aws: add platform config for g7e instance type

Enable OFI NCCL plugin to support AWS g7e.8xlarge, g7e.12xlarge,
g7e.24xlarge, g7e.48xlarge instance types.

Signed-off-by: Rongbing Zhou <rongbiz@amazon.com>
(cherry picked from commit 8dc2ce4)

Author: rongbingzhou

src/platform-aws.cpp

@@ -208,6 +208,34 @@ static struct ec2_platform_data platform_data_map[] = {
 		.domain_per_thread = true,
 		.env = {},
 	},
+	{
+		.name = "g7e.8xlarge",
+		.regex = NULL,
+		.topology = NULL,
+		.default_dup_conns = 0,
+		.latency = 35.0,
+		.gdr_required = false,
+		.default_protocol = PROTOCOL::RDMA,
+		.domain_per_thread = true,
+		.env = {
+			{ "NCCL_BUFFSIZE", "8388608" },
+			{ "NCCL_P2P_NET_CHUNKSIZE", "524288" },
+		},
+	},
+	{
+		.name = "g7e",
+		.regex = "^g7e\\.(12|24|48)xlarge",
+		.topology = NULL,
+		.default_dup_conns = 0,
+		.latency = 35.0,
+		.gdr_required = true,
+		.default_protocol = PROTOCOL::RDMA,
+		.domain_per_thread = true,
+		.env = {
+			{ "NCCL_BUFFSIZE", "8388608" },
+			{ "NCCL_P2P_NET_CHUNKSIZE", "524288" },
+		},
+	},
 	{
 		.name = "trn1",
 		.regex = "^trn1.*",

tests/unit/aws_platform_mapper.cpp

@@ -65,6 +65,16 @@ static int check_known_platforms(void)
 	ret += check_value(platform_data_list, len, "p6e-gb200.36xlarge", "p-series");
 	ret += check_value(platform_data_list, len, "g5.48xlarge", "g5.48xlarge");
 	ret += check_value(platform_data_list, len, "g6.16xlarge", NULL);
+	ret += check_value(platform_data_list, len, "g7e.8xlarge", "g7e.8xlarge");
+	ret += check_value(platform_data_list, len, "g7.8xlarge", NULL);
+	ret += check_value(platform_data_list, len, "g7e.12xlarge", "g7e");
+	ret += check_value(platform_data_list, len, "g7e.24xlarge", "g7e");
+	ret += check_value(platform_data_list, len, "g7e.48xlarge", "g7e");
+	ret += check_value(platform_data_list, len, "g7e.xlarge", NULL);
+	ret += check_value(platform_data_list, len, "g7e.1xlarge", NULL);
+	ret += check_value(platform_data_list, len, "g7e.2xlarge", NULL);
+	ret += check_value(platform_data_list, len, "g7e.4xlarge", NULL);
+	ret += check_value(platform_data_list, len, "g7.48xlarge", NULL);
 
 	// obviously future platforms
 	ret += check_value(platform_data_list, len, "p100.2048xlarge", "p-series");