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

don't run linearize().uop tests in get_action_space test (#10766)

Browse Source 
* don't run linearize().uop tests in get_action_space test

this part takes 2 minutes in CI and has nothing to do with action space. also not sure if the "for some reason" comment is still relevant

* -n=auto test/models
This commit is contained in:
chenyu
2025-06-10 14:23:53 -07:00
committed by GitHub
parent 52c49dd4f3
commit 5e7ad70aae
2 changed files with 1 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
extra/optimization/get_action_space.py
View File

@@ -26,15 +26,6 @@ if __name__ == "__main__":
#if not lin.apply_tensor_cores():
lin.apply_opts(hand_coded_optimizations(lin))
test_rebuild(lin)
# confirm linearize can be called twice
uops1 = lin.linearize().uops
uops2 = lin.linearize().uops
for x,y in zip(uops1, uops2):
# for some reason DEFINE_ACC is changing the arg
if x.op != y.op or x.dtype != y.dtype: # or x.arg != y.arg:
uops1.print()
uops2.print()
raise Exception(f"UOPS MISMATCH {x} {y}")
print(len(tactions), len(actions))
print(sorted(list(tactions)))