OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Upgrade CI to ROCm5.0

Browse Source 
cc @jaglinux @pruthvistony @arindamroy-eng @kyleczh

Depends on https://github.com/pytorch/pytorch/pull/73906

Fixes https://github.com/pytorch/pytorch/issues/65884 as well
Pull Request resolved: https://github.com/pytorch/pytorch/pull/73105
Approved by: https://github.com/osalpekar, https://github.com/malfet
This commit is contained in:
Jithun Nair 2022-03-30 02:58:23 +00:00 committed by PyTorch MergeBot
parent 3269729c68
commit dfe6e88adc
5 changed files with 23 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
.circleci/docker/build.sh
View File

@ -222,14 +222,6 @@ case "$image" in
DB=yes
VISION=yes
;;
pytorch-linux-bionic-rocm4.3.1-py3.7)
ANACONDA_PYTHON_VERSION=3.7
GCC_VERSION=9
PROTOBUF=yes
DB=yes
VISION=yes
ROCM_VERSION=4.3.1
;;
pytorch-linux-bionic-rocm4.5-py3.7)
ANACONDA_PYTHON_VERSION=3.7
GCC_VERSION=9
@ -238,6 +230,14 @@ case "$image" in
VISION=yes
ROCM_VERSION=4.5.2
;;
pytorch-linux-bionic-rocm5.0-py3.7)
ANACONDA_PYTHON_VERSION=3.7
GCC_VERSION=9
PROTOBUF=yes
DB=yes
VISION=yes
ROCM_VERSION=5.0
;;
*)
# Catch-all for builds that are not hardcoded.
PROTOBUF=yes

2
.circleci/docker/common/install_rocm.sh
View File

@ -35,7 +35,7 @@ ver() {
}
# Map ROCm version to AMDGPU version
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" )
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" )
install_ubuntu() {
apt-get update

8
.github/scripts/generate_ci_workflows.py vendored
View File

@ -667,8 +667,8 @@ LINUX_WORKFLOWS = [
),
CIWorkflow(
arch="linux",
build_environment="linux-bionic-rocm4.5-py3.7",
docker_image_base=f"{DOCKER_REGISTRY}/pytorch/pytorch-linux-bionic-rocm4.5-py3.7",
build_environment="linux-bionic-rocm5.0-py3.7",
docker_image_base=f"{DOCKER_REGISTRY}/pytorch/pytorch-linux-bionic-rocm5.0-py3.7",
test_runner_type=LINUX_ROCM_TEST_RUNNER,
num_test_shards=2,
enable_distributed_test=False,
@ -936,8 +936,8 @@ MACOS_WORKFLOWS = [
]
DOCKER_IMAGES = {
f"{DOCKER_REGISTRY}/pytorch/pytorch-linux-bionic-rocm4.3.1-py3.7", # for rocm
f"{DOCKER_REGISTRY}/pytorch/pytorch-linux-bionic-rocm4.5-py3.7", # for rocm
f"{DOCKER_REGISTRY}/pytorch/pytorch-linux-bionic-rocm4.5-py3.7", # for rocm
f"{DOCKER_REGISTRY}/pytorch/pytorch-linux-bionic-rocm5.0-py3.7", # for rocm
}
DOCKER_IMAGES.update({

2
.github/workflows/docker-builds.yml vendored
View File

@ -27,8 +27,8 @@ jobs:
- docker-image-name: pytorch-linux-bionic-cuda10.2-cudnn7-py3.9-gcc7
- docker-image-name: pytorch-linux-bionic-cuda11.5-cudnn8-py3-gcc7
- docker-image-name: pytorch-linux-bionic-py3.7-clang9
- docker-image-name: pytorch-linux-bionic-rocm4.3.1-py3.7
- docker-image-name: pytorch-linux-bionic-rocm4.5-py3.7
- docker-image-name: pytorch-linux-bionic-rocm5.0-py3.7
- docker-image-name: pytorch-linux-xenial-cuda10.2-cudnn7-py3-gcc7
- docker-image-name: pytorch-linux-xenial-cuda11.3-cudnn8-py3-gcc7
- docker-image-name: pytorch-linux-xenial-py3-clang5-android-ndk-r19c

18
.github/workflows/pull.yml vendored
View File

@ -175,20 +175,20 @@ jobs:
{ config: "distributed", shard: 1, num_shards: 1, runner: "linux.8xlarge.nvidia.gpu" },
]}
linux-bionic-rocm4_5-py3_7-build:
name: linux-bionic-rocm4.5-py3.7
linux-bionic-rocm5_0-py3_7-build:
name: linux-bionic-rocm5.0-py3.7
uses: ./.github/workflows/_linux-build.yml
with:
build-environment: linux-bionic-rocm4.5-py3.7
docker-image-name: pytorch-linux-bionic-rocm4.5-py3.7
build-environment: linux-bionic-rocm5.0-py3.7
docker-image-name: pytorch-linux-bionic-rocm5.0-py3.7
linux-bionic-rocm4_5-py3_7-test:
name: linux-bionic-rocm4.5-py3.7
linux-bionic-rocm5_0-py3_7-test:
name: linux-bionic-rocm5.0-py3.7
uses: ./.github/workflows/_rocm-test.yml
needs: linux-bionic-rocm4_5-py3_7-build
needs: linux-bionic-rocm5_0-py3_7-build
with:
build-environment: linux-bionic-rocm4.5-py3.7
docker-image: ${{ needs.linux-bionic-rocm4_5-py3_7-build.outputs.docker-image }}
build-environment: linux-bionic-rocm5.0-py3.7
docker-image: ${{ needs.linux-bionic-rocm5_0-py3_7-build.outputs.docker-image }}
test-matrix: |
{ include: [
{ config: "default", shard: 1, num_shards: 2, runner: "linux.rocm.gpu" },