Skip to content

Commit

Permalink
neuron: add inf2e.32xlarge
Browse files Browse the repository at this point in the history
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>
  • Loading branch information
maxtmann committed Feb 24, 2025
1 parent 0911a0e commit 60f5e73
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions src/platform-aws.c
Original file line number Diff line number Diff line change
Expand Up @@ -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,
},
};


Expand Down

0 comments on commit 60f5e73

Please sign in to comment.