[luci/pass] Fix typos in FuseXxxToFC test (Samsung#13449)

This will fix typos in fuse Add/Mul to FullyConnected pass unit tests. ONE-DCO-1.0-Signed-off-by: SaeHie Park <[email protected]>
dayo09 · Jul 17, 2024 · 8e5ad80 · 8e5ad80
1 parent fb3bfa0
commit 8e5ad80
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/compiler/luci/pass/src/FuseAddToFullyConnectedBiasPass.test.cpp b/compiler/luci/pass/src/FuseAddToFullyConnectedBiasPass.test.cpp
@@ -59,17 +59,17 @@ template <loco::DataType DT> class FuseAddToFullyConnectedBiasPassTestGraph : pu
     _add_s->size<DT>(3);
     for (uint32_t i = 0; i < 3; ++i)
     {
-      _add_s->at<DT>(0) = 1.0f;
+      _add_s->at<DT>(i) = 1.0f;
     }
 
     _fc_w->rank(2);
-    _fc_w->dim(0) = 3;
+    _fc_w->dim(0) = 6;
     _fc_w->dim(1) = 4;
     _fc_w->dtype(DT);
     _fc_w->size<DT>(4 * 6);
     for (uint32_t i = 0; i < 4 * 6; ++i)
     {
-      _fc_w->at<DT>(0) = 1.0f;
+      _fc_w->at<DT>(i) = 1.0f;
     }
 
     _fc_b->rank(1);
@@ -78,13 +78,13 @@ template <loco::DataType DT> class FuseAddToFullyConnectedBiasPassTestGraph : pu
     _fc_b->size<DT>(6);
     for (uint32_t i = 0; i < 6; ++i)
     {
-      _fc_b->at<DT>(0) = 1.0f;
+      _fc_b->at<DT>(i) = 1.0f;
     }
 
     _add->x(input());
     _add->y(_add_s);
     _fc->input(_add);
-    _fc->weights(_fc_b);
+    _fc->weights(_fc_w);
     _fc->bias(_fc_b);
 
     output()->from(_fc);

diff --git a/compiler/luci/pass/src/FuseMulToFullyConnectedWeightsPass.test.cpp b/compiler/luci/pass/src/FuseMulToFullyConnectedWeightsPass.test.cpp
@@ -59,17 +59,17 @@ template <loco::DataType DT> class FuseMulToFullyConnectedWeightsPassTestGraph :
     _mul_s->size<DT>(3);
     for (uint32_t i = 0; i < 3; ++i)
     {
-      _mul_s->at<DT>(0) = 1.0f;
+      _mul_s->at<DT>(i) = 1.0f;
     }
 
     _fc_w->rank(2);
-    _fc_w->dim(0) = 3;
+    _fc_w->dim(0) = 6;
     _fc_w->dim(1) = 4;
     _fc_w->dtype(DT);
     _fc_w->size<DT>(4 * 6);
     for (uint32_t i = 0; i < 4 * 6; ++i)
     {
-      _fc_w->at<DT>(0) = 1.0f;
+      _fc_w->at<DT>(i) = 1.0f;
     }
 
     _fc_b->rank(1);
@@ -78,13 +78,13 @@ template <loco::DataType DT> class FuseMulToFullyConnectedWeightsPassTestGraph :
     _fc_b->size<DT>(6);
     for (uint32_t i = 0; i < 6; ++i)
     {
-      _fc_b->at<DT>(0) = 1.0f;
+      _fc_b->at<DT>(i) = 1.0f;
     }
 
     _mul->x(input());
     _mul->y(_mul_s);
     _fc->input(_mul);
-    _fc->weights(_fc_b);
+    _fc->weights(_fc_w);
     _fc->bias(_fc_b);
 
     output()->from(_fc);