From 83bd6b077eba575df1dcd7f5664c660826654ffa Mon Sep 17 00:00:00 2001 From: Vinayak Gokhale Date: Wed, 11 Dec 2024 09:57:22 -0600 Subject: [PATCH] Update CI to use pytorch:latest (#679) * Change integration tests to use pytorch:latest * Update post merge CI to use pytorch latest docker --- .github/workflows/amd_perf_kernel_Integration_tests.yml | 2 +- .github/workflows/amd_perf_kernel_postmerge_tests.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/amd_perf_kernel_Integration_tests.yml b/.github/workflows/amd_perf_kernel_Integration_tests.yml index 5a120905f24e..b6a12841a646 100644 --- a/.github/workflows/amd_perf_kernel_Integration_tests.yml +++ b/.github/workflows/amd_perf_kernel_Integration_tests.yml @@ -100,7 +100,7 @@ jobs: matrix: runner: ${{fromJson(needs.Runner-Preparation-AMD.outputs.matrix-HIP)}} container: - image: rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.4 + image: rocm/pytorch:latest options: --device=/dev/kfd --device=/dev/dri --security-opt seccomp=unconfined --group-add video --user root steps: - name: Checkout diff --git a/.github/workflows/amd_perf_kernel_postmerge_tests.yml b/.github/workflows/amd_perf_kernel_postmerge_tests.yml index 8b9f70b50041..66b0bfa71eb1 100644 --- a/.github/workflows/amd_perf_kernel_postmerge_tests.yml +++ b/.github/workflows/amd_perf_kernel_postmerge_tests.yml @@ -41,7 +41,7 @@ jobs: matrix: runner: ${{fromJson(needs.Runner-Preparation-AMD.outputs.matrix-HIP)}} container: - image: rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2 + image: rocm/pytorch:latest options: --device=/dev/kfd --device=/dev/dri --security-opt seccomp=unconfined --group-add video --user root steps: - name: Checkout