From 1cb1cfe594aea605ec9508a773c76e0097460e32 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 26 Aug 2019 16:32:40 +0900 Subject: [PATCH] [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 --- compiler/loco/src/Service/MultiDialectShapeInferenceRule.test.cpp | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/compiler/loco/src/Service/MultiDialectShapeInferenceRule.test.cpp b/compiler/loco/src/Service/MultiDialectShapeInferenceRule.test.cpp index 5198d1d..ace7947 100644 --- 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 +#include #include @@ -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()); -- 2.7.4