TOFIX: elezar suggestions

Signed-off-by: Evan Lezar <[email protected]>

Author: elezar

.github/workflows/image.yaml

@@ -38,11 +38,15 @@ jobs:
         - vectorAdd
         - nbody
         - deviceQuery
+        - nvbandwidth
         exclude:
         - dist: ubi9
           sample: deviceQuery
         - dist: ubi9
           sample: nbody
+        - dist: ubi8
+          sample: nvbandwidth
+
     steps:
       - uses: actions/checkout@v4
         name: Check out code
@@ -72,30 +76,3 @@ jobs:
           echo "Building ${{ matrix.sample }} on ${{ matrix.dist }}"
           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
-        with:
-          image: tonistiigi/binfmt:master
-      - 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
-          BUILD_MULTI_ARCH_IMAGES: true
-        run: |
-          echo "nvbandwidth ${NVBANDWIDTH_VERSION}"
-          make -f multinodeNVL/nvbandwidth/Makefile build

deployments/container/Makefile

@@ -86,7 +86,7 @@ endif
 endif
 
 build-%: DIST = $(*)
-ifeq ($(SAMPLE),nbody)
+ifeq ($(SAMPLE),$(filter $(SAMPLE),nbody nvbandwidth))
 build-%: DOCKERFILE = $(CURDIR)/deployments/container/$(SAMPLE)/Dockerfile
 else
 build-%: DOCKERFILE = $(CURDIR)/deployments/container/Dockerfile.$(DOCKERFILE_SUFFIX)

multinodeNVL/nvbandwidth/Dockerfile renamed to deployments/container/nvbandwidth/Dockerfile

@@ -12,9 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-ARG CUDA_VERSION=12.5.0
-
-FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS builder
+FROM nvcr.io/nvidia/cuda:12.6.2-devel-ubuntu22.04 AS builder
 
 ENV DEBIAN_FRONTEND=noninteractive