Skip to content

Commit 87b3b18

Browse files
authored
check ipex version instead of torch version (#1442)
Signed-off-by: Xin He <xin3.he@intel.com>
1 parent 03bc3b7 commit 87b3b18

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

neural_compressor/adaptor/pytorch.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,15 @@ def get_torch_version():
5353
return version
5454

5555

56+
def get_ipex_version():
57+
try:
58+
ipex_version = ipex.__version__.split("+")[0]
59+
except ValueError as e: # pragma: no cover
60+
assert False, "Got an unknown version of intel_extension_for_pytorch: {}".format(e)
61+
version = Version(ipex_version)
62+
return version
63+
64+
5665
def get_torch_white_list(approach):
5766
version = get_torch_version()
5867
import torch.quantization as tq
@@ -2603,7 +2612,7 @@ class PyTorch_IPEXAdaptor(TemplateAdaptor):
26032612

26042613
def __init__(self, framework_specific_info):
26052614
super(PyTorch_IPEXAdaptor, self).__init__(framework_specific_info)
2606-
self.version = get_torch_version()
2615+
self.version = get_ipex_version()
26072616
query_config_file = "pytorch_ipex.yaml"
26082617
self.query_handler = PyTorchQuery(
26092618
device=self.device, local_config_file=os.path.join(os.path.dirname(__file__), query_config_file)

0 commit comments

Comments
 (0)