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.
1 parent d3fe294 commit 3335a84Copy full SHA for 3335a84
docker/caffe2/jenkins/common/install_rocm.sh
@@ -63,6 +63,15 @@ install_rocrand() {
63
dpkg -i /opt/rocm/debians/rocrand.deb
64
}
65
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
+
75
# Install Python packages depending on the base OS
76
if [ -f /etc/lsb-release ]; then
77
install_ubuntu
@@ -76,3 +85,4 @@ fi
85
install_hip_thrust
86
install_rocrand
78
87
install_hcsparse
88
+install_hipsparse
0 commit comments