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

[LLM] IPEX import support bigdl-core-xe-21 (#9769)

Browse source 
Add support for bigdl-core-xe-21.
This commit is contained in:
Qiyuan Gong 2023-12-28 15:23:58 +08:00 committed by GitHub
parent a8baf68865
commit f0f9d45eac
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
python/llm/src/bigdl/llm/utils/ipex_importer.py
View file

@ -35,9 +35,18 @@ class IPEXImporter:
""" """
# Check if xpu version installed # Check if xpu version installed
try: try:
# Check if bigdl-core-xe is installed
distribution('bigdl-core-xe') distribution('bigdl-core-xe')
return True return True
except PackageNotFoundError: except PackageNotFoundError:
# bigdl-core-xe not found
# Check if bigdl-core-xe-21 is installed
try:
distribution('bigdl-core-xe-21')
return True
except PackageNotFoundError:
# bigdl-core-xe not found
return False
return False return False
def import_ipex(self): def import_ipex(self):