Skip to content

Commit 69fe7f5

Browse files
authored
Fix TK gemmbench after renaming of tkw.reduction() to tkw.iterate() (#77)
See iree-org/iree-turbine@5cee489
1 parent dc1784c commit 69fe7f5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

iree_kernel_benchmark/gemmbench/gemm_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def gemm(
247247

248248
# This microkernel encodes the fact that if the reduction
249249
# dimension were tiled, then we would need to materialize a loop.
250-
@tkw.reduction(K, init_args=[c_reg])
250+
@tkw.iterate(K, init_args=[c_reg])
251251
def repeat(acc: tkl.Register[M, N, tkl.f32]) -> tkl.Register[M, N, tkl.f32]:
252252
# a_reg: tkw.Register[M, K, tkl.f16]
253253
a_reg = tkw.read(a, elements_per_thread=LOAD_ELEMS_PER_THREAD)

0 commit comments

Comments
 (0)