mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
remove the different rtol / atol for openpilot CUDA in benchmark (#2907)
not sure what the issue was but seems to be fixed on master
This commit is contained in:
1
test/external/external_model_benchmark.py
vendored
1
test/external/external_model_benchmark.py
vendored
@@ -99,7 +99,6 @@ def benchmark_model(m, devices, validate_outs=False):
|
||||
|
||||
if validate_outs:
|
||||
rtol, atol = 2e-3, 2e-3 # tolerance for fp16 models
|
||||
if m == "openpilot" and 'CUDA' in devices: rtol, atol = 0.1, 0.1 # TODO: why is this broken?
|
||||
inputs = {k:Tensor(inp) for k,inp in np_inputs.items()}
|
||||
tinygrad_model = get_run_onnx(onnx_model)
|
||||
tinygrad_out = tinygrad_model(inputs)
|
||||
|
||||
Reference in New Issue
Block a user