Skip to content

Commit 44b263c

Browse files
authored
Merge pull request #1060 from elezar/add-ada-lovelace-label
Add ada architecture label for compute capability 8.9
2 parents b6b81a6 + aaccdfd commit 44b263c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

internal/lm/resource.go

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,10 @@ func getArchFamily(computeMajor, computeMinor int) string {
298298
}
299299
return "turing"
300300
case 8:
301-
return "ampere"
301+
if computeMinor < 9 {
302+
return "ampere"
303+
}
304+
return "ada-lovelace"
302305
case 9:
303306
return "hopper"
304307
}

0 commit comments

Comments
 (0)