chore: add A10 GPUs as Nvidia GPUs (#5050)

Signed-off-by: Pranav Pandit <pranavp@microsoft.com>
Azure · Jan 23, 2023 · 234b41a · 234b41a
1 parent 904b3ff
commit 234b41a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/api/common/helper.go b/pkg/api/common/helper.go
@@ -81,6 +81,13 @@ func IsNvidiaEnabledSKU(vmSize string) bool {
 	   that we have an agreement with NVIDIA for this specific gpu.
 	*/
 	dm := map[string]bool{
+		// A10
+		"Standard_NV6ads_A10_v5":   true,
+		"Standard_NV12ads_A10_v5":  true,
+		"Standard_NV18ads_A10_v5":  true,
+		"Standard_NV36ads_A10_v5":  true,
+		"Standard_NV36adms_A10_v5": true,
+		"Standard_NV72ads_A10_v5":  true,
 		// K80
 		"Standard_NC6":   true,
 		"Standard_NC12":  true,