[Hotfix] A unittest (#19)

junrushao · Jan 22, 2022 · 5793c98 · 5793c98
1 parent dbe279f
commit 5793c98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_meta_schedule_postproc_rewrite_unbound_block.py b/tests/python/unittest/test_meta_schedule_postproc_rewrite_unbound_block.py
@@ -99,15 +99,15 @@ def main(A: T.Buffer[(1, 256, 256), "float32"], D: T.Buffer[(1,), "float32"]) ->
                     with T.block("C"):
                         b = T.axis.S(1, 0)
                         i, j = T.axis.remap("RR", [i1, i2])
-                        T.where(i0_fused_0 * 32 + i0_fused_1 < 1)
+                        T.where(i0_fused_1 < 1)
                         with T.init():
                             C[b] = T.float32(0)
                         C[b] = C[b] + A[b, i, j] * A[b, i, j]
         for i0_fused_0 in T.thread_binding(1, thread="blockIdx.x"):
             for i0_fused_1 in T.thread_binding(32, thread="threadIdx.x"):
                 with T.block("D"):
                     b = T.axis.S(1, 0)
-                    T.where(i0_fused_0 * 32 + i0_fused_1 < 1)
+                    T.where(i0_fused_1 < 1)
                     D[b] = T.sqrt(C[b], dtype="float32")