Merge pull request #1237 from NVIDIA/blackwell-arch

elezar · web-flow · commit f666bc3f836a · 2025-04-22T11:29:42.000+02:00
[gpu-feature-discovery] detect blackwell architecture
diff --git a/internal/lm/resource.go b/internal/lm/resource.go
@@ -304,6 +304,9 @@ func getArchFamily(computeMajor, computeMinor int) string {
 		return "ada-lovelace"
 	case 9:
 		return "hopper"
+	// The Blackwell GPU family is bifurcated into two cuda compute capabilities 10.0 and 12.0
+	case 10, 12:
+		return "blackwell"
 	}
 	return "undefined"
 }

Original file line number	Diff line number	Diff line change
`@@ -304,6 +304,9 @@ func getArchFamily(computeMajor, computeMinor int) string {`
`304`	`304`	`return "ada-lovelace"`
`305`	`305`	`case 9:`
`306`	`306`	`return "hopper"`
	`307`	`+ // The Blackwell GPU family is bifurcated into two cuda compute capabilities 10.0 and 12.0`
	`308`	`+ case 10, 12:`
	`309`	`+ return "blackwell"`
`307`	`310`	`}`
`308`	`311`	`return "undefined"`
`309`	`312`	`}`