Skip to content

Commit 6b89eeb

Browse files
committed
Add another sycl link
1 parent 2b3e977 commit 6b89eeb

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

scripts/install-conda.sh

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,20 @@
11
#!/bin/bash
22

3+
link_sycl() {
4+
mkdir -p $HOME/miniforge3/envs/triton/$1
5+
ln -snf /opt/intel/oneapi/compiler/2024.1/include/sycl $HOME/miniforge3/envs/triton/$1/
6+
ln -snf /opt/intel/oneapi/compiler/2024.1/include/sycl/CL $HOME/miniforge3/envs/triton/$1/
7+
}
8+
39
install_env() {
410
export PATH="$HOME/miniforge3/bin:$PATH"
511
conda create -n triton --override-channels -c conda-forge python=$python_version.*
612
conda env update -f scripts/triton.yml
713
find /opt/intel/oneapi/ \( -name '*.so' -or -name '*.so.*' \) -exec cp -n {} $HOME/miniforge3/envs/triton/lib \;
814
ln -snf /usr/include/level_zero $HOME/miniforge3/envs/triton/bin/../x86_64-conda-linux-gnu/sysroot/usr/include/level_zero
915
find /usr -name libze_\* -exec cp -n {} $HOME/miniforge3/envs/triton/lib \;
10-
mkdir -p $HOME/miniforge3/envs/triton/lib/python$python_version/site-packages/triton/backends/intel/include
11-
ln -snf /opt/intel/oneapi/compiler/2024.1/include/sycl $HOME/miniforge3/envs/triton/lib/python$python_version/site-packages/triton/backends/intel/include
12-
ln -snf /opt/intel/oneapi/compiler/2024.1/include/sycl/CL $HOME/miniforge3/envs/triton/lib/python$python_version/site-packages/triton/backends/intel/include
16+
link_sycl lib/python$python_version/site-packages/triton/backends/intel/include
17+
link_sycl x86_64-conda-linux-gnu/sysroot/usr/include
1318
}
1419

1520
print_env_info() {

0 commit comments

Comments
 (0)