red fix

microsoft · Sep 17, 2024 · 62de446 · 62de446
1 parent 0acc369
commit 62de446
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/testing/python/operators/test_general_matmul_tile_schedule.py b/testing/python/operators/test_general_matmul_tile_schedule.py
@@ -16,8 +16,12 @@
 
 
 def check_reduce(rt_mod):
-    source = rt_mod.imported_modules[0].get_source()
-    assert "red_buf" in source
+    # source = rt_mod.imported_modules[0].get_source()
+    # assert "red_buf" in source
+    # TODO(lei): After improve lower_thraed_all_reduce pass
+    # The red_buf has been merged into dynamic shared memory
+    # ref to: https://github.com/microsoft/BitBLAS/pull/183
+    return True
 
 
 # fmt: off