github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

add renderer class (#4524)

Browse Source 
* add renderer class

* tests pass

* fix pylint

* fix tensor cores
This commit is contained in:
George Hotz
2024-05-10 21:40:02 -07:00
committed by GitHub
parent b00b6b16f0
commit 347a3acb37
31 changed files with 536 additions and 527 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/optimization/search.py
View File

@@ -25,7 +25,7 @@ if __name__ == '__main__':
for i, ast_str in enumerate(ast_strs):
print(f"optimizing {i}/{len(ast_strs)}\nast={ast_str}")
lin = ast_str_to_lin(ast_str, opts=device.compiler.compiler_opts)
lin = ast_str_to_lin(ast_str, opts=device.renderer)
rawbufs = bufs_from_lin(lin)
lin = beam_search(lin, rawbufs, getenv("BEAM", 8), bool(getenv("BEAM_ESTIMATE", 1)))