From ac3a115d832298e4c3cf72f95f6d4914111fabb1 Mon Sep 17 00:00:00 2001 From: Benjamin Berman Date: Fri, 19 Jul 2024 23:16:39 -0700 Subject: [PATCH] Re-enable ROCm tests --- .github/workflows/test.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index a1186f440..a60e88899 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -55,8 +55,8 @@ jobs: # todo: re-enable these when they actually work lol # - labels: [self-hosted, Linux, X64, ipex-a380-6gb] # container: "intel/intel-optimized-pytorch:2.1.20-xpu-pip-base" -# - labels: [self-hosted, Linux, X64, rocm-7600-8gb] -# container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2" + - labels: [self-hosted, Linux, X64, rocm-7600-8gb] + container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2" - labels: [self-hosted, Linux, X64, cuda-3060-12gb] container: "nvcr.io/nvidia/pytorch:24.03-py3" steps: