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

fix on cpu

Browse Source
This commit is contained in:
geohot
2025-10-07 16:43:52 +08:00
parent 51f3a5cbb4
commit f129d75ee5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/thunder/clone.py
+2 -2
View File
@@ -45,8 +45,8 @@ def mma_AB(outacc:UOp, a:UOp, b:UOp, *endrngs):
if __name__ == "__main__":
# TODO: support string ranges
tg_id_y = UOp.range(M // (M_BLOCK * TILE_DIM), -3, AxisType.GLOBAL)
tg_id_x = UOp.range(N // (N_BLOCK * TILE_DIM), -2, AxisType.GLOBAL)
tg_id_y = UOp.range(M // (M_BLOCK * TILE_DIM), -3, AxisType.GLOBAL if Device.DEFAULT != "CPU" else AxisType.LOOP)
tg_id_x = UOp.range(N // (N_BLOCK * TILE_DIM), -2, AxisType.GLOBAL if Device.DEFAULT != "CPU" else AxisType.LOOP)
gl_d = glbl("gl0_d", dtypes.float, (N, M))
gl_a = glbl("gl1_a", dtypes.float, (N, K))