Skip to content

Commit b51d2fe

Browse files
authored
Merge pull request #1130 from jeruane/cherry-pick-fix-nvidia-gpu-resource-name-1.1
cherry pick: "Fix nvidia gpu resource name on AWS" to 1.1
2 parents 71f0bc4 + e68d5fb commit b51d2fe

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cluster-autoscaler/cloudprovider/aws/aws_manager.go

+2-1
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ import (
3535
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
3636
"k8s.io/apimachinery/pkg/util/wait"
3737
"k8s.io/autoscaler/cluster-autoscaler/cloudprovider"
38+
"k8s.io/autoscaler/cluster-autoscaler/utils/gpu"
3839
provider_aws "k8s.io/kubernetes/pkg/cloudprovider/providers/aws"
3940
kubeletapis "k8s.io/kubernetes/pkg/kubelet/apis"
4041
)
@@ -250,7 +251,7 @@ func (m *AwsManager) buildNodeFromTemplate(asg *Asg, template *asgTemplate) (*ap
250251
// TODO: get a real value.
251252
node.Status.Capacity[apiv1.ResourcePods] = *resource.NewQuantity(110, resource.DecimalSI)
252253
node.Status.Capacity[apiv1.ResourceCPU] = *resource.NewQuantity(template.InstanceType.VCPU, resource.DecimalSI)
253-
node.Status.Capacity[apiv1.ResourceNvidiaGPU] = *resource.NewQuantity(template.InstanceType.GPU, resource.DecimalSI)
254+
node.Status.Capacity[gpu.ResourceNvidiaGPU] = *resource.NewQuantity(template.InstanceType.GPU, resource.DecimalSI)
254255
node.Status.Capacity[apiv1.ResourceMemory] = *resource.NewQuantity(template.InstanceType.MemoryMb*1024*1024, resource.DecimalSI)
255256

256257
// TODO: use proper allocatable!!

0 commit comments

Comments
 (0)