File tree 1 file changed +16
-2
lines changed
docker/caffe2/jenkins/common
1 file changed +16
-2
lines changed Original file line number Diff line number Diff line change @@ -21,8 +21,6 @@ install_ubuntu() {
21
21
miopengemm \
22
22
rocblas \
23
23
hipblas \
24
- rocrand \
25
- hcsparse \
26
24
rocm-profiler \
27
25
cxlactivitylogger
28
26
@@ -65,6 +63,20 @@ install_hcrng() {
65
63
dpkg -i /opt/rocm/debians/hcrng.deb
66
64
}
67
65
66
+ # This will be removed after merging an upcoming PR.
67
+ install_hcsparse () {
68
+ mkdir -p /opt/rocm/debians
69
+ curl https://s3.amazonaws.com/ossci-linux/hcsparse-master-907a505-Linux.deb -o /opt/rocm/debians/hcsparse.deb
70
+ dpkg -i /opt/rocm/debians/hcsparse.deb
71
+ }
72
+
73
+ # Install an updated version of rocRand that's PyTorch compatible.
74
+ install_rocrand () {
75
+ mkdir -p /opt/rocm/debians
76
+ curl https://s3.amazonaws.com/ossci-linux/rocrand-1.8.0-Linux.deb -o /opt/rocm/debians/rocrand.deb
77
+ dpkg -i /opt/rocm/debians/rocrand.deb
78
+ }
79
+
68
80
# Install Python packages depending on the base OS
69
81
if [ -f /etc/lsb-release ]; then
70
82
install_ubuntu
77
89
78
90
install_hip_thrust
79
91
install_hcrng
92
+ install_rocrand
93
+ install_hcsparse
You can’t perform that action at this time.
0 commit comments