diff --git a/testing/python/operators/test_general_matmul_tile_schedule.py b/testing/python/operators/test_general_matmul_tile_schedule.py
index ef1728efd..58f595984 100644
--- 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