Latest tinygrad (#36615)

* Latest tinygrad

* jit batch size

* bump again

* limit upcast

* latest tgf

* upstream tg
pull/36606/merge
Harald Schäfer 2 days ago committed by GitHub
parent 9c19ec8409
commit a1795f80dd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      .gitmodules
  2. 6
      selfdrive/modeld/SConscript
  3. 2
      tinygrad_repo

2
.gitmodules vendored

@ -15,4 +15,4 @@
url = ../../commaai/teleoprtc
[submodule "tinygrad"]
path = tinygrad_repo
url = https://github.com/commaai/tinygrad.git
url = https://github.com/tinygrad/tinygrad.git

@ -50,9 +50,9 @@ def tg_compile(flags, model_name):
# Compile small models
for model_name in ['driving_vision', 'driving_policy', 'dmonitoring_model']:
flags = {
'larch64': 'DEV=QCOM',
'Darwin': f'DEV=CPU HOME={os.path.expanduser("~")} IMAGE=0', # tinygrad calls brew which needs a $HOME in the env
}.get(arch, 'DEV=CPU CPU_LLVM=1 IMAGE=0')
'larch64': 'DEV=QCOM FLOAT16=1 NOLOCALS=1 IMAGE=2 JIT_BATCH_SIZE=0',
'Darwin': f'DEV=CPU HOME={os.path.expanduser("~")}', # tinygrad calls brew which needs a $HOME in the env
}.get(arch, 'DEV=CPU CPU_LLVM=1')
tg_compile(flags, model_name)
# Compile BIG model if USB GPU is available

@ -1 +1 @@
Subproject commit 7296c74cbd2666da7dce95d7ca6dab5340653a5c
Subproject commit 547304c471b26ada0b34f400ccba67f3e1eb5965
Loading…
Cancel
Save