Switch PC running to tinygrad with LLVM (#34660)

* try llvm on cpu

* update lock

* Make sure test fails

* With BEAM=3 its fast

* small fixes

* more fixes

* Fix macos

* do dm too

* rm ort
This commit is contained in:
Harald Schäfer
2025-02-24 14:51:10 -08:00
committed by GitHub
parent c13b1f95e1
commit d41311ca20
7 changed files with 96 additions and 193 deletions

View File

@@ -41,8 +41,10 @@ lenv.Command(fn + "_metadata.pkl", [fn + ".onnx"] + tinygrad_files + script_file
pythonpath_string = 'PYTHONPATH="${PYTHONPATH}:' + env.Dir("#tinygrad_repo").abspath + '"'
if arch == 'larch64':
device_string = 'QCOM=1'
else:
elif arch == 'Darwin':
device_string = 'CLANG=1 IMAGE=0'
else:
device_string = 'LLVM=1 LLVMOPT=1 BEAM=0 IMAGE=0'
for model_name in ['supercombo', 'dmonitoring_model']:
fn = File(f"models/{model_name}").abspath