Skip to content

Commit fbe1d24

Browse files
authored
Merge pull request #151 from iotamudelta/rocsparse_install
First step to rocSPARSE: install
2 parents abb8d10 + 3335a84 commit fbe1d24

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
@@ -64,6 +64,15 @@ install_rocrand() {
6464
dpkg -i /opt/rocm/debians/rocrand.deb
6565
}
6666

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

0 commit comments

Comments
 (0)