diff --git a/src/relax/transform/fuse_tir.cc b/src/relax/transform/fuse_tir.cc index 612e1459c826..fe247645dc24 100644 --- a/src/relax/transform/fuse_tir.cc +++ b/src/relax/transform/fuse_tir.cc @@ -139,6 +139,16 @@ class SymbolicMatcher : ExprFunctor(); + if (rhs) { + VisitExpr(op->true_value, rhs->true_value); + VisitExpr(op->false_value, rhs->false_value); + } else { + must_prove_ = must_prove_ && (GetRef(op) == other); + } + } + arith::Analyzer* analyzer_; Map* var_remap_; PrimExpr must_prove_ = Bool(true);