From aaccdfd2dc80554d3b67474d027923ccde39cb50 Mon Sep 17 00:00:00 2001 From: Evan Lezar Date: Thu, 28 Nov 2024 14:11:13 +0100 Subject: [PATCH] Add ada-lovelace architecture label for compute capability 8.9 Signed-off-by: Evan Lezar --- internal/lm/resource.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/internal/lm/resource.go b/internal/lm/resource.go index 5ed73ca18..56fd7f74b 100644 --- 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" }