This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/ci-docker-staging by this push:
new 4f24b6c disable testing qmv3
4f24b6c is described below
commit 4f24b6c5e6d91b444aad2ecfda3637dd3f266f9e
Author: Masahiro Masuda <[email protected]>
AuthorDate: Fri Jan 7 13:29:10 2022 +0900
disable testing qmv3
---
tests/python/frontend/pytorch/qnn_test.py | 34 +++++++++++++------------------
1 file changed, 14 insertions(+), 20 deletions(-)
diff --git a/tests/python/frontend/pytorch/qnn_test.py
b/tests/python/frontend/pytorch/qnn_test.py
index 9f145b7..15da955 100644
--- a/tests/python/frontend/pytorch/qnn_test.py
+++ b/tests/python/frontend/pytorch/qnn_test.py
@@ -378,26 +378,20 @@ def test_quantized_imagenet():
from torchvision.models.quantization import mobilenet as qmobilenet
from torchvision.models.quantization import inception as qinception
from torchvision.models.quantization import googlenet as qgooglenet
-
- qmodels = []
-
- for per_channel in [False, True]:
- qmodels += [
- ("resnet18", qresnet.resnet18(pretrained=True), per_channel),
- ("mobilenet_v2", qmobilenet.mobilenet_v2(pretrained=True),
per_channel),
- # disable inception test for now, since loading it takes ~5min on
torchvision-0.5 due to scipy bug
- # See
https://discuss.pytorch.org/t/torchvisions-inception-v3-takes-much-longer-to-load-than-other-models/68756
- # ("inception_v3", qinception.inception_v3(pretrained=True),
per_channel),
- # tracing quantized googlenet broken as of v1.6
- # ("googlenet", qgooglenet(pretrained=True), per_channel),
- ]
-
- if is_version_greater_than("1.7.1"):
- from torchvision.models.quantization import mobilenet_v3_large as
qmobilenet_v3_large
-
- qmodels.append(
- ("mobilenet_v3_large", qmobilenet_v3_large(pretrained=True,
quantize=True).eval(), True)
- )
+ from torchvision.models.quantization import mobilenet_v3_large as
qmobilenet_v3_large
+
+ per_channel = True
+ qmodels = [
+ ("resnet18", qresnet.resnet18(pretrained=True), per_channel),
+ ("mobilenet_v2", qmobilenet.mobilenet_v2(pretrained=True),
per_channel),
+ ("inception_v3", qinception.inception_v3(pretrained=True),
per_channel),
+ # tracing quantized googlenet broken as of v1.6
+ # ("googlenet", qgooglenet(pretrained=True), per_channel),
+ # As of v1.10, quantized mobilenet v3 has a weird segfault issue
+ # during make_conv_packed_param
+ # See
https://ci.tlcpack.ai/blue/organizations/jenkins/tvm/detail/ci-docker-staging/192
+ # ("mobilenet_v3_large", qmobilenet_v3_large(pretrained=True,
quantize=True).eval(), True)
+ ]
results = []
