mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

temporary debugs for CI crash

Browse source
This commit is contained in:
Szymon Ożóg 2023-10-07 17:44:57 +02:00 • committed by GitHub
commit fee3ea96c8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
test/test_uops.py
View file

@ -8,6 +8,7 @@ from tinygrad.codegen.linearizer import UOps, UOp
def _uops_to_prg(uops):
src = Device[Device.DEFAULT].renderer("test", uops)
print(src, Device.DEFAULT, Device[Device.DEFAULT].renderer)
return ASTRunner("test", src[0] if getenv("TRITON") else src, [1], [1], runtime_args=src[2] if getenv("TRITON") else {}).build(Device[Device.DEFAULT].runtime)
def uop(uops:List[UOp], uop:UOps, dtype:Optional[DType], vin:Tuple[UOp, ...], arg:Any=None) -> UOp: