Skip to content

GitLab

Commit 0b58bdea authored by huchen's avatar huchen
Browse files

support dtk2210

parent c4dd1fd4
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
compile.sh
View file @ 0b58bdea
source /opt/dtk-22.04.2/env.sh source /opt/dtk/env.sh
export CPLUS_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}:/opencl/include export CPLUS_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}:/opencl/include
export CPLUS_INCLUDE_PATH=${ROCM_PATH}/hiprand/include/:${ROCM_PATH}/rocrand/include:$CPLUS_INCLUDE_PATH export CPLUS_INCLUDE_PATH=${ROCM_PATH}/hiprand/include/:${ROCM_PATH}/rocrand/include:$CPLUS_INCLUDE_PATH
export LD_LIBRARY_PATH=${ROCM_PATH}/hipblas/lib/:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=${ROCM_PATH}/hipblas/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=${ROCM_PATH}/llvm/lib:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=${ROCM_PATH}/llvm/lib:$LD_LIBRARY_PATH
NAME="colossalai-0.0.2-cp37-cp37m-linux_x86_64"
rm dist/${NAME}.whl
MAX_JOBS=32 python3 setup.py -v bdist_wheel MAX_JOBS=32 python3 setup.py -v bdist_wheel
pip3 install dist/${NAME}.whl -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com --no-deps --force-reinstall
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment