|
|
@ -49,11 +49,8 @@ def tg_compile(flags, model_name): |
|
|
|
|
|
|
|
|
|
|
|
# Compile small models |
|
|
|
# Compile small models |
|
|
|
for model_name in ['driving_vision', 'driving_policy', 'dmonitoring_model']: |
|
|
|
for model_name in ['driving_vision', 'driving_policy', 'dmonitoring_model']: |
|
|
|
flags = { |
|
|
|
TG_DEVICE = {'larch64': 'QCOM','Darwin': 'CPU'}.get(arch, 'LLVM') |
|
|
|
'larch64': 'DEV=QCOM', |
|
|
|
tg_compile(f'DEV={TG_DEVICE}', model_name) |
|
|
|
'Darwin': 'DEV=CPU IMAGE=0 JIT=2', |
|
|
|
|
|
|
|
}.get(arch, 'DEV=LLVM LLVMOPT=1 BEAM=0 IMAGE=0 JIT=2') |
|
|
|
|
|
|
|
tg_compile(flags, model_name) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Compile BIG model if USB GPU is available |
|
|
|
# Compile BIG model if USB GPU is available |
|
|
|
if "USBGPU" in os.environ: |
|
|
|
if "USBGPU" in os.environ: |
|
|
|