feat: add support for ada lovelace and ampere GPU architecture (#197)

Signed-off-by: Amber Xue <ambermingxin@nvidia.com>

Author: ambermingxin

docs/usage.md

@@ -92,7 +92,7 @@ Validates the local prerequisites required for installation and enrollment.
 
 **What it checks:**
 - NVIDIA GPU presence
-- supported GPU architecture (`Hopper`, `Blackwell`, `Rubin`)
+- supported GPU architecture (`Hopper`, `Blackwell`, `Rubin`, `Ampere`, `Ada Lovelace`)
 - NVIDIA driver major version (`510` or newer)
 - DCGM HostEngine reachability
 - DCGM HostEngine minimum version (`4.2.3`)

internal/precheck/precheck.go

@@ -33,7 +33,7 @@ import (
 	"github.com/NVIDIA/fleet-intelligence-agent/internal/dcgmversion"
 )
 
-var supportedArchitectures = []string{"Hopper", "Blackwell", "Rubin"}
+var supportedArchitectures = []string{"Hopper", "Blackwell", "Rubin", "Ampere", "Ada Lovelace"}
 
 const minimumDCGMVersion = "4.2.3"
 const minimumDriverMajorVersion = 510
@@ -206,11 +206,15 @@ func evaluateArchitecture(input *Input) Check {
 	}
 
 	if !slices.ContainsFunc(supportedArchitectures, func(s string) bool {
-		return strings.EqualFold(s, input.GPUInfo.Architecture)
+		return architectureMatches(s, input.GPUInfo.Architecture)
 	}) {
 		return Check{
-			Name:    "gpu-architecture",
-			Message: "Unsupported GPU architecture: " + input.GPUInfo.Architecture + "; supported architectures are Hopper, Blackwell, and Rubin",
+			Name: "gpu-architecture",
+			Message: fmt.Sprintf(
+				"Unsupported GPU architecture: %s; supported architectures are %s",
+				input.GPUInfo.Architecture,
+				formatSupportedArchitectures(supportedArchitectures),
+			),
 		}
 	}
 
@@ -221,6 +225,31 @@ func evaluateArchitecture(input *Input) Check {
 	}
 }
 
+func architectureMatches(supported, detected string) bool {
+	return normalizeArchitectureName(supported) == normalizeArchitectureName(detected)
+}
+
+func normalizeArchitectureName(v string) string {
+	normalized := strings.TrimSpace(strings.ToLower(v))
+	normalized = strings.ReplaceAll(normalized, "-", "")
+	normalized = strings.ReplaceAll(normalized, "_", "")
+	normalized = strings.ReplaceAll(normalized, " ", "")
+	return normalized
+}
+
+func formatSupportedArchitectures(architectures []string) string {
+	switch len(architectures) {
+	case 0:
+		return ""
+	case 1:
+		return architectures[0]
+	case 2:
+		return architectures[0] + " and " + architectures[1]
+	default:
+		return strings.Join(architectures[:len(architectures)-1], ", ") + ", and " + architectures[len(architectures)-1]
+	}
+}
+
 func evaluateDriver(input *Input) Check {
 	if !gpuHardwareDetected(input) {
 		return Check{

internal/precheck/precheck_test.go

@@ -58,6 +58,30 @@ func TestEvaluateArchitecture(t *testing.T) {
 			},
 			wantPassed: true,
 		},
+		{
+			name: "passes for ampere",
+			input: Input{
+				GPUInfo:          gpuInfo("Ampere"),
+				GPUDriverVersion: "575.57.08",
+			},
+			wantPassed: true,
+		},
+		{
+			name: "passes for ada lovelace",
+			input: Input{
+				GPUInfo:          gpuInfo("Ada Lovelace"),
+				GPUDriverVersion: "575.57.08",
+			},
+			wantPassed: true,
+		},
+		{
+			name: "passes for ada-lovelace",
+			input: Input{
+				GPUInfo:          gpuInfo("Ada-Lovelace"),
+				GPUDriverVersion: "575.57.08",
+			},
+			wantPassed: true,
+		},
 		{
 			name: "fails for missing gpu",
 			input: Input{
@@ -79,12 +103,12 @@ func TestEvaluateArchitecture(t *testing.T) {
 		{
 			name: "fails for unsupported architecture",
 			input: Input{
-				GPUInfo:          gpuInfo("Ampere"),
+				GPUInfo:          gpuInfo("Turing"),
 				GPUDriverVersion: "575.57.08",
 			},
 			wantPassed: false,
 			wantMessages: []string{
-				"Unsupported GPU architecture: Ampere; supported architectures are Hopper, Blackwell, and Rubin",
+				"Unsupported GPU architecture: Turing; supported architectures are Hopper, Blackwell, Rubin, Ampere, and Ada Lovelace",
 			},
 		},
 		{
@@ -118,7 +142,7 @@ func TestEvaluateArchitecture(t *testing.T) {
 func TestSupportedArchitectures(t *testing.T) {
 	t.Parallel()
 
-	require.ElementsMatch(t, []string{"Hopper", "Blackwell", "Rubin"}, SupportedArchitectures())
+	require.ElementsMatch(t, []string{"Hopper", "Blackwell", "Rubin", "Ampere", "Ada Lovelace"}, SupportedArchitectures())
 }
 
 func TestEvaluateDriverAndNVAT(t *testing.T) {