From 85453288d7630fb0d3e675fee2e262ef5772d4bf Mon Sep 17 00:00:00 2001 From: Comma Device Date: Thu, 18 Aug 2022 08:30:12 -0700 Subject: [PATCH] run_onnx_torch --- openpilot/compile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openpilot/compile.py b/openpilot/compile.py index e8fc88a53d..77378ba627 100644 --- a/openpilot/compile.py +++ b/openpilot/compile.py @@ -22,7 +22,6 @@ import tinygrad.ops as ops from tinygrad.llops.ops_gpu import CL, CLProgram, CLBuffer from extra.utils import fetch from extra.onnx import get_run_onnx -from test.test_onnx import run_onnx_torch from tinygrad.tensor import Tensor from tinygrad.helpers import prod @@ -165,6 +164,7 @@ 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)