github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' into call_is_kernel

Browse Source
This commit is contained in:
George Hotz
2026-02-06 11:33:06 +08:00
committed by GitHub
parent 9183355104 d41836f135
commit 7391ab77f2
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/schedule/indexing.py
View File

@@ -19,8 +19,7 @@ def realize_srcs(ctx:dict[UOp, None], rb:UOp) -> None:
def realize_assign(ctx:dict[UOp, None], a:UOp) -> None:
if a.src[1].op not in ALWAYS_CONTIGUOUS: ctx[a.src[1]] = None
# if it's a kernel, we don't realize it
if a.src[1].op is not Ops.KERNEL: ctx[a] = None
ctx[a] = None
pm_generate_realize_map = PatternMatcher([
# always realize SINK src