Handle rank 0 and 1

Samsung · Aug 12, 2024 · 8977ef9 · 8977ef9
1 parent 9e22b26
commit 8977ef9
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/compiler/luci/pass/src/FuseMulWithFullyConnectedPass.cpp b/compiler/luci/pass/src/FuseMulWithFullyConnectedPass.cpp
@@ -136,15 +136,20 @@ bool fuse_mul_with_fc(luci::CircleFullyConnected *fc)
   // Check that all dimensions are ones, checks broadcast capabilites.
   // Last dimesion of multiplication must be compatible with FC.
   // N-D case (N>1):
-  if (multiplication->rank() >= 1)
+  if (multiplication->rank() > 1)
   {
     // Check channel-wise broadcasting:
     for (uint32_t i = 0; i < rank - 1; i++)
       RETURN_FALSE_UNLESS(multiplication->dim(i).value() == 1);
     // Check the last dimesion of Mul is the same with the first dimension of FullyConnected
     RETURN_FALSE_UNLESS(multiplication->dim(rank - 1) == weights->dim(0));
   }
-  // Scalar case:
+  // 1-D or scalar case:
+  else if (multiplication->rank() == 1)
+  {
+    RETURN_FALSE_UNLESS(multiplication->size<loco::DataType::FLOAT32>() == 1 ||
+                        multiplication->size<loco::DataType::FLOAT32>() == weights->dim(0));
+  }
   else if (multiplication->rank() == 0)
   {
     RETURN_FALSE_UNLESS(multiplication->size<loco::DataType::FLOAT32>() == 1);