test torch 2.4

databricks · Aug 19, 2024 · 3a8cea0 · 3a8cea0
1 parent ede0471
commit 3a8cea0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -29,6 +29,14 @@ jobs:
           gpu_num: 2
           python_version: 3.11
           container: mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu20.04
+        - name: "python3.11-pytorch2.4.0-gpus1"
+          gpu_num: 1
+          python_version: 3.11
+          container: mosaicml/pytorch:2.4.0_cu121-python3.11-ubuntu20.04
+        - name: "python3.11-pytorch2.4.0-gpus2"
+          gpu_num: 2
+          python_version: 3.11
+          container: mosaicml/pytorch:2.4.0_cu121-python3.11-ubuntu20.04
     steps:
     - name: Run PR GPU tests
       uses: mosaicml/ci-testing/.github/actions/[email protected]