From f129d75ee5215939db2d1a32fc89ad93c3308b2e Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 7 Oct 2025 16:43:52 +0800 Subject: [PATCH] fix on cpu --- extra/thunder/clone.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extra/thunder/clone.py b/extra/thunder/clone.py index 72453533d4..23877f20b9 100644 --- a/extra/thunder/clone.py +++ b/extra/thunder/clone.py @@ -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))