make the GPU sku a parameter

Author: dcasati

.github/workflows/deploy.yml

@@ -9,6 +9,7 @@ env:
   REGISTRY_NAME: gbbpvt
   CLUSTER_NAME: pvt-aks-h100
   TEMPLATE_FILE: main.bicep
+  GPUSKU: Standard_NC40ads_H100_v5
 
 jobs:
   deploy:
@@ -43,4 +44,4 @@ jobs:
           subscription-id: ${{ secrets.AZURE_SUBSCRIPTION_ID }}
           template-file: ${{ env.TEMPLATE_FILE }}
           resource-group-name: ${{ env.RESOURCE_GROUP }}
-          parameters: '{"registryName":"${{ env.REGISTRY_NAME }}","clusterName":"${{ env.CLUSTER_NAME }}","resourceGroupName":"${{ env.RESOURCE_GROUP }}","location":"${{ env.LOCATION }}"}'
+          parameters: '{"registryName":"${{ env.REGISTRY_NAME }}","clusterName":"${{ env.CLUSTER_NAME }}","resourceGroupName":"${{ env.RESOURCE_GROUP }}","location":"${{ env.LOCATION }}"}, "gpuVMSKU":"${{ env.GPUSKU }}"'

aks.bicep

@@ -4,6 +4,7 @@ param vnetName string = 'vnet-${managedClusterName}'
 param subnetName string = 'subnet-${managedClusterName}'
 param location string
 param registryName string
+param gpuSKU string
 
 resource vnet 'Microsoft.Network/virtualNetworks@2023-05-01' = {
   name: vnetName
@@ -78,7 +79,7 @@ resource managedCluster 'Microsoft.ContainerService/managedClusters@2024-10-01'
       {
         name: 'gpunp'
         count: 1
-        vmSize: 'Standard_NC40ads_H100_v5'
+        vmSize: gpuSKU
         osDiskSizeGB: 322
         osDiskType: 'Ephemeral'
         kubeletDiskType: 'OS'

main.bicep

@@ -3,6 +3,7 @@ param clusterName string
 param managedClusterName string = clusterName
 param resourceGroupName string = 'rg-${managedClusterName}'
 param location string = 'eastus2'
+param gpuVMSKU string = 'Standard_NC40ads_H100_v5'
 
 module infra 'aks.bicep' = {
   name: 'provisionInfra'
@@ -11,5 +12,6 @@ module infra 'aks.bicep' = {
     clusterName: managedClusterName
     location: location
     registryName: registryName
+    gpuSKU: gpuVMSKU
   }
 }