Skip to content

Commit

Permalink
update test case
Browse files Browse the repository at this point in the history
  • Loading branch information
dsikka committed Oct 8, 2024
1 parent 6f4cae7 commit 52e9a45
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ def _test_layer_dynamic_quantization_status(
# check inputs always have an observer if quantized but never scale/zp
assert not hasattr(module, "input_scale")
assert not hasattr(module, "input_zero_point")
assert hasattr(module, "input_observer") == inputs
assert not hasattr(module, "input_observer")

# check weights always have scale/zp and observer only if not frozen
assert hasattr(module, "weight_scale") == weights
Expand Down

0 comments on commit 52e9a45

Please sign in to comment.