From 74d77a375787a40d87d052241209877472b41a05 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Tue, 23 Sep 2025 15:33:23 -0700
Subject: [PATCH] Fix container image for rocm

---
 .github/workflows/test.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index ad8b61eb1..821252441 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -64,7 +64,7 @@ jobs:
         runner:
           - labels: [self-hosted, Linux, X64, rocm-7600-8gb]
             friendly_name: "Python 3.12 ROCm 7.0 Torch 2.7.1"
-            container: "rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.7.1"
+            container: "rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.7.1"
     steps:
       - run: |
           apt-get update