diff --git a/test.py b/test.py index ed8413839..2856fb64c 100644 --- a/test.py +++ b/test.py @@ -176,10 +176,6 @@ def _load_tests(): if os.getenv("USE_CANARY_MODELS"): model_paths.extend(_list_canary_model_paths()) for path in model_paths: - # TODO: skipping quantized tests for now due to BC-breaking changes for prepare - # api, enable after PyTorch 1.13 release - if "quantized" in path: - continue for device in devices: _load_test(path, device) diff --git a/torchbenchmark/util/env_check.py b/torchbenchmark/util/env_check.py index 9473f7bec..14eb9760b 100644 --- a/torchbenchmark/util/env_check.py +++ b/torchbenchmark/util/env_check.py @@ -30,6 +30,8 @@ "Super_SloMo", "vgg16", "mtml_ctr_instagram_model", + "resnet50_quantized_qat", + "mobilenet_v2_quantized_qat", ] CI_SKIP_OPTIMIZER = { # TIMM