compile.sh

source /opt/dtk/env.sh
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 LD_LIBRARY_PATH=${ROCM_PATH}/hipblas/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

pip3 install dist/${NAME}.whl  -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com --no-deps --force-reinstall