components: Update DCGM exporter version for AzL3 to 4.6.0 (#7284)

Signed-off-by: Suraj Deshmukh <[email protected]>

Author: surajssd

.github/workflows/validate-components.yml

@@ -1,7 +1,7 @@
 name: Validate Components
 on: pull_request
 
-jobs: 
+jobs:
   cue:
     runs-on: ubuntu-latest
     environment: test
@@ -40,3 +40,16 @@ jobs:
           export PATH="$PATH:$GOPATH/bin"
           cue vet -c ./schemas/components.cue ./vhdbuilder/packer/windows/components-test.json
 
+  version-consistency:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v5
+    - uses: actions/setup-go@v6
+      with:
+        go-version: '1.24'
+    - name: Run GPU managed components version consistency test
+      working-directory: ./e2e
+      run: |
+        set -ex
+        go test -v -run ^Test_Version_Consistency_GPU_Managed_Components$ .
+

e2e/scenario_gpu_managed_experience_test.go

@@ -2,6 +2,7 @@ package e2e
 
 import (
 	"context"
+	"regexp"
 	"testing"
 	"time"
 
@@ -29,6 +30,81 @@ func getDCGMPackageNames(os string) []string {
 	return packages
 }
 
+// extractMajorMinorPatchVersion extracts the major.minor.patch version from a
+// version string
+//
+// Examples:
+//
+//	"4.6.0-1" -> "4.6.0"
+//	"4.5.2-1.azl3" -> "4.5.2"
+//	"1:4.4.1-1" -> "4.4.1" (handles epoch prefix)
+func extractMajorMinorPatchVersion(version string) string {
+	// Remove epoch prefix (e.g., "1:" in "1:4.4.1-1")
+	version = regexp.MustCompile(`^\d+:`).ReplaceAllString(version, "")
+
+	// Match major.minor.patch pattern
+	re := regexp.MustCompile(`^(\d+\.\d+\.\d+)`)
+	matches := re.FindStringSubmatch(version)
+	if len(matches) > 1 {
+		return matches[1]
+	}
+	return ""
+}
+
+type packageOSVariant struct {
+	pkgName   string
+	osName    string
+	osRelease string
+}
+
+func Test_Version_Consistency_GPU_Managed_Components(t *testing.T) {
+	allPackageVariants := [][]packageOSVariant{
+		{
+			{"nvidia-device-plugin", "ubuntu", "r2404"},
+			{"nvidia-device-plugin", "ubuntu", "r2204"},
+			{"nvidia-device-plugin", "azurelinux", "v3.0"},
+		},
+		{
+			{"datacenter-gpu-manager-4-core", "ubuntu", "r2404"},
+			{"datacenter-gpu-manager-4-core", "ubuntu", "r2204"},
+			{"datacenter-gpu-manager-4-core", "azurelinux", "v3.0"},
+		},
+		{
+			{"datacenter-gpu-manager-4-proprietary", "ubuntu", "r2404"},
+			{"datacenter-gpu-manager-4-proprietary", "ubuntu", "r2204"},
+			{"datacenter-gpu-manager-4-proprietary", "azurelinux", "v3.0"},
+		},
+		{
+			{"datacenter-gpu-manager-exporter", "ubuntu", "r2404"},
+			{"datacenter-gpu-manager-exporter", "ubuntu", "r2204"},
+			{"dcgm-exporter", "azurelinux", "v3.0"},
+		},
+	}
+
+	for _, packageGroup := range allPackageVariants {
+		expectedVersion := ""
+		for _, pkgVar := range packageGroup {
+			componentVersions := components.GetExpectedPackageVersions(pkgVar.pkgName, pkgVar.osName, pkgVar.osRelease)
+			require.Lenf(t, componentVersions, 1,
+				"Expected exactly one %s version for %s %s but got %d",
+				pkgVar.pkgName, pkgVar.osName, pkgVar.osRelease, len(componentVersions))
+
+			pkgVersion := extractMajorMinorPatchVersion(componentVersions[0])
+			require.NotEmptyf(t, pkgVersion, "Failed to extract major.minor.patch version from %s for %s %s",
+				componentVersions[0], pkgVar.osName, pkgVar.osRelease)
+
+			// For the first iteration, set the expectedVersion
+			if expectedVersion == "" {
+				expectedVersion = pkgVersion
+				continue
+			}
+			require.Equalf(t, expectedVersion, pkgVersion,
+				"Expected all %s versions to have the same major.minor.patch version, but found mismatch: %s vs %s for %s.%s",
+				pkgVar.pkgName, expectedVersion, pkgVersion, pkgVar.osName, pkgVar.osRelease)
+		}
+	}
+}
+
 func Test_Ubuntu2404_NvidiaDevicePluginRunning(t *testing.T) {
 	RunScenario(t, &Scenario{
 		Description: "Tests that NVIDIA device plugin and DCGM Exporter are running & functional on Ubuntu 24.04 GPU nodes",

parts/common/components.json

@@ -749,7 +749,7 @@
       "downloadURL": "mcr.microsoft.com/aks/aks-gpu-grid:*",
       "gpuVersion": {
         "renovateTag": "registry=https://mcr.microsoft.com, name=aks/aks-gpu-grid",
-          "latestVersion": "550.144.06-20251021155159"
+        "latestVersion": "550.144.06-20251021155159"
       }
     }
   ],
@@ -1633,7 +1633,7 @@
             "versionsV2": [
               {
                 "renovateTag": "RPM_registry=https://packages.microsoft.com/azurelinux/3.0/prod/cloud-native/x86_64/repodata, name=dcgm-exporter, os=azurelinux, release=3.0",
-                "latestVersion": "4.5.2-1.azl3"
+                "latestVersion": "4.6.0-1.azl3"
               }
             ]
           }