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

add smin/smax (#7253)

Browse Source 
* add smin/smax

* don't create var with var

* better test errors

* add failing test

* enable shape simplification

* fix tests

* Update view.py

* simpler and simplify
This commit is contained in:
George Hotz
2024-10-24 15:10:49 +07:00
committed by GitHub
parent de7b9d7c42
commit 532b7b018c
6 changed files with 36 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/helpers.py
View File

@@ -21,7 +21,7 @@ def assert_jit_cache_len(fxn, expected_len):
return
# until we have a better way of typing the prg in ExecItem
if issubclass(type(fxn.jit_cache[0].prg), Runner) and not type(fxn.jit_cache[0].prg).__name__.endswith('Graph'):
assert len(fxn.jit_cache) == expected_len, len(fxn.jit_cache)
assert len(fxn.jit_cache) == expected_len, f"expected {expected_len}, got {len(fxn.jit_cache)}"
else:
assert len(fxn.jit_cache) == 1, len(fxn.jit_cache)
# until we have a better way of typing the prg in ExecItem