Skip to content

Commit

Permalink
skip test for ort v1.16.0
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Sep 21, 2023
1 parent 4d6880e commit b20ffa8
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/onnxruntime/test_quantization.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,9 @@
from pathlib import Path

from onnx import load as onnx_load
from onnxruntime import __version__ as ort_version
from onnxruntime.quantization import QuantFormat, QuantizationMode, QuantType
from packaging.version import Version, parse
from parameterized import parameterized
from transformers import AutoTokenizer

Expand Down Expand Up @@ -112,9 +114,9 @@ def test_dynamic_quantization(self, model_cls, model_name, expected_quantized_ma
self.assertEqual(expected_quantized_matmuls, num_quantized_matmul)
gc.collect()

@unittest.skipIf(parse(ort_version) == Version("1.16.0"), "not supported with this onnxruntime version")
def test_dynamic_quantization_subgraphs(self):
qconfig = AutoQuantizationConfig.avx512(is_static=False, per_channel=True)
# with tempfile.TemporaryDirectory() as tmp_dir:
tmp_dir = tempfile.mkdtemp()
output_dir = Path(tmp_dir)
model = ORTModelForCausalLM.from_pretrained(
Expand Down

0 comments on commit b20ffa8

Please sign in to comment.