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

[ROCm] upgrade CI distributed test to ROCm 5.0

Browse Source 
This change was missed by earlier CI upgrade #73105.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/74959
Approved by: https://github.com/suo, https://github.com/malfet
This commit is contained in:
Jeff Daily 2022-04-06 17:19:39 +00:00 committed by PyTorch MergeBot
parent b780e8a640
commit eb43e60a92
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -678,8 +678,8 @@ LINUX_WORKFLOWS = [
),
CIWorkflow(
arch="linux",
build_environment="linux-bionic-rocm4.5-py3.7-distributed",
docker_image_base=f"{DOCKER_REGISTRY}/pytorch/pytorch-linux-bionic-rocm4.5-py3.7",
build_environment="linux-bionic-rocm5.0-py3.7-distributed",
docker_image_base=f"{DOCKER_REGISTRY}/pytorch/pytorch-linux-bionic-rocm5.0-py3.7",
test_runner_type=LINUX_ROCM_TEST_RUNNER,
enable_default_test=False,
ciflow_config=CIFlowConfig(

10
.github/workflows/trunk.yml vendored
View File

@ -95,18 +95,18 @@ jobs:
docker-image-name: pytorch-linux-xenial-cuda11.3-cudnn8-py3-gcc7
linux-bionic-rocm4_5-py3_7-distributed-build:
name: linux-bionic-rocm4.5-py3.7-distributed
name: linux-bionic-rocm5.0-py3.7-distributed
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-distributed-test:
name: linux-bionic-rocm4.5-py3.7-distributed
name: linux-bionic-rocm5.0-py3.7-distributed
uses: ./.github/workflows/_rocm-test.yml
needs: linux-bionic-rocm4_5-py3_7-distributed-build
with:
build-environment: linux-bionic-rocm4.5-py3.7
build-environment: linux-bionic-rocm5.0-py3.7
docker-image: ${{ needs.linux-bionic-rocm4_5-py3_7-distributed-build.outputs.docker-image }}
test-matrix: |
{ include: [