File tree Expand file tree Collapse file tree 2 files changed +24
-16
lines changed Expand file tree Collapse file tree 2 files changed +24
-16
lines changed Original file line number Diff line number Diff line change @@ -49,9 +49,31 @@ python third_party/torch-xpu-ops/.github/scripts/apply_torch_pr.py
49
49
git submodule sync && git submodule update --init --recursive
50
50
python -m pip install -r requirements.txt
51
51
python -m pip install mkl-static mkl-include
52
- # python -m pip install -U cmake==3.31.6
52
+ export USE_STATIC_MKL=1
53
53
export USE_ONEMKL=1
54
54
export USE_XCCL=1
55
+ export PYTORCH_EXTRA_INSTALL_REQUIREMENTS=" \
56
+ intel-cmplr-lib-rt==2025.1.1 | \
57
+ intel-cmplr-lib-ur==2025.1.1 | \
58
+ intel-cmplr-lic-rt==2025.1.1 | \
59
+ intel-sycl-rt==2025.1.1 | \
60
+ oneccl-devel==2021.15.1; platform_system == 'Linux' and platform_machine == 'x86_64' | \
61
+ oneccl==2021.15.1; platform_system == 'Linux' and platform_machine == 'x86_64' | \
62
+ impi-rt==2021.15.0; platform_system == 'Linux' and platform_machine == 'x86_64' | \
63
+ onemkl-sycl-blas==2025.1.0 | \
64
+ onemkl-sycl-dft==2025.1.0 | \
65
+ onemkl-sycl-lapack==2025.1.0 | \
66
+ onemkl-sycl-rng==2025.1.0 | \
67
+ onemkl-sycl-sparse==2025.1.0 | \
68
+ dpcpp-cpp-rt==2025.1.1 | \
69
+ intel-opencl-rt==2025.1.1 | \
70
+ mkl==2025.1.0 | \
71
+ intel-openmp==2025.1.1 | \
72
+ tbb==2022.1.0 | \
73
+ tcmlib==1.3.0 | \
74
+ umf==0.10.0 | \
75
+ intel-pti==0.12.0
76
+ "
55
77
56
78
# Build
57
79
sed -i " s/checkout --quiet \$ {TORCH_XPU_OPS_COMMIT}/log -n 1/g" caffe2/CMakeLists.txt
Original file line number Diff line number Diff line change 1
1
#! /bin/bash
2
2
3
3
source /opt/intel/oneapi/compiler/latest/env/vars.sh
4
- source /opt/intel/oneapi/umf/latest/env/vars.sh
5
4
source /opt/intel/oneapi/pti/latest/env/vars.sh
5
+ source /opt/intel/oneapi/umf/latest/env/vars.sh
6
6
source /opt/intel/oneapi/ccl/latest/env/vars.sh
7
7
source /opt/intel/oneapi/mpi/latest/env/vars.sh
8
- source /opt/intel/oneapi/mkl/latest/env/vars.sh
9
- export PYTORCH_EXTRA_INSTALL_REQUIREMENTS=" \
10
- intel-cmplr-lib-rt==2025.0.5 |\
11
- intel-cmplr-lib-ur==2025.0.5 |\
12
- intel-cmplr-lic-rt==2025.0.5 |\
13
- intel-sycl-rt==2025.0.5 |\
14
- impi-devel==2021.14.2 |\
15
- oneccl-devel==2021.14.1 |\
16
- mkl-devel==2025.0.1 |\
17
- onemkl-sycl-blas==2025.0.1 |\
18
- onemkl-sycl-dft==2025.0.1 |\
19
- onemkl-sycl-lapack==2025.0.1 |\
20
- tcmlib==1.2.0 | umf==0.9.1 | intel-pti==0.10.2 \
21
- "
You can’t perform that action at this time.
0 commit comments