github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 22:48:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

hand_coded_optimizations returns list[Opt] [pr] (#9938)

Browse Source 
new api looks like `k.apply_opts(hand_coded_optimizations(k))`
This commit is contained in:
chenyu
2025-04-19 20:26:59 -04:00
committed by GitHub
parent 720f20865b
commit 6c30948df6
14 changed files with 27 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/optimization/get_action_space.py
View File

@@ -24,7 +24,7 @@ if __name__ == "__main__":
for ast_str in tqdm(ast_strs):
lin = ast_str_to_lin(ast_str)
#if not lin.apply_tensor_cores():
lin = hand_coded_optimizations(lin)
lin.apply_opts(hand_coded_optimizations(lin))
test_rebuild(lin)
# confirm linearize can be called twice
uops1 = lin.linearize().uops

2
extra/optimization/test_net.py
View File

@@ -35,7 +35,7 @@ if __name__ == "__main__":
rawbufs = bufs_from_lin(lin)
linhc = deepcopy(lin)
linhc = hand_coded_optimizations(linhc)
linhc.applied_opts(hand_coded_optimizations(linhc))
tmhc = time_linearizer(linhc, rawbufs)
print(f"{tmhc*1e6:10.2f} HC ", linhc.colored_shape())