From 1d13847cf392277eed3f03ef24212f45dc6a1aa8 Mon Sep 17 00:00:00 2001
From: Eta <esyra@coreweave.com>
Date: Tue, 3 Dec 2024 11:43:47 -0600
Subject: [PATCH] ci: Build for CUDA 12.3

This commit is just to trigger a custom build,
and is not intended to be merged.
---
 .github/workflows/ubuntu-20.yml | 40 ++++-----------------------------
 .github/workflows/ubuntu-22.yml | 40 ++++-----------------------------
 2 files changed, 8 insertions(+), 72 deletions(-)

diff --git a/.github/workflows/ubuntu-20.yml b/.github/workflows/ubuntu-20.yml
index 9770cd2..f5ad0a6 100644
--- a/.github/workflows/ubuntu-20.yml
+++ b/.github/workflows/ubuntu-20.yml
@@ -7,7 +7,7 @@ on:
       - .github/workflows/build.yml
 
 jobs:
-  cu122:
+  cu123:
     uses: ./.github/workflows/build.yml
     secrets:
       ORG_BUILDKIT_CLIENT_TOKEN: ${{ secrets.ORG_BUILDKIT_CLIENT_TOKEN }}
@@ -17,40 +17,8 @@ jobs:
       folder: .
       dockerfile: Dockerfile.ubuntu20
       base-image: nvidia/cuda
-      base-tag: 12.2.2-devel-ubuntu20.04
-      cuda-version: "12.2.2"
+      base-tag: 12.3.2-devel-ubuntu20.04
+      cuda-version: "12.3.2"
       nccl-version: 2.23.4-1
-      cuda-samples-version: "12.2"
-      hpcx-distribution: "hpcx-v2.21-gcc-doca_ofed-ubuntu20.04-cuda12"
-
-  cu124:
-    uses: ./.github/workflows/build.yml
-    secrets:
-      ORG_BUILDKIT_CLIENT_TOKEN: ${{ secrets.ORG_BUILDKIT_CLIENT_TOKEN }}
-      BUILDKIT_CONSUMER_DOPPLER_PROJECT: ${{ secrets.BUILDKIT_CONSUMER_DOPPLER_PROJECT }}
-      BUILDKIT_CONSUMER_ENDPOINT: ${{ secrets.BUILDKIT_CONSUMER_ENDPOINT }}
-    with:
-      folder: .
-      dockerfile: Dockerfile.ubuntu20
-      base-image: nvidia/cuda
-      base-tag: 12.4.1-devel-ubuntu20.04
-      cuda-version: "12.4.1"
-      nccl-version: 2.23.4-1
-      cuda-samples-version: "12.4"
-      hpcx-distribution: "hpcx-v2.21-gcc-doca_ofed-ubuntu20.04-cuda12"
-
-  cu126:
-    uses: ./.github/workflows/build.yml
-    secrets:
-      ORG_BUILDKIT_CLIENT_TOKEN: ${{ secrets.ORG_BUILDKIT_CLIENT_TOKEN }}
-      BUILDKIT_CONSUMER_DOPPLER_PROJECT: ${{ secrets.BUILDKIT_CONSUMER_DOPPLER_PROJECT }}
-      BUILDKIT_CONSUMER_ENDPOINT: ${{ secrets.BUILDKIT_CONSUMER_ENDPOINT }}
-    with:
-      folder: .
-      dockerfile: Dockerfile.ubuntu20
-      base-image: nvidia/cuda
-      base-tag: 12.6.2-devel-ubuntu20.04
-      cuda-version: "12.6.2"
-      nccl-version: 2.23.4-1
-      cuda-samples-version: "12.5"
+      cuda-samples-version: "12.3"
       hpcx-distribution: "hpcx-v2.21-gcc-doca_ofed-ubuntu20.04-cuda12"
diff --git a/.github/workflows/ubuntu-22.yml b/.github/workflows/ubuntu-22.yml
index 5d04aa9..d75d226 100644
--- a/.github/workflows/ubuntu-22.yml
+++ b/.github/workflows/ubuntu-22.yml
@@ -7,7 +7,7 @@ on:
       - .github/workflows/build.yml
 
 jobs:
-  cu122:
+  cu123:
     uses: ./.github/workflows/build.yml
     secrets:
       ORG_BUILDKIT_CLIENT_TOKEN: ${{ secrets.ORG_BUILDKIT_CLIENT_TOKEN }}
@@ -17,40 +17,8 @@ jobs:
       folder: .
       dockerfile: Dockerfile.ubuntu22
       base-image: nvidia/cuda
-      base-tag: 12.2.2-devel-ubuntu22.04
-      cuda-version: "12.2.2"
+      base-tag: 12.3.2-devel-ubuntu22.04
+      cuda-version: "12.3.2"
       nccl-version: 2.23.4-1
-      cuda-samples-version: "12.2"
-      hpcx-distribution: "hpcx-v2.21-gcc-doca_ofed-ubuntu22.04-cuda12"
-
-  cu124:
-    uses: ./.github/workflows/build.yml
-    secrets:
-      ORG_BUILDKIT_CLIENT_TOKEN: ${{ secrets.ORG_BUILDKIT_CLIENT_TOKEN }}
-      BUILDKIT_CONSUMER_DOPPLER_PROJECT: ${{ secrets.BUILDKIT_CONSUMER_DOPPLER_PROJECT }}
-      BUILDKIT_CONSUMER_ENDPOINT: ${{ secrets.BUILDKIT_CONSUMER_ENDPOINT }}
-    with:
-      folder: .
-      dockerfile: Dockerfile.ubuntu22
-      base-image: nvidia/cuda
-      base-tag: 12.4.1-devel-ubuntu22.04
-      cuda-version: "12.4.1"
-      nccl-version: 2.23.4-1
-      cuda-samples-version: "12.4"
-      hpcx-distribution: "hpcx-v2.21-gcc-doca_ofed-ubuntu22.04-cuda12"
-
-  cu126:
-    uses: ./.github/workflows/build.yml
-    secrets:
-      ORG_BUILDKIT_CLIENT_TOKEN: ${{ secrets.ORG_BUILDKIT_CLIENT_TOKEN }}
-      BUILDKIT_CONSUMER_DOPPLER_PROJECT: ${{ secrets.BUILDKIT_CONSUMER_DOPPLER_PROJECT }}
-      BUILDKIT_CONSUMER_ENDPOINT: ${{ secrets.BUILDKIT_CONSUMER_ENDPOINT }}
-    with:
-      folder: .
-      dockerfile: Dockerfile.ubuntu22
-      base-image: nvidia/cuda
-      base-tag: 12.6.2-devel-ubuntu22.04
-      cuda-version: "12.6.2"
-      nccl-version: 2.23.4-1
-      cuda-samples-version: "12.5"
+      cuda-samples-version: "12.3"
       hpcx-distribution: "hpcx-v2.21-gcc-doca_ofed-ubuntu22.04-cuda12"