Skip to content

Commit

Permalink
remove a test
Browse files Browse the repository at this point in the history
  • Loading branch information
MasterJH5574 committed Jul 19, 2021
1 parent 1cd6956 commit 60b7cdd
Showing 1 changed file with 0 additions and 15 deletions.
15 changes: 0 additions & 15 deletions tests/python/unittest/test_tir_schedule_reduction.py
Original file line number Diff line number Diff line change
Expand Up @@ -338,21 +338,6 @@ def rowsum_zero_dim_rfactor(a: ty.handle, b: ty.handle) -> None:
B[()] = (B[()] + B_rf[vi0_1])


@tvm.script.tir
def rowsum_with_predicate(a: ty.handle, b: ty.handle) -> None:
A = tir.match_buffer(a, (10, 1027))
B = tir.match_buffer(b, (10,))

for i0, i1 in tir.grid(10, 1027):
with tir.block([10, tir.reduce_axis(0, 1027)], "B") as [vi, vk]:
tir.where(i0 > 1)
tir.bind(vi, i0)
tir.bind(vk, i1)
with tir.init():
B[vi] = 0.0
B[vi] = B[vi] + A[vi, vk]


@tvm.script.tir
def multiple_reduction_blocks(a: ty.handle, f: ty.handle) -> None:
A = tir.match_buffer(a, (16, 16, 16))
Expand Down

0 comments on commit 60b7cdd

Please sign in to comment.