From dab6496dbe9d959dfde89125644ed14d9044326a Mon Sep 17 00:00:00 2001 From: Kyle Chen Date: Wed, 15 Sep 2021 11:48:33 -0700 Subject: [PATCH] [ROCm] Update CI images for ROCm 4.3.1 (#64610) Summary: Signed-off-by: Kyle Chen reference: https://github.com/pytorch/pytorch/issues/58017 jithunnair-amd jeffdaily arindamroy-eng cc jeffdaily sunway513 jithunnair-amd ROCmSupport Pull Request resolved: https://github.com/pytorch/pytorch/pull/64610 Reviewed By: seemethere Differential Revision: D30964582 Pulled By: malfet fbshipit-source-id: a8335d3d32d7f1557d3cf6cb055ad0f9c49ef7aa --- .circleci/cimodel/data/simple/docker_definitions.py | 3 +-- .circleci/config.yml | 9 +++------ .circleci/docker/build.sh | 12 ++++++------ 3 files changed, 10 insertions(+), 14 deletions(-) diff --git a/.circleci/cimodel/data/simple/docker_definitions.py b/.circleci/cimodel/data/simple/docker_definitions.py index 77a11d9..fd5ac66 100644 --- a/.circleci/cimodel/data/simple/docker_definitions.py +++ b/.circleci/cimodel/data/simple/docker_definitions.py @@ -22,10 +22,9 @@ IMAGE_NAMES = [ "pytorch-linux-xenial-py3.6-gcc5.4", # this one is used in doc builds "pytorch-linux-xenial-py3.6-gcc7.2", "pytorch-linux-xenial-py3.6-gcc7", - "pytorch-linux-bionic-rocm3.9-py3.6", - "pytorch-linux-bionic-rocm4.0.1-py3.6", "pytorch-linux-bionic-rocm4.1-py3.6", "pytorch-linux-bionic-rocm4.2-py3.6", + "pytorch-linux-bionic-rocm4.3.1-py3.6", ] # This entry should be an element from the list above diff --git a/.circleci/config.yml b/.circleci/config.yml index 82f5ce1..0950b03 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -7078,17 +7078,14 @@ workflows: name: "docker-pytorch-linux-xenial-py3.6-gcc7" image_name: "pytorch-linux-xenial-py3.6-gcc7" - docker_build_job: - name: "docker-pytorch-linux-bionic-rocm3.9-py3.6" - image_name: "pytorch-linux-bionic-rocm3.9-py3.6" - - docker_build_job: - name: "docker-pytorch-linux-bionic-rocm4.0.1-py3.6" - image_name: "pytorch-linux-bionic-rocm4.0.1-py3.6" - - docker_build_job: name: "docker-pytorch-linux-bionic-rocm4.1-py3.6" image_name: "pytorch-linux-bionic-rocm4.1-py3.6" - docker_build_job: name: "docker-pytorch-linux-bionic-rocm4.2-py3.6" image_name: "pytorch-linux-bionic-rocm4.2-py3.6" + - docker_build_job: + name: "docker-pytorch-linux-bionic-rocm4.3.1-py3.6" + image_name: "pytorch-linux-bionic-rocm4.3.1-py3.6" - pytorch_linux_build: name: pytorch_linux_xenial_py3_6_gcc5_4_build requires: diff --git a/.circleci/docker/build.sh b/.circleci/docker/build.sh index 18d19ae..bebbd3f 100755 --- a/.circleci/docker/build.sh +++ b/.circleci/docker/build.sh @@ -225,29 +225,29 @@ case "$image" in VISION=yes ROCM_VERSION=3.9 ;; - pytorch-linux-bionic-rocm4.0.1-py3.6) + pytorch-linux-bionic-rocm4.1-py3.6) ANACONDA_PYTHON_VERSION=3.6 GCC_VERSION=9 PROTOBUF=yes DB=yes VISION=yes - ROCM_VERSION=4.0.1 + ROCM_VERSION=4.1 ;; - pytorch-linux-bionic-rocm4.1-py3.6) + pytorch-linux-bionic-rocm4.2-py3.6) ANACONDA_PYTHON_VERSION=3.6 GCC_VERSION=9 PROTOBUF=yes DB=yes VISION=yes - ROCM_VERSION=4.1 + ROCM_VERSION=4.2 ;; - pytorch-linux-bionic-rocm4.2-py3.6) + pytorch-linux-bionic-rocm4.3.1-py3.6) ANACONDA_PYTHON_VERSION=3.6 GCC_VERSION=9 PROTOBUF=yes DB=yes VISION=yes - ROCM_VERSION=4.2 + ROCM_VERSION=4.3.1 ;; *) # Catch-all for builds that are not hardcoded. -- 2.7.4