This repository was archived by the owner on Oct 25, 2024. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +10
-2
lines changed
intel_extension_for_transformers/qbits Expand file tree Collapse file tree 3 files changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -4,6 +4,7 @@ source /intel-extension-for-transformers/.github/workflows/script/change_color.s
44cd /intel-extension-for-transformers
55export CMAKE_ARGS=" -DNE_DNNL_CACHE_DIR=/cache"
66pip install -U pip
7+ pip install -r requirements.txt
78$BOLD_YELLOW && echo " ---------------- git submodule update --init --recursive -------------" && $RESET
89git config --global --add safe.directory " *"
910git submodule update --init --recursive
Original file line number Diff line number Diff line change 1616# limitations under the License.
1717
1818import torch
19- if not torch .xpu ._is_compiled ():
20- from intel_extension_for_transformers .qbits_py import * # pylint: disable=E0401, E0611
19+ import intel_extension_for_transformers
20+ if "gpu" not in intel_extension_for_transformers .__version__ :
21+ from intel_extension_for_transformers .qbits_py import * # pylint: disable=E0401, E0611
Original file line number Diff line number Diff line change 88from pathlib import Path
99from setuptools import Extension , find_packages , setup
1010from setuptools .command .build_ext import build_ext
11+ from setuptools_scm import get_version
1112
1213result = subprocess .Popen ("pip install -r requirements.txt" , shell = True )
1314result .wait ()
1415
16+
1517def is_intel_gpu_available ():
1618 import torch
1719 import intel_extension_for_pytorch as ipex
@@ -286,6 +288,9 @@ def check_submodules():
286288 "intel_extension_for_transformers/transformers/runtime/" ),
287289 ])
288290 cmdclass = {'build_ext' : CMakeBuild }
291+ itrex_version = get_version ()
292+ if IS_INTEL_GPU :
293+ itrex_version = itrex_version + "-gpu"
289294
290295 setup (
291296 name = "intel-extension-for-transformers" ,
@@ -324,4 +329,5 @@ def check_submodules():
324329 ],
325330 setup_requires = ['setuptools_scm' ],
326331 use_scm_version = True ,
332+ version = itrex_version
327333 )
You can’t perform that action at this time.
0 commit comments