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

LLM: update setup to provide new install option to support ipex 2.1 & oneapi 2024 (#9647)

Browse source 
* update setup

* default to 2.0 now

* meet code review
This commit is contained in:
Ruonan Wang 2023-12-13 17:31:56 +08:00 committed by GitHub
parent 09ca540f9b
commit 9b9cd51de1
1 changed files with 24 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
python/llm/setup.py
View file

@ -272,16 +272,28 @@ def setup_package():
all_requires += CONVERT_DEP
# Linux install with -f https://developer.intel.com/ipex-whl-stable-xpu
xpu_requires = copy.deepcopy(all_requires)
xpu_requires.remove('torch')
xpu_requires += ["torch==2.0.1a0;platform_system=='Linux'",
"torch==2.0.0a0+gitc6a572f;platform_system=='Windows'",
"torchvision==0.15.2a0;platform_system=='Linux'",
"intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'",
"intel_extension_for_pytorch==2.0.110+gitba7f6c1;platform_system=='Windows'",
"bigdl-core-xe==" + VERSION + ";(platform_system=='Linux' or platform_system=='Windows')",
"bigdl-core-xe-esimd==" + VERSION + ";platform_system=='Linux'"]
xpu_20_requires = copy.deepcopy(all_requires)
xpu_20_requires.remove('torch')
# xpu_20 only works for linux now
xpu_20_requires += ["torch==2.0.1a0;platform_system=='Linux'",
"torchvision==0.15.2a0;platform_system=='Linux'",
"intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'",
"bigdl-core-xe==" + VERSION + ";platform_system=='Linux'",
"bigdl-core-xe-esimd==" + VERSION + ";platform_system=='Linux'"]
xpu_21_requires = copy.deepcopy(all_requires)
xpu_21_requires.remove('torch')
xpu_21_requires += ["torch==2.1.0a0",
"torchvision==0.16.0a0",
"intel_extension_for_pytorch==2.1.10+xpu",
"bigdl-core-xe-21==" + VERSION,
"bigdl-core-xe-esimd-21==" + VERSION + ";platform_system=='Linux'"]
# default to ipex 2.0 for linux and 2.1 for windows
xpu_requires = copy.deepcopy(xpu_20_requires)
xpu_requires.extend(["torch==2.1.0a0;platform_system=='Windows'",
"torchvision==0.16.0a0;platform_system=='Windows'",
"intel_extension_for_pytorch==2.1.10+xpu;platform_system=='Windows'",
"bigdl-core-xe-21==" + VERSION + ";platform_system=='Windows'"])
serving_requires = ['py-cpuinfo']
serving_requires += SERVING_DEP
@ -307,7 +319,9 @@ def setup_package():
]
},
extras_require={"all": all_requires,
"xpu": xpu_requires,
"xpu": xpu_requires, # default to ipex 2.0 for linux and 2.1 for windows
"xpu_20": xpu_20_requires,
"xpu_21": xpu_21_requires,
"serving": serving_requires},
classifiers=[
'License :: OSI Approved :: Apache Software License',