fix file name

This commit is contained in:
Jason Wen
2025-01-05 08:27:23 -05:00
parent 41fef87680
commit 485eef68da

View File

@@ -66,7 +66,7 @@ if arch == "larch64" or GetOption('pc_thneed'):
if not GetOption('pc_thneed'):
# use FLOAT16 on device for speed + don't cache the CL kernels for space
tinygrad_opts = 'FLOAT16=1 PYOPENCL_NO_CACHE=1'
cmd = f'{pythonpath_string} {tinygrad_opts} python3 {Dir("#sunnypilot/modeld").abspath}/redirect_tinygrad.py {fn}.onnx {fn}.thneed'
cmd = f'{pythonpath_string} {tinygrad_opts} python3 {Dir("#sunnypilot/modeld").abspath}/run_compile.py {fn}.onnx {fn}.thneed'
lenv.Command(fn + ".thneed", [fn + ".onnx"] + tinygrad_files, cmd)