9c7833faf9
* Vegetarian Filet o Fish model * fix.. atc.. * test cluster_speed_limit * fix.. cluster_speed_limit.. 2 * fix.. clusterspeedlimit3 * cruise speed to roadlimit speed * fix.. * fix.. eng * deltaUp/Down for lanechange * fix.. atc desire... * fix.. * ff * ff * fix.. * fix.. eng * fix engsound * Update desire_helper.py * fix.. connect... * fix curve_min speed * Revert "fix curve_min speed" This reverts commit fcc9c2eb14eb3504abef3e420db93e8882e56f37. * Reapply "fix curve_min speed" This reverts commit 2d2bba476c58a7b4e13bac3c3ad0e4694c95515d. * fix.. auto speed up.. roadlimit * fix.. atc auto lanechange... * Update desire_helper.py * Update cruise.py * debug atc... * fix.. waze alert offset.. * fix.. * test atc.. * fix.. * fix.. atc * atc test.. * fix.. atc * fix.. atc2 * fix.. atc3 * KerryGold Model. latsmooth_sec = 0.0 * lat smooth seconds 0.13 * fix comment * fix.. auto cruise, and speed unit * change lanemode switching. * erase mazda lkas button.
32 lines
951 B
Python
32 lines
951 B
Python
import random
|
|
from extra.optimization.helpers import load_worlds, ast_str_to_lin
|
|
from tinygrad.engine.search import actions
|
|
from tinygrad.codegen.kernel import Kernel
|
|
from tinygrad.codegen.heuristic import hand_coded_optimizations
|
|
from tinygrad.helpers import tqdm
|
|
|
|
tactions = set()
|
|
def test_rebuild(lin):
|
|
linr = Kernel(lin.ast)
|
|
for o in lin.applied_opts:
|
|
assert o in actions, f"{o} is not 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)
|
|
random.shuffle(ast_strs)
|
|
ast_strs = ast_strs[:2000]
|
|
for ast_str in tqdm(ast_strs):
|
|
lin = ast_str_to_lin(ast_str)
|
|
#if not lin.apply_tensor_cores():
|
|
lin.apply_opts(hand_coded_optimizations(lin))
|
|
test_rebuild(lin)
|
|
|
|
print(len(tactions), len(actions))
|
|
print(sorted(list(tactions)))
|