github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 23:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: missing return in METAL dm benchmark (#6749)

Browse Source
This commit is contained in:
qazal
2024-09-26 09:12:38 +08:00
committed by GitHub
parent 8a15ccb414
commit 12e4a4900a
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/external/external_model_benchmark.py vendored
View File

@@ -74,7 +74,9 @@ def benchmark_model(m, devices, validate_outs=False):
del inputs, tinygrad_model, tinygrad_jitted_model
except RuntimeError as e:
# TODO: we don't run the dm model on METAL for now
if Device.DEFAULT == "METAL": assert "buffer count limit" in str(e)
if Device.DEFAULT == "METAL":
assert "buffer count limit" in str(e)
return
else: raise e
# convert model to torch