Re-enable ROCm tests

This commit is contained in:
Benjamin Berman 2024-07-19 23:16:39 -07:00 committed by GitHub
parent cf9fdc5feb
commit ac3a115d83
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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: