diff --git a/cluster-autoscaler/cloudprovider/gce/gce_manager.go b/cluster-autoscaler/cloudprovider/gce/gce_manager.go index 21c2aecc1e..b4b29aebfd 100644 --- a/cluster-autoscaler/cloudprovider/gce/gce_manager.go +++ b/cluster-autoscaler/cloudprovider/gce/gce_manager.go @@ -96,9 +96,17 @@ var ( "https://www.googleapis.com/auth/devstorage.read_only", "https://www.googleapis.com/auth/service.management.readonly", "https://www.googleapis.com/auth/servicecontrol"} - supportedResources = map[string]bool{cloudprovider.ResourceNameCores: true, cloudprovider.ResourceNameMemory: true} + supportedResources = map[string]bool{} ) +func init() { + supportedResources[cloudprovider.ResourceNameCores] = true + supportedResources[cloudprovider.ResourceNameMemory] = true + for _, gpuType := range supportedGpuTypes { + supportedResources[gpuType] = true + } +} + type migInformation struct { config *Mig basename string diff --git a/cluster-autoscaler/cloudprovider/gce/gpu.go b/cluster-autoscaler/cloudprovider/gce/gpu.go index 0d5b52deb9..04c342a421 100644 --- a/cluster-autoscaler/cloudprovider/gce/gpu.go +++ b/cluster-autoscaler/cloudprovider/gce/gpu.go @@ -83,6 +83,12 @@ var ( 8: 96, }, } + + supportedGpuTypes = []string{ + "nvidia-tesla-k80", + "nvidia-tesla-p100", + "nvidia-tesla-v100", + } ) func validateGpuConfig(gpuType string, gpuCount int64, zone, machineType string) error {