[loco] fix MultiDialectShapeInferenceRule.test (#6913)

This will add CanonicalShapeInferenceRule to MultiDialectShapeInferenceRule as there are Canonical dialect node Pull and Push in the graph

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/compiler/loco/src/Service/MultiDialectShapeInferenceRule.test.cpp b/compiler/loco/src/Service/MultiDialectShapeInferenceRule.test.cpp
index 5198d1d..ace7947 100644 (file)
--- a/compiler/loco/src/Service/MultiDialectShapeInferenceRule.test.cpp
+++ b/compiler/loco/src/Service/MultiDialectShapeInferenceRule.test.cpp
@@ -14,10 +14,12 @@
  * limitations under the License.
  */
 
+#include "loco/Service/CanonicalShapeInferenceRule.h"
 #include "loco/Service/MultiDialectShapeInferenceRule.h"
 #include "loco/Service/ShapeInference.h"
 
 #include <loco/IR/Dialect.h>
+#include <loco/IR/CanonicalDialect.h>
 
 #include <gtest/gtest.h>
 
@@ -103,12 +105,15 @@ TEST(MultiDialectShapeInferenceRuleTest, test1)
   ASSERT_FALSE(loco::shape_known(t45_node));
 
   // Run Type Inference
+  loco::CanonicalShapeInferenceRule canonical_rule;
   TestShapeInferenceRule<2, 3> t23_rule;
   TestShapeInferenceRule<4, 5> t45_rule;
 
   loco::MultiDialectShapeInferenceRule rules;
 
-  rules.bind(TestDialect<2, 3>::get(), &t23_rule).bind(TestDialect<4, 5>::get(), &t45_rule);
+  rules.bind(loco::CanonicalDialect::get(), &canonical_rule)
+      .bind(TestDialect<2, 3>::get(), &t23_rule)
+      .bind(TestDialect<4, 5>::get(), &t45_rule);
 
   loco::apply(&rules).to(g.get());