mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-07 03:00:26 -04:00
c36d306606
* create cache for q learning * make linter happy * global beam * where it belongs * bugfix * ditch the kopt, use the beam * faster lin and DEBUG=2 okay * remove kopt, move search to features
28 lines
778 B
Python
28 lines
778 B
Python
from tqdm import tqdm
|
|
from extra.optimization.helpers import load_worlds, ast_str_to_lin
|
|
from tinygrad.features.search import actions
|
|
from tinygrad.codegen.linearizer import Linearizer
|
|
|
|
tactions = set()
|
|
def test_rebuild(lin):
|
|
linr = Linearizer(lin.ast)
|
|
for o in lin.applied_opts:
|
|
assert o in actions
|
|
tactions.add(o)
|
|
linr.apply_opt(o)
|
|
|
|
assert len(lin.sts) == len(linr.sts)
|
|
for st1,st2 in zip(lin.sts, linr.sts):
|
|
assert st1 == st2, f"{st1} != {st2}"
|
|
|
|
if __name__ == "__main__":
|
|
ast_strs = load_worlds(False, False, False)
|
|
for ast_str in tqdm(ast_strs):
|
|
lin = ast_str_to_lin(ast_str)
|
|
#if not lin.apply_tensor_cores():
|
|
lin.hand_coded_optimizations()
|
|
test_rebuild(lin)
|
|
|
|
print(len(tactions), len(actions))
|
|
print(sorted(list(tactions)))
|