Skip to content

Commit 8d4b6d6

Browse files
Enrico Usaienrico-usai
authored andcommitted
integ-tests: do not use EFA instances for head node
The big head node instance type was used to speed up NCCL tests but we don't need EFA enabled instances for it because we're not enabling it from the Launch Template. Signed-off-by: Enrico Usai <[email protected]>
1 parent 915db25 commit 8d4b6d6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integration-tests/tests/efa/test_efa.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def test_efa(
5353
# 2 instances are enough for other EFA tests.
5454
max_queue_size = 4 if instance in osu_benchmarks_instances else 2
5555
slots_per_instance = fetch_instance_slots(region, instance)
56-
head_node_instance = "c5n.18xlarge" if architecture == "x86_64" else "c6gn.16xlarge"
56+
head_node_instance = "c5.18xlarge" if architecture == "x86_64" else "c6g.16xlarge"
5757
cluster_config = pcluster_config_reader(max_queue_size=max_queue_size, head_node_instance=head_node_instance)
5858
cluster = clusters_factory(cluster_config)
5959
remote_command_executor = RemoteCommandExecutor(cluster)

0 commit comments

Comments
 (0)