platform: Set RDMA protocol as default for trn1/trn1n platforms

With switching to this plugin implementation, we want to take advantage of the RDMA protocol on trn1/trn1n platforms without setting environment variables. This commit sets RDMA protocol as default for trn1/trn1n platforms. Signed-off-by: Michael Axtmann <[email protected]>
maxtmann · Aug 16, 2024 · ef67c46 · ef67c46
1 parent 8e098c1
commit ef67c46
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/platform-aws.c b/src/platform-aws.c
@@ -73,6 +73,18 @@ struct ec2_platform_data {
 		.net_flush_required = true,
 		.default_protocol = "SENDRECV",
 	},
+	{
+		.name = "trn1.32xlarge",
+		.gdr_required = true,
+		.net_flush_required = true,
+		.default_protocol = "RDMA",
+	},
+	{
+		.name = "trn1n.32xlarge",
+		.gdr_required = true,
+		.net_flush_required = true,
+		.default_protocol = "RDMA",
+	}
 };
 
 /*