Added support for in2 instance types

Added support for the AWS Inferentia 2 instance types based on the NeuronCore v2 chip architecture
kubernetes · Feb 22, 2023 · 63161aa · 63161aa
1 parent c611acd
commit 63161aa
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/cluster-autoscaler/cloudprovider/aws/ec2_instance_types.go b/cluster-autoscaler/cloudprovider/aws/ec2_instance_types.go
@@ -1614,6 +1614,34 @@ var InstanceTypes = map[string]*InstanceType{
 		GPU:          0,
 		Architecture: "amd64",
 	},
+	"inf2.48xlarge": {
+		InstanceType: "inf2.48xlarge",
+		VCPU:         192,
+		MemoryMb:     786432,
+		GPU:          0,
+		Architecture: "amd64",
+	},
+	"inf2.24xlarge": {
+		InstanceType: "inf2.24xlarge",
+		VCPU:         96,
+		MemoryMb:     393216,
+		GPU:          0,
+		Architecture: "amd64",
+	},
+	"inf2.8xlarge": {
+		InstanceType: "inf2.8xlarge",
+		VCPU:         32,
+		MemoryMb:     131072,
+		GPU:          0,
+		Architecture: "amd64",
+	},
+	"inf2.xlarge": {
+		InstanceType: "inf2.xlarge",
+		VCPU:         4,
+		MemoryMb:     16384,
+		GPU:          0,
+		Architecture: "amd64",
+	},	
 	"is4gen.2xlarge": {
 		InstanceType: "is4gen.2xlarge",
 		VCPU:         8,