Merge pull request #1060 from elezar/add-ada-lovelace-label

Add ada architecture label for compute capability 8.9
NVIDIA · Dec 6, 2024 · 44b263c · 44b263c
2 parents b6b81a6 + aaccdfd
commit 44b263c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/internal/lm/resource.go b/internal/lm/resource.go
@@ -298,7 +298,10 @@ func getArchFamily(computeMajor, computeMinor int) string {
 		}
 		return "turing"
 	case 8:
-		return "ampere"
+		if computeMinor < 9 {
+			return "ampere"
+		}
+		return "ada-lovelace"
 	case 9:
 		return "hopper"
 	}