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

Ops.ALU is no more, the arg is just an op (#7525)

Browse Source 
* op arg alu [pr]

* more

* more passing

* fix more tests

* more tests passing

* fix single failing test

* so much cleaner

* noop to not have process replay trigger

* fix ptx
This commit is contained in:
George Hotz
2024-11-05 00:22:22 +08:00
committed by GitHub
parent 36488a2a43
commit 99bd4372a5
22 changed files with 171 additions and 185 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/fuzz_linearizer.py vendored
View File

@@ -252,7 +252,7 @@ def fuzz_linearizer(lin: Kernel, rtol=1e-2, atol=1e-2, opts_list=None):
def _is_simple(lin: Kernel) -> bool:
if len(lin.ast.src) > 1: return False
ast:UOp = lin.ast.src[0]
if ast.src[0].arg is UnaryOps.CAST and ast.src[0].src[0].op is Ops.LOAD: return True
if ast.src[0].op is UnaryOps.CAST and ast.src[0].src[0].op is Ops.LOAD: return True
return False
if __name__ == "__main__":