diff options
| author | iotamudelta <dieterich@ogolem.org> | 2018-08-13 11:22:43 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-08-13 11:39:43 -0700 |
| commit | 75651d5b586e423d83db47220b373952e3eae8b9 (patch) | |
| tree | 69157cb842bc444b948d3387ede6e4ebd983c4ad /docker | |
| parent | cd81217f8e030ce46751eed73285d5c870bd5540 (diff) | |
| download | pytorch-75651d5b586e423d83db47220b373952e3eae8b9.tar.gz pytorch-75651d5b586e423d83db47220b373952e3eae8b9.tar.bz2 pytorch-75651d5b586e423d83db47220b373952e3eae8b9.zip | |
improve use of ROCm libraries, enable more tests, small fixes (#10406)
Summary:
* some small leftovers from the last PR review
* enable more unit test sets for CI
* replace use of hcRNG w/ rocRAND (docker image was already updated w/ newer rocRAND)
* use rocBLAS instead of hipBLAS to allow convergence w/ Caffe2
* use strided_batched gemm interface also from the batched internal interface
* re-enable Dropout.cu as we now have philox w/ rocRAND
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10406
Reviewed By: Jorghi12
Differential Revision: D9277093
Pulled By: ezyang
fbshipit-source-id: 7ef2f6fe4ead77e501ed7aea5c3743afe2466ca2
Diffstat (limited to 'docker')
| -rw-r--r-- | docker/caffe2/jenkins/common/install_rocm.sh | 15 |
1 files changed, 0 insertions, 15 deletions
diff --git a/docker/caffe2/jenkins/common/install_rocm.sh b/docker/caffe2/jenkins/common/install_rocm.sh index f76cf90f92..e93a5bc955 100644 --- a/docker/caffe2/jenkins/common/install_rocm.sh +++ b/docker/caffe2/jenkins/common/install_rocm.sh @@ -20,16 +20,9 @@ install_ubuntu() { miopen-hip \ miopengemm \ rocblas \ - hipblas \ rocm-profiler \ cxlactivitylogger - pushd $HOME - # install hcrng - curl https://s3.amazonaws.com/ossci-linux/hcrng-master-a8c6a0b-Linux.deb -o hcrng.deb - dpkg -i hcrng.deb - rm hcrng.deb - # hotfix a bug in hip's cmake files, this has been fixed in # https://github.com/ROCm-Developer-Tools/HIP/pull/516 but for # some reason it has not included in the latest rocm release @@ -57,13 +50,6 @@ install_hip_thrust() { } # This will be removed after merging an upcoming PR. -install_hcrng() { - mkdir -p /opt/rocm/debians - curl https://s3.amazonaws.com/ossci-linux/hcrng-master-a8c6a0b-Linux.deb -o /opt/rocm/debians/hcrng.deb - dpkg -i /opt/rocm/debians/hcrng.deb -} - -# This will be removed after merging an upcoming PR. install_hcsparse() { mkdir -p /opt/rocm/debians curl https://s3.amazonaws.com/ossci-linux/hcsparse-master-907a505-Linux.deb -o /opt/rocm/debians/hcsparse.deb @@ -88,6 +74,5 @@ else fi install_hip_thrust -install_hcrng install_rocrand install_hcsparse |