File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff line change 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+
39install_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
1520print_env_info () {
You can’t perform that action at this time.
0 commit comments