fixes for sharp

Author: edwardsp

.github/workflows/build_nccl_image_dev.yaml

@@ -47,9 +47,9 @@ jobs:
       with:
         images: ghcr.io/azure/ai-infrastructure-on-azure/nccl-test
         tags: |
-          type=raw,value=dev2503
-          type=raw,value=dev2503-{{sha}}
-          type=raw,value=dev2503-pytorch-{{date 'YYYYMMDD-hhmmss' tz='UTC'}}
+          type=raw,value=dev
+          type=raw,value=dev-{{sha}}
+          type=raw,value=dev-pytorch-{{date 'YYYYMMDD-hhmmss' tz='UTC'}}
 
     - name: Build and push Dev Image
       uses: docker/build-push-action@v6
@@ -67,14 +67,14 @@ jobs:
         echo "## Dev NCCL Test Image Built Successfully! 🚀" >> $GITHUB_STEP_SUMMARY
         echo "" >> $GITHUB_STEP_SUMMARY
         echo "**Image Tags:**" >> $GITHUB_STEP_SUMMARY
-        echo "- \`ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev2503\`" >> $GITHUB_STEP_SUMMARY
-        echo "- \`ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev2503-${{ github.sha }}\`" >> $GITHUB_STEP_SUMMARY
+        echo "- \`ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev\`" >> $GITHUB_STEP_SUMMARY
+        echo "- \`ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev-${{ github.sha }}\`" >> $GITHUB_STEP_SUMMARY
         echo "" >> $GITHUB_STEP_SUMMARY
         echo "**Changes:**" >> $GITHUB_STEP_SUMMARY
-        echo "- Updated to use PyTorch base image: \`nvcr.io/nvidia/pytorch:25.03-py3\`" >> $GITHUB_STEP_SUMMARY
+        echo "- Updated to use PyTorch base image: \`nvcr.io/nvidia/pytorch:25.06-py3\`" >> $GITHUB_STEP_SUMMARY
         echo "- Includes NCCL tests for distributed training validation" >> $GITHUB_STEP_SUMMARY
         echo "" >> $GITHUB_STEP_SUMMARY
         echo "**Usage:**" >> $GITHUB_STEP_SUMMARY
         echo "\`\`\`bash" >> $GITHUB_STEP_SUMMARY
-        echo "docker pull ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev2503" >> $GITHUB_STEP_SUMMARY
+        echo "docker pull ghcr.io/azure/ai-infrastructure-on-azure/nccl-test:dev" >> $GITHUB_STEP_SUMMARY
         echo "\`\`\`" >> $GITHUB_STEP_SUMMARY

infrastructure_validations/aks/NCCL/docker/Dockerfile

@@ -1,4 +1,4 @@
-ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:25.03-py3
+ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:25.06-py3
 FROM ${PYTORCH_IMAGE}
 
 ENV DEBIAN_FRONTEND=noninteractive

infrastructure_validations/aks/NCCL/helm/nccl-test/values.yaml

@@ -47,11 +47,11 @@ ncclTest:
 
     # UCX settings
     UCX_TLS: "rc"
-    UCX_NET_DEVICES: "mlx5_ib0:1"
+    UCX_NET_DEVICES: "mlx5_0:1"
 
     # SHARP settings
     NCCL_COLLNET_ENABLE: "1" # set to "0" to disable SHARP
-    SHARP_SMX_UCX_INTERFACE: "mlx5_ib0:1"
+    SHARP_SMX_UCX_INTERFACE: "mlx5_0:1"
     SHARP_COLL_ENABLE_SAT: "1"
     SHARP_COLL_LOG_LEVEL: "3"
     SHARP_COLL_ENABLE_PCI_RELAXED_ORDERING: "1"