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

get program fields in __post_init__ [run_process_replay] (#5878)

Browse Source 
* get program fields in __post_init__ [run_process_replay]

* remove print
This commit is contained in:
George Hotz
2024-08-02 09:57:12 -07:00
committed by GitHub
parent 8611fa6c99
commit 23e8c39288
3 changed files with 36 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_uops.py
View File

@@ -19,7 +19,8 @@ def _uops_to_prg(uops_list, print_uops=False):
src = Device[Device.DEFAULT].renderer.render("test", uops.uops)
if print_uops: uops.print()
has_local = Device[Device.DEFAULT].renderer.has_local
return CompiledRunner(Program("test", src, Device.DEFAULT, [1,1,1] if has_local else None, [1,1,1] if has_local else None, uops=uops.uops))
return CompiledRunner(Program("test", src, Device.DEFAULT, uops=uops.uops,
global_size=[1,1,1] if has_local else None, local_size=[1,1,1] if has_local else None))
def uop(uops:List[UOp], uop:UOps, dtype:Optional[DType], src:Tuple[UOp, ...], arg:Any=None) -> UOp:
uops.append(UOp(uop, dtype, tuple(src), arg))