From b2976e64ec4327ae28c3de9c088c85a9cd64c5ef Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 19 Mar 2025 23:43:59 -0400 Subject: [PATCH] add 12.8.1 cuda to the base matrix --- .github/workflows/base.yml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index e45436e93..cb61e8494 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -40,6 +40,12 @@ jobs: python_version: "3.11" pytorch: 2.6.0 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" + - cuda: "128" + cuda_version: 12.8.1 + cudnn_version: "" + python_version: "3.11" + pytorch: 2.6.0 + torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" steps: - name: Checkout uses: actions/checkout@v4