github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 06:58:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix wrong op version being used if versions equal (#934)

Browse Source
This commit is contained in:
M4tthewDE
2023-06-05 16:45:10 +02:00
committed by GitHub
parent 79613eb83e
commit 70f12fdb57
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/onnx.py
View File

@@ -181,7 +181,7 @@ def get_run_onnx(onnx_model: ModelProto):
fxn = getattr(onnx_ops, n.op_type)
if isinstance(fxn, dict):
for k in sorted(fxn.keys()):
if k < onnx_model_version:
if k <= onnx_model_version:
real_fxn = fxn[k]
else:
real_fxn = fxn