We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bf5c8fb commit 7cbfb56Copy full SHA for 7cbfb56
tests/unit/onnx/test_quantize_int8.py
@@ -33,14 +33,9 @@ def assert_nodes_are_quantized(nodes):
33
for node in nodes:
34
for inp_idx, inp in enumerate(node.inputs):
35
if isinstance(inp, gs.Variable):
36
- if should_be_quantized:
37
- assert node.i(inp_idx).op == "DequantizeLinear", (
38
- f"Input '{inp.name}' of node '{node.name}' is not quantized but should be!"
39
- )
40
- else:
41
- assert node.i(inp_idx).op != "DequantizeLinear", (
42
- f"Input '{inp.name}' of node '{node.name}' is quantized but should not be!"
43
+ assert node.i(inp_idx).op == "DequantizeLinear", (
+ f"Input '{inp.name}' of node '{node.name}' is not quantized but should be!"
+ )
44
return True
45
46
0 commit comments