mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-14 08:57:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix race in test_quantize_onnx (#16555)

Browse Source
This commit is contained in:
Christopher Milan
2026-06-09 15:39:48 -07:00
committed by GitHub
parent 2bdc360606
commit f9d88d3c3a
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/backend/test_quantize_onnx.py
View File

@@ -1,6 +1,6 @@
# ruff: noqa: E501
import numpy as np
import unittest
import tempfile, unittest
from tinygrad import Tensor, Context, Device, dtypes, UOp
from tinygrad.uop.ops import Ops
from tinygrad.codegen.opt import Opt, OptOps
@@ -57,8 +57,9 @@ def get_quantized_model(sz):
self.cnt += 1
if self.cnt == 100: return None
return {"input": np.random.uniform(size=(sz, sz)).astype(np.float32)}
out_file = "/tmp/test_out.onnx"
quantize_static(create_gemm_model("/tmp/test_in.onnx", sz, sz, sz), out_file,
tmpdir = tempfile.mkdtemp()
out_file = f"{tmpdir}/test_out.onnx"
quantize_static(create_gemm_model(f"{tmpdir}/test_in.onnx", sz, sz, sz), out_file,
FakeDataReader(), quant_format=QuantFormat.QDQ, per_channel=False, reduce_range=False,
activation_type=QuantType.QUInt8, weight_type=QuantType.QInt8,
extra_options={"ActivationSymmetric": False})