tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 20:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

ln

Browse Source
This commit is contained in:
geohot
2026-03-12 08:23:06 +08:00
parent 5ed68aeed5
commit b384c27314
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/gemm/amd_uop_matmul.py
-1
View File
@@ -116,7 +116,6 @@ def hand_spec_kernel3():
A_local_slice = A_local[k, :].reshape(WAVES_PER_BLOCK_M, REG_TILES_PER_WAVE_M, LANES_PER_WAVE_M, TM)[waveIdy, :, laneIdy, :]
A_col = copy(A_col, A_local_slice , 300, set=True, upcast=True)
B_row = UOp.placeholder((REG_TILES_PER_WAVE_N, TN), dtypes.float, slot=1, addrspace=AddrSpace.REG)
B_local_slice = B_local[k, :].reshape(WAVES_PER_BLOCK_N, REG_TILES_PER_WAVE_N, LANES_PER_WAVE_N, TN)[waveIdx, :, laneIdx, :]
B_row = copy(B_row, B_local_slice, 400, set=True, upcast=True)