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

no torch test if no torch

Browse Source
This commit is contained in:
George Hotz
2022-08-29 15:29:19 -07:00
parent 5eba228844
commit 880707f2d2
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
openpilot/compile.py
View File

@@ -169,10 +169,13 @@ def compile(input, output_fn):
# float32 only
FLOAT16 = int(os.getenv("FLOAT16", 0))
if FLOAT16 == 0:
from test.test_onnx import run_onnx_torch
torch_out = run_onnx_torch(onnx_model, np_inputs).numpy()
print(tinygrad_out_np, torch_out)
np.testing.assert_allclose(torch_out, tinygrad_out_np, atol=1e-4, rtol=1e-2)
try:
from test.test_onnx import run_onnx_torch
torch_out = run_onnx_torch(onnx_model, np_inputs).numpy()
print(tinygrad_out_np, torch_out)
np.testing.assert_allclose(torch_out, tinygrad_out_np, atol=1e-4, rtol=1e-2)
except ModuleNotFoundError:
pass
# save local_cl_cache as thneed
import struct, json