Add ci build

Signed-off-by: Swati Gupta <[email protected]>

Author: guptaNswati

.github/workflows/image.yaml

@@ -29,7 +29,7 @@ on:
       - release-*
 
 jobs:
-  build:
+  build-cuda-samples:
     runs-on: ubuntu-latest
     strategy:
       matrix:
@@ -78,10 +78,39 @@ jobs:
           registry: ghcr.io
           username: ${{ github.actor }}
           password: ${{ secrets.GITHUB_TOKEN }}
-      - name: Build image
+      - name: Build cuda sample image
         env:
           IMAGE_NAME: ghcr.io/${LOWERCASE_REPO_OWNER}/k8s-samples
           VERSION: ${COMMIT_SHORT_SHA}
         run: |
           echo "${VERSION}"
           SAMPLE=${{ matrix.sample }} make -f deployments/container/Makefile build-${{ matrix.dist }}
+
+  build-nvbandwidth:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v4
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+      - name: Build nvbandwidth image
+        env:
+          IMAGE_NAME: ghcr.io/nvidia/k8s-samples:nvbandwidth
+          CUDA_VERSION: 12.5.0
+          NVBANDWIDTH_VERSION: v0.7
+          PUSH_ON_BUILD: true
+        run: |
+          echo "nvbandwidth ${NVBANDWIDTH_VERSION}"
+          docker build --platform=linux/arm64 \
+            --build-arg CUDA_VERSION=${CUDA_VERSION} \
+            --build-arg NVBANDWIDTH_VERSION=${NVBANDWIDTH_VERSION} \
+            -t "${IMAGE_NAME}-${NVBANDWIDTH_VERSION}" \
+            -f multinodeNVL/nvbandwidth/Dockerfile .