-
Notifications
You must be signed in to change notification settings - Fork 0
/
set_env.sh
34 lines (28 loc) · 1.18 KB
/
set_env.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# compiler
source ${HOME}/intel/oneapi/compiler/latest/env/vars.sh
#MKL
export MKL_DPCPP_ROOT=${HOME}/intel/oneapi/mkl/latest
export LD_LIBRARY_PATH=${MKL_DPCPP_ROOT}/lib:${MKL_DPCPP_ROOT}/lib64:${MKL_DPCPP_ROOT}/lib/intel64:${LD_LIBRARY_PATH}
export LIBRARY_PATH=${MKL_DPCPP_ROOT}/lib:${MKL_DPCPP_ROOT}/lib64:${MKL_DPCPP_ROOT}/lib/intel64:$LIBRARY_PATH
export CMAKE_INCLUDE_PATH=${MKL_DPCPP_ROOT}/include/
export CMAKE_LIBRARY_PATH=${MKL_DPCPP_ROOT}/lib/intel64/
#CCL
source ${HOME}/intel/oneapi/ccl/latest/env/vars.sh
#PTI for kineto profiler
source ${HOME}/intel/oneapi/pti/latest/env/vars.sh
export BUILD_SEPARATE_OPS=ON
export USE_AOT_DEVLIST='pvc'
export BUILD_WITH_CPU=OFF
export USE_MKL=OFF
export USE_XETLA=OFF
export USE_DS_KERNELS=OFF
export USE_ONEMKL=OFF
export USE_PTI=OFF
export USE_KINETO=OFF
export USE_XCCL=ON
# export SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=1
# export SYCL_PI_LEVEL_ZERO_DEVICE_SCOPE_EVENTS=0
# export ExperimentalCopyThroughLock=1
alias pyclean="python setup.py clean"
alias pyin="pip uninstall -y intel_extension_for_pytorch;python setup.py install"
alias gitme="git config --local user.name 'hanchao';git config --local user.email '[email protected]'"