Skip to content

Commit 7cbfb56

Browse files
committed
Revert modification of nodes_are_quantized function
Signed-off-by: gcunhase <[email protected]>
1 parent bf5c8fb commit 7cbfb56

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

tests/unit/onnx/test_quantize_int8.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -33,14 +33,9 @@ def assert_nodes_are_quantized(nodes):
3333
for node in nodes:
3434
for inp_idx, inp in enumerate(node.inputs):
3535
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-
)
36+
assert node.i(inp_idx).op == "DequantizeLinear", (
37+
f"Input '{inp.name}' of node '{node.name}' is not quantized but should be!"
38+
)
4439
return True
4540

4641

0 commit comments

Comments
 (0)