Skip to content

Commit

Permalink
Add ada-lovelace architecture label for compute capability 8.9
Browse files Browse the repository at this point in the history
Signed-off-by: Evan Lezar <elezar@nvidia.com>
  • Loading branch information
elezar committed Dec 6, 2024
1 parent 96cc24a commit 75e83d7
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion internal/lm/resource.go
Original file line number Diff line number Diff line change
Expand Up @@ -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"
}
Expand Down

0 comments on commit 75e83d7

Please sign in to comment.