Skip to content

Commit ec36742

Browse files
committed
add config test
1 parent 4471552 commit ec36742

File tree

3 files changed

+10
-16
lines changed

3 files changed

+10
-16
lines changed

optimum/intel/__init__.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -60,11 +60,11 @@
6060
raise OptionalDependencyNotAvailable()
6161
except OptionalDependencyNotAvailable:
6262
_import_structure["utils.dummy_openvino_and_nncf_objects"].extend(
63-
["OVQuantizer", "OVTrainingArguments", "OVQuantizationConfig", "OVWeightQuantizationConfig"]
63+
["OVQuantizer", "OVTrainingArguments", "OVQuantizationConfig", "OVWeightQuantizationConfig", "OVDynamicQuantizationConfig"]
6464
)
6565
else:
6666
_import_structure["openvino"].extend(
67-
["OVQuantizer", "OVTrainingArguments", "OVQuantizationConfig", "OVWeightQuantizationConfig"]
67+
["OVQuantizer", "OVTrainingArguments", "OVQuantizationConfig", "OVWeightQuantizationConfig", "OVDynamicQuantizationConfig"]
6868
)
6969

7070

@@ -195,9 +195,10 @@
195195
OVQuantizer,
196196
OVTrainingArguments,
197197
OVWeightQuantizationConfig,
198+
OVDynamicQuantizationConfig,
198199
)
199200
else:
200-
from .openvino import OVQuantizationConfig, OVQuantizer, OVTrainingArguments, OVWeightQuantizationConfig
201+
from .openvino import OVQuantizationConfig, OVQuantizer, OVTrainingArguments, OVWeightQuantizationConfig, OVDynamicQuantizationConfig
201202

202203
try:
203204
if not (is_openvino_available() and is_nncf_available() and is_accelerate_available()):

optimum/intel/openvino/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
from .trainer import OVTrainer
4444

4545

46-
from .configuration import OVConfig, OVQuantizationConfig, OVWeightQuantizationConfig
46+
from .configuration import OVConfig, OVQuantizationConfig, OVWeightQuantizationConfig, OVDynamicQuantizationConfig
4747
from .modeling import (
4848
OVModelForAudioClassification,
4949
OVModelForAudioFrameClassification,

tests/openvino/test_quantization.py

+5-12
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@
6262
OVTrainer,
6363
OVQuantizationConfig,
6464
OVWeightQuantizationConfig,
65+
OVDynamicQuantizationConfig,
6566
)
6667
from optimum.intel.openvino.configuration import OVQuantizationMethod, OVQuantizationConfigBase
6768

@@ -684,12 +685,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
684685
QUANTIZATION_CONFIGS = (
685686
(None,),
686687
(OVWeightQuantizationConfig(),),
687-
(
688-
OVWeightQuantizationConfig(
689-
bits=8,
690-
sym=True,
691-
),
692-
),
688+
(OVWeightQuantizationConfig(bits=8,sym=True),),
693689
(
694690
OVWeightQuantizationConfig(
695691
dataset="wikitext",
@@ -717,6 +713,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
717713
),
718714
),
719715
(OVQuantizationConfig(ignored_scope=nncf.IgnoredScope(names=["op_name"])),),
716+
(OVDynamicQuantizationConfig(bits=8,sym=True),),
720717
)
721718

722719
QUANTIZATION_CONFIG_DICTS = (
@@ -746,11 +743,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
746743
),
747744
(dict(num_samples=100), OVWeightQuantizationConfig, "Can't determine type of OV quantization config"),
748745
(dict(abc="def"), OVWeightQuantizationConfig, "Can't determine type of OV quantization config"),
749-
(
750-
dict(bits=8, fast_bias_correction=True),
751-
OVWeightQuantizationConfig,
752-
"Can't determine type of OV quantization config",
753-
),
746+
(dict(bits=8, fast_bias_correction=True, dataset="wikitext"), OVWeightQuantizationConfig, "Can't determine type of OV quantization config"),
754747
(dict(model_type="transformer"), OVQuantizationConfig, None),
755748
(
756749
dict(
@@ -768,7 +761,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
768761
(dict(weight_only=False), OVQuantizationConfig, None),
769762
(dict(abc="def", weight_only=False), OVQuantizationConfig, None),
770763
(dict(abc="def", weight_only=True), OVWeightQuantizationConfig, None),
771-
(dict(bits=8, fast_bias_correction=True, weight_only=True), OVWeightQuantizationConfig, None),
764+
(dict(bits=8, fast_bias_correction=True, dataset="wikitext", weight_only=True), OVWeightQuantizationConfig, None),
772765
(dict(bits=8, fast_bias_correction=True, weight_only=False), OVQuantizationConfig, None),
773766
)
774767

0 commit comments

Comments
 (0)