Skip to content

Commit 47ab539

Browse files
committed
skip ipex ci
Signed-off-by: Kaihui-intel <kaihui.tang@intel.com>
1 parent 46eff9b commit 47ab539

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

test/3x/torch/quantization/test_smooth_quant.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
import pytest
66
import torch
7+
from packaging.version import Version
78

89
from neural_compressor.torch.quantization import SmoothQuantConfig, convert, get_default_sq_config, prepare, quantize
910
from neural_compressor.torch.utils import is_ipex_available
@@ -33,8 +34,7 @@ def forward(self, x):
3334
def run_fn(model):
3435
for i in range(10):
3536
model(example_inputs)
36-
37-
37+
@pytest.mark.skipif(not Version(torch.__version__) < Version("2.9.0"), reason="only for torch<2.9.0 [ipex]")
3838
class TestSmoothQuant:
3939
def teardown_class(self):
4040
shutil.rmtree("saved_results", ignore_errors=True)

test/3x/torch/quantization/test_static_quant.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
import pytest
66
import torch
7+
from packaging.version import Version
78

89
try:
910
import intel_extension_for_pytorch as ipex
@@ -52,7 +53,7 @@ def run_fn(model):
5253
model(torch.rand((1, 30)))
5354
model(torch.rand((1, 30)))
5455

55-
56+
@pytest.mark.skipif(not Version(torch.__version__) < Version("2.9.0"), reason="only for torch<2.9.0 [ipex]")
5657
class TestStaticQuant:
5758
def setup_class(self):
5859
self.fp32_model = build_simple_torch_model()

test/3x/torch/quantization/weight_only/test_transformers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,13 +23,15 @@
2323

2424
ipex_version = get_ipex_version()
2525

26+
2627
try:
2728
import auto_round
2829

2930
auto_round_installed = True
3031
except ImportError:
3132
auto_round_installed = False
3233

34+
@pytest.mark.skipif(not Version(torch.__version__) < Version("2.9.0"), reason="only for torch<2.9.0 [ipex]")
3335
class TestTansformersLikeAPI:
3436
def setup_class(self):
3537
self.model_name_or_path = "hf-tiny-model-private/tiny-random-GPTJForCausalLM"

0 commit comments

Comments
 (0)