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

Merge branch 'master' into warp_fun

Browse Source
This commit is contained in:
George Hotz
2025-06-30 18:18:52 -07:00
committed by GitHub
parent fc59db0aaa 5628e2054c
commit 5907f2d443
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/codegen/gpudims.py
+1
View File
@@ -57,6 +57,7 @@ def add_gpudims(ctx:Renderer, s:UOp):
s_topo = list(s.toposort())
if any(x.op is Ops.SPECIAL for x in s_topo): return None
ranges = sorted([x for x in s_topo if x.op is Ops.RANGE and x.arg < (ki.global_dims+ki.local_dims)], key=lambda x: x.arg)
if not len(ranges): return None
global_shape = tuple([ssimplify(r.src[0]) for r in ranges if r.arg < ki.global_dims])
local_shape = tuple([ssimplify(r.src[0]) for r in ranges if r.arg >= ki.global_dims])
if ki.dont_use_locals: