Install Thrust package and stop patching (#16911)

Summary: Pull Request resolved: pytorch/pytorch#16911 I think the Thrust package has want we want for /opt/rocm/include/thrust. We probably can stop patching it now. Reviewed By: bddppq Differential Revision: D14015177 fbshipit-source-id: 8d9128783a790c39083a1b8b4771c2c18bd67d46
TonyTangYu · Feb 11, 2019 · 48c5d0a · 48c5d0a
1 parent 8042edc
commit 48c5d0a
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/docker/caffe2/jenkins/common/install_rocm.sh b/docker/caffe2/jenkins/common/install_rocm.sh
@@ -76,13 +76,8 @@ install_centos() {
                    rocsparse \
                    hipsparse \
                    rocrand \
-                   rccl
-
-  # Needed for now, will be replaced once hip-thrust is packaged for CentOS
-  git clone --recursive https://github.com/ROCmSoftwarePlatform/Thrust.git /data/Thrust
-  rm -rf /data/Thrust/thrust/system/cuda/detail/cub-hip
-  git clone --recursive https://github.com/ROCmSoftwarePlatform/cub-hip.git /data/Thrust/thrust/system/cuda/detail/cub-hip
-  ln -s /data/Thrust/thrust /opt/rocm/include/thrust
+                   rccl \
+                   Thrust
 }
 
 # Install Python packages depending on the base OS