ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Control sys.modules ipex duplicate check with BIGDL_CHECK_DUPLICATE_IMPORT (#11453)

Browse source 
* Control sys.modules ipex duplicate check with BIGDL_CHECK_DUPLICATE_IMPORT。
This commit is contained in:
Qiyuan Gong 2024-06-27 17:21:45 +08:00 committed by GitHub
parent c6e5ad668d
commit 4e4ecd5095
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
python/llm/src/ipex_llm/utils/ipex_importer.py
View file

@ -124,9 +124,10 @@ class IPEXImporter:
""" """
if self.is_xpu_version_installed(): if self.is_xpu_version_installed():
# Check if user import ipex manually # Check if user import ipex manually
if 'ipex' in sys.modules or 'intel_extension_for_pytorch' in sys.modules: if BIGDL_CHECK_DUPLICATE_IMPORT:
log4Error.invalidInputError(False, if 'ipex' in sys.modules or 'intel_extension_for_pytorch' in sys.modules:
ipex_duplicate_import_error) log4Error.invalidInputError(False,
ipex_duplicate_import_error)
self.directly_import_ipex() self.directly_import_ipex()
self.ipex_version = ipex.__version__ self.ipex_version = ipex.__version__
# Replace builtin import to avoid duplicate ipex import # Replace builtin import to avoid duplicate ipex import