diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index a60e88899..16e99ad27 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -51,12 +51,11 @@ jobs:
       matrix:
         runner:
           - labels: [self-hosted, Linux, X64, cpu]
-            container: "ubuntu"
-            # 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"
+            container: "ubuntu:22.04"
           - labels: [self-hosted, Linux, X64, rocm-7600-8gb]
-            container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2"
+            container: "rocm/pytorch:rocm6.1.3_ubuntu22.04_py3.10_pytorch_release-2.1.2"
+          - labels: [self-hosted, Linux, X64, rocm-7600-8gb]
+            container: "rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.4"
           - labels: [self-hosted, Linux, X64, cuda-3060-12gb]
             container: "nvcr.io/nvidia/pytorch:24.03-py3"
     steps: