From b08cdc5178d96002570b8d5af0d1d6b1aea9d0ed Mon Sep 17 00:00:00 2001
From: Francesco Fiorentino <francesco_fiorentino@apple.com>
Date: Mon, 8 May 2023 12:36:50 +0100
Subject: [PATCH] Added support for the AWS Trainium instance types based on
 the NeuronCore v2 chip architecture

---
 .../cloudprovider/aws/ec2_instance_types.go   | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/cluster-autoscaler/cloudprovider/aws/ec2_instance_types.go b/cluster-autoscaler/cloudprovider/aws/ec2_instance_types.go
index ef8ef42e08c9..393c3740cd4b 100644
--- 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,