diff --git a/examples/openpilot/compile3.py b/examples/openpilot/compile3.py index 66c34f7248b25..bb604fa3dd6e7 100644 --- a/examples/openpilot/compile3.py +++ b/examples/openpilot/compile3.py @@ -1,8 +1,9 @@ import os, sys, pickle, time, re import numpy as np -if "FLOAT16" not in os.environ: os.environ["FLOAT16"] = "1" -if "IMAGE" not in os.environ: os.environ["IMAGE"] = "2" -if "NOLOCALS" not in os.environ: os.environ[ "NOLOCALS" ] = "1" +if "DEV" not in os.environ or os.environ.get("DEV") == "QCOM": + if "FLOAT16" not in os.environ: os.environ["FLOAT16"] = "1" + if "IMAGE" not in os.environ: os.environ["IMAGE"] = "2" + if "NOLOCALS" not in os.environ: os.environ["NOLOCALS"] = "1" if "JIT_BATCH_SIZE" not in os.environ: os.environ["JIT_BATCH_SIZE"] = "0" from tinygrad import fetch, Tensor, TinyJit, Context, GlobalCounters, Device, dtypes