From ba1d1142beee45cfdda97d1c1659a5697ddbfa36 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Fri, 31 Oct 2025 09:02:35 +0800 Subject: [PATCH] remove mod --- extra/gemm/amd_uop_matmul.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extra/gemm/amd_uop_matmul.py b/extra/gemm/amd_uop_matmul.py index be69031aef..66bdcb68cc 100644 --- a/extra/gemm/amd_uop_matmul.py +++ b/extra/gemm/amd_uop_matmul.py @@ -90,7 +90,7 @@ def hand_spec_kernel3(kernel5=getenv("K5", 0)): i = UOp.range(BN * BK // BLOCK_SIZE, 1) index_x = threadIdx_x % BN index_y = (threadIdx_x // BN) + (BLOCK_SIZE // BN) * i - Bs_store = Bs[index_y, index_x % BN].store(b[kId_range, index_y, blockIdx_x, index_x]).end(i) + Bs_store = Bs[index_y, index_x].store(b[kId_range, index_y, blockIdx_x, index_x]).end(i) a = a.reshape((N // BM, BM, N // BK, BK)) i = UOp.range(BM * BK // BLOCK_SIZE, 2)