Skip to content

Commit 3335a84

Browse files
committed
Install the pre-release versions of rocSPARSE and hipSPARSE
1 parent d3fe294 commit 3335a84

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

docker/caffe2/jenkins/common/install_rocm.sh

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,15 @@ install_rocrand() {
6363
dpkg -i /opt/rocm/debians/rocrand.deb
6464
}
6565

66+
# Install rocSPARSE/hipSPARSE that will be released soon - can co-exist w/ hcSPARSE which will be removed soon
67+
install_hipsparse() {
68+
mkdir -p /opt/rocm/debians
69+
curl https://s3.amazonaws.com/ossci-linux/rocsparse-0.1.1.0.deb -o /opt/rocm/debians/rocsparse.deb
70+
curl https://s3.amazonaws.com/ossci-linux/hipsparse-0.1.1.0.deb -o /opt/rocm/debians/hipsparse.deb
71+
dpkg -i /opt/rocm/debians/rocsparse.deb
72+
dpkg -i /opt/rocm/debians/hipsparse.deb
73+
}
74+
6675
# Install Python packages depending on the base OS
6776
if [ -f /etc/lsb-release ]; then
6877
install_ubuntu
@@ -76,3 +85,4 @@ fi
7685
install_hip_thrust
7786
install_rocrand
7887
install_hcsparse
88+
install_hipsparse

0 commit comments

Comments
 (0)