neuron: add inf2e.32xlarge

Inf2e uses EFA but it's missing in platform data map. By adding the platform data map, domain_per_thread is enabled correctly. Signed-off-by: Se Wang Oh <seoh@amazon.com> Signed-off-by: Michael Axtmann <axtmannm@amazon.com>
aws · Feb 25, 2025 · 71a1e68 · 71a1e68
1 parent a4832e0
commit 71a1e68
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/platform-aws.c b/src/platform-aws.c
@@ -118,6 +118,17 @@ static struct ec2_platform_data platform_data_map[] = {
 		.default_protocol = "RDMA",
 		.domain_per_thread = 1,
 	},
+	{
+		.name = "inf",
+		.regex = "^inf.*",
+		.topology = NULL,
+		.default_dup_conns = 0,
+		.latency = 75.0,
+		.gdr_required = true,
+		.net_flush_required = true,
+		.default_protocol = "SENDRECV",
+		.domain_per_thread = 1,
+	},
 };
 
 

diff --git a/tests/unit/aws_platform_mapper.c b/tests/unit/aws_platform_mapper.c
@@ -49,7 +49,7 @@ static int check_known_platforms(void)
 	ret += check_value(platform_data_list, len, "trn1.2xlarge", "trn1");
 	ret += check_value(platform_data_list, len, "trn2.48xlarge", "trn2");
 	ret += check_value(platform_data_list, len, "trn2u.48xlarge", "trn2");
-	ret += check_value(platform_data_list, len, "inf2.48xlarge", NULL);
+	ret += check_value(platform_data_list, len, "inf2.48xlarge", "inf");
 	ret += check_value(platform_data_list, len, "p3.2xlarge", NULL);
 	ret += check_value(platform_data_list, len, "p3.8xlarge", NULL);
 	ret += check_value(platform_data_list, len, "p3.16xlarge", NULL);