@@ -453,12 +453,8 @@ jobs:
453453 no_output_timeout : " 1h"
454454 command : |
455455 set -e
456- # TODO: Remove this after we figure out why rocm tests are failing
457- if [[ "${DOCKER_IMAGE}" == *rocm3.5* ]]; then
458- export DOCKER_TAG="ab1632df-fa59-40e6-8c23-98e004f61148"
459- fi
460- if [[ "${DOCKER_IMAGE}" == *rocm3.7* ]]; then
461- export DOCKER_TAG="1045c7b891104cb4fd23399eab413b6213e48aeb"
456+ if [[ "${DOCKER_IMAGE}" == *rocm3.9* ]]; then
457+ export DOCKER_TAG="f3d89a32912f62815e4feaeed47e564e887dffd6"
462458 fi
463459 if [[ ${BUILD_ENVIRONMENT} == *"pure_torch"* ]]; then
464460 echo 'BUILD_CAFFE2=OFF' >> "${BASH_ENV}"
@@ -538,12 +534,8 @@ jobs:
538534 command : |
539535 set -e
540536 export PYTHONUNBUFFERED=1
541- # TODO: Remove this after we figure out why rocm tests are failing
542- if [[ "${DOCKER_IMAGE}" == *rocm3.5* ]]; then
543- export DOCKER_TAG="ab1632df-fa59-40e6-8c23-98e004f61148"
544- fi
545- if [[ "${DOCKER_IMAGE}" == *rocm3.7* ]]; then
546- export DOCKER_TAG="1045c7b891104cb4fd23399eab413b6213e48aeb"
537+ if [[ "${DOCKER_IMAGE}" == *rocm3.9* ]]; then
538+ export DOCKER_TAG="f3d89a32912f62815e4feaeed47e564e887dffd6"
547539 fi
548540 # See Note [Special build images]
549541 output_image=${DOCKER_IMAGE}:${DOCKER_TAG}-${CIRCLE_SHA1}
@@ -7280,9 +7272,6 @@ workflows:
72807272 - docker_build_job :
72817273 name : " docker-pytorch-linux-xenial-py3.6-gcc7"
72827274 image_name : " pytorch-linux-xenial-py3.6-gcc7"
7283- - docker_build_job :
7284- name : " docker-pytorch-linux-bionic-rocm3.7-py3.6"
7285- image_name : " pytorch-linux-bionic-rocm3.7-py3.6"
72867275 - docker_build_job :
72877276 name : " docker-pytorch-linux-bionic-rocm3.8-py3.6"
72887277 image_name : " pytorch-linux-bionic-rocm3.8-py3.6"
@@ -7713,11 +7702,11 @@ workflows:
77137702 docker_image : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-py3.8-gcc9"
77147703 resource_class : large
77157704 - pytorch_linux_build :
7716- name : pytorch_linux_bionic_rocm3_7_py3_6_build
7705+ name : pytorch_linux_bionic_rocm3_9_py3_6_build
77177706 requires :
7718- - " docker-pytorch-linux-bionic-rocm3.7 -py3.6"
7719- build_environment : " pytorch-linux-bionic-rocm3.7 -py3.6-build"
7720- docker_image : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-rocm3.7 -py3.6"
7707+ - " docker-pytorch-linux-bionic-rocm3.9 -py3.6"
7708+ build_environment : " pytorch-linux-bionic-rocm3.9 -py3.6-build"
7709+ docker_image : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-rocm3.9 -py3.6"
77217710 resource_class : xlarge
77227711 - pytorch_macos_10_13_py3_build :
77237712 name : pytorch_macos_10_13_py3_build
0 commit comments