Merge pull request #5732 from ff-apple/aws_1.24_trn

Added support for the AWS Trainium instance types
kubernetes · May 8, 2023 · a087f37 · a087f37
2 parents 35a4253 + b08cdc5
commit a087f37
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/cluster-autoscaler/cloudprovider/aws/ec2_instance_types.go b/cluster-autoscaler/cloudprovider/aws/ec2_instance_types.go
@@ -3189,6 +3189,27 @@ var InstanceTypes = map[string]*InstanceType{
 		GPU:          0,
 		Architecture: "arm64",
 	},
+	"trn1.2xlarge": {
+		InstanceType: "trn1.2xlarge",
+		VCPU:         8,
+		MemoryMb:     32768,
+		GPU:          0,
+		Architecture: "amd64",
+	},
+	"trn1.32xlarge": {
+		InstanceType: "trn1.32xlarge",
+		VCPU:         128,
+		MemoryMb:     524288,
+		GPU:          0,
+		Architecture: "amd64",
+	},
+	"trn1n.32xlarge": {
+		InstanceType: "trn1n.32xlarge",
+		VCPU:         128,
+		MemoryMb:     524288,
+		GPU:          0,
+		Architecture: "amd64",
+	},
 	"u-12tb1.112xlarge": {
 		InstanceType: "u-12tb1.112xlarge",
 		VCPU:         448,