OPT=99 IMAGE=2

commaai · Sep 20, 2024 · 3dee3e0 · 3dee3e0
1 parent fa9e6b1
commit 3dee3e0
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/selfdrive/modeld/runners/tinygradmodel.py b/selfdrive/modeld/runners/tinygradmodel.py
@@ -1,6 +1,7 @@
 import os
 import sys
 from pathlib import Path
+
 OPENPILOT_ROOT = Path(__file__).parent.parent.parent.parent
 if not (OPENPILOT_ROOT / 'tinygrad_repo').exists():
   OPENPILOT_ROOT = OPENPILOT_ROOT.parent  # This takes us from openpilot/openpilot to openpilot but it's probably wrong
@@ -9,7 +10,12 @@
 assert (TINYGRAD_REPO_PATH / 'extra').exists()
 sys.path.append(str(TINYGRAD_REPO_PATH))
 
-os.environ['GPU'] = '1'
+if os.getenv("OPT", None) is None:
+  os.environ['OPT'] = '99'
+if os.getenv("GPU", None) is None:
+  os.environ['GPU'] = '1'
+if os.getenv("IMAGE", None) is None:
+  os.environ['IMAGE'] = '2'
 
 import onnx
 import numpy as np