nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

remove mod

Browse Source
This commit is contained in:
geohot
2025-10-31 09:02:35 +08:00
parent 59ad5d51f5
commit ba1d1142be
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/gemm/amd_uop_matmul.py
+1 -1
View File
@@ -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)