lock ROCm to v3.5.0 on pfnCI
@@ -64,16 +64,25 @@ main() {
'py37.amd' )
cd "$(dirname "${BASH_SOURCE}")"/..
apt update -qqy
- apt install python3.7-dev -qqy
+ apt install python3.7-dev python3-apt python3-pip python3-setuptools -qqy
python3.7 -m pip install cython numpy
- wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | apt-key add -
- echo 'deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main' | tee /etc/apt/sources.list.d/rocm.list
+
+ # We lock to ROCm v3.5.0 to focus on improving CuPy's support, because many
+ # symlinks (such as libhiprtc.so) are missing in newer versions...
+ wget -qO - http://repo.radeon.com/rocm/apt/3.5/rocm.gpg.key | apt-key add -
+ echo 'deb [arch=amd64] http://repo.radeon.com/rocm/apt/3.5/ xenial main' | tee /etc/apt/sources.list.d/rocm.list
+ # TODO(leofang): revisit this when supporting newer ROCm
+ #wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | apt-key add -
+ #echo 'deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main' | tee /etc/apt/sources.list.d/rocm.list
+
apt update -qqy
- apt install rocm-dev hipblas hipsparse rocrand -qqy
+ apt install rocm-dev hipblas hipsparse rocsparse rocrand rocthrust rocsolver -qqy
export HCC_AMDGPU_TARGET=gfx900
export ROCM_HOME=/opt/rocm
export CUPY_INSTALL_USE_HIP=1
- export PATH=$ROCM_HOME/bin:$PATH
+ export PATH=$PATH:$ROCM_HOME/bin:$ROCM_HOME/profiler/bin:$ROCM_HOME/opencl/bin/x86_64
+ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$ROCM_HOME/lib
+ echo $(hipconfig)
python3.7 -m pip install -v .
;;
# Docker builds.
