We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents abb8d10 + 3335a84 commit fbe1d24Copy full SHA for fbe1d24
docker/caffe2/jenkins/common/install_rocm.sh
@@ -64,6 +64,15 @@ install_rocrand() {
64
dpkg -i /opt/rocm/debians/rocrand.deb
65
}
66
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
+
76
# Install Python packages depending on the base OS
77
if [ -f /etc/lsb-release ]; then
78
install_ubuntu
@@ -77,3 +86,4 @@ fi
86
install_hip_thrust
87
install_rocrand
79
88
install_hcsparse
89
+install_hipsparse
0 commit comments