Test case rollback and fixes.

Author: srkreddy1238

src/relax/transform/legalize_ops.cc

@@ -161,7 +161,7 @@ class LegalizeMutator : public ExprMutator {
     return NullOpt;
   }
 
-  Expr UpdateVDeviceOutStructInfo(Expr expr, Call& visited_call) {
+  Expr UpdateVDeviceOutStructInfo(Expr expr, const Call& visited_call) {
     static const auto& infer_struct_info_map = Op::GetAttrMap<FInferStructInfo>("FInferStructInfo");
     static const Op& call_tir_op = Op::Get("relax.call_tir");
     auto* op_node = visited_call->op.as<OpNode>();

src/relax/transform/optimize_to_vdevice_for_scope_change.cc

@@ -102,8 +102,7 @@ class CollectConsumerDetails : public ExprVisitor {
  public:
   using ExprVisitor::VisitExpr_;
 
-  Map<Expr, Array<Expr>> Collect(
-      const IRModule& mod, Function func, const Target& target) {
+  Map<Expr, Array<Expr>> Collect(const IRModule& mod, Function func, const Target& target) {
     mod_ = mod;
     target_ = target;
     VisitExpr(func->body);
@@ -159,7 +158,6 @@ class CollectConsumerDetails : public ExprVisitor {
   }
 
  private:
-
   /* Map of each Var consumption by a call node */
   Map<Expr, Array<Expr>> consumers;
   Map<Expr, Expr> arg_to_binding;
@@ -172,7 +170,8 @@ namespace transform {
 Pass OptimizeToVDeviceForScopeChange() {
   auto pass_func = [=](Function func, IRModule mod, PassContext pc) {
     /* here Target doesn't matter as the consumers we use only to find multiple consumers */
-    auto consumers = CollectConsumerDetails().Collect(mod, Downcast<Function>(func), Target("opencl"));
+    auto consumers =
+        CollectConsumerDetails().Collect(mod, Downcast<Function>(func), Target("opencl"));
     auto [pattern, rewriter] = CreatePatterns(consumers);
     return RewriteCall(pattern, rewriter, func);
   };

src/relax/transform/remove_redundant_assignments.cc

@@ -44,7 +44,7 @@ class RemoveRedundantAssignments : public ExprMutator {
  public:
   using ExprMutator::VisitExpr_;
 
-  IRModule Run(IRModule& mod) {
+  IRModule Run(const IRModule& mod) {
     mod_ = mod;
     for (const auto& [gv, func] : mod_->functions) {
       if (func->IsInstance<relax::FunctionNode>()) {

tests/python/relax/test_transform_fuse_tir.py

@@ -1049,7 +1049,7 @@ def concatenate(
             ),
             T_concat: T.Buffer((T.int64(2), T.int64(4), T.int64(64), T.int64(64)), "float32"),
         ):
-            T.func_attr({"op_pattern": 2, "tir.noalias": T.bool(True)})
+            T.func_attr({"tir.noalias": T.bool(True)})
             for ax0, ax1, ax2, ax3 in T.grid(T.int64(2), T.int64(4), T.int64(64), T.int64(64)):
                 with T.block("T_concat"):
                     v_ax0, v_ax1, v_ax2, v_ax3 = T.axis.remap("SSSS", [ax0, ax1, ax2, ax3])
@@ -1069,7 +1069,7 @@ def transpose2(
             rxplaceholder: T.Buffer((T.int64(2), T.int64(4), T.int64(64), T.int64(64)), "float32"),
             T_transpose: T.Buffer((T.int64(2), T.int64(64), T.int64(64), T.int64(4)), "float32"),
         ):
-            T.func_attr({"op_pattern": 2, "tir.noalias": T.bool(True)})
+            T.func_attr({"tir.noalias": T.bool(True)})
             for ax0, ax1, ax2, ax3 in T.grid(T.int64(2), T.int64(64), T.int64(64), T.int64(4)):
                 with T.block("T_transpose"):
                     v_ax0, v_ax1, v_ax2, v_ax3 = T.axis.remap("SSSS", [ax0, ax1, ax2, ax3])
@@ -1117,7 +1117,7 @@ def fused_concatenate_transpose2(
                 (T.int64(2), T.int64(64), T.int64(64), T.int64(4)), "float32"
             ),
         ):
-            T.func_attr({"op_pattern": 2, "tir.noalias": T.bool(True)})
+            T.func_attr({"tir.noalias": T.bool(True)})
             T_concat_handle_intermediate = T.alloc_buffer(
                 (T.int64(2), T.int64(4), T.int64(64), T.int64(64))
             )
@@ -1242,7 +1242,7 @@ def reshape(
             A: T.Buffer((T.int64(4), T.int64(8), T.int64(2048)), "float32"),
             T_reshape: T.Buffer((T.int64(4), T.int64(8), T.int64(32), T.int64(64)), "float32"),
         ):
-            T.func_attr({"op_pattern": 2, "tir.noalias": T.bool(True)})
+            T.func_attr({"tir.noalias": T.bool(True)})
             # with T.block("root"):
             for ax0, ax1, ax2, ax3 in T.grid(T.int64(4), T.int64(8), T.int64(32), T.int64(64)):
                 with T.block("T_reshape"):
@@ -1307,7 +1307,7 @@ def fused_reshape(
                 (T.int64(4), T.int64(8), T.int64(32), T.int64(64)), "float32"
             ),
         ):
-            T.func_attr({"op_pattern": 2, "tir.noalias": T.bool(True)})
+            T.func_attr({"tir.noalias": T.bool(True)})
             # with T.block("root"):
             for ax0, ax1, ax2, ax3 in T.grid(T.int64(4), T.int64(8), T.int64(32), T.int64(64)):
                 with T.block("T_reshape"):