github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

MetaOps.KERNEL (#5543)

Browse Source
This commit is contained in:
George Hotz
2024-07-17 19:41:23 -07:00
committed by GitHub
parent d3b098299d
commit fa7e734b49
19 changed files with 39 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/handcode_opt.py
View File

@@ -69,7 +69,7 @@ if __name__ == "__main__":
print(f"optimizing for {Device.DEFAULT}")
sched = globals()[f"get_sched_{getenv('MODEL', 'resnet')}"]()
sched = [x for x in sched if x.ast.op is MetaOps.SINK]
sched = [x for x in sched if x.ast.op is MetaOps.KERNEL]
# focus on one kernel
if getenv("KERNEL", -1) >= 0: sched = sched[getenv("KERNEL", -1):getenv("KERNEL", -1)+1]