Skip to content

Commit

Permalink
add comment in test
Browse files Browse the repository at this point in the history
  • Loading branch information
abraham-arun committed Sep 28, 2021
1 parent 308b484 commit be9dc43
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/python/frontend/pytorch/qnn_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -296,6 +296,7 @@ def test_quantized_modules():
raw_module.eval()
inp = torch.rand(ishape)

# quantized conv_transpose2d is supported only with qnnpack engine before torch v1.8.0.
if module_name == "conv_transpose" and not is_version_greater_than("1.7.1"):
prev_engine = torch.backends.quantized.engine
torch.backends.quantized.engine = "qnnpack"
Expand Down

0 comments on commit be9dc43

Please sign in to comment.