Skip to content

Commit 32ed98f

Browse files
[integ-tests] Increase NCCL benchmark baseline
The baselines are 90% of the current performance
1 parent 90efbbe commit 32ed98f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integration-tests/tests/common/nccl_common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,8 @@ def install_and_run_nccl_benchmarks(remote_command_executor, mpi_module, schedul
6969
"p4d.24xlarge": 26.0,
7070
# p5.48xlarge - Expected "in-place busbw" bandwidth with 2 nodes, 8 tasks per node is about 250GB/s
7171
"p5.48xlarge": 250.0,
72-
"p6-b200.48xlarge": 300,
73-
"p6e-gb200.36xlarge": 500,
72+
"p6-b200.48xlarge": 570, # Initial testing performance 631.17
73+
"p6e-gb200.36xlarge": 650, # Initial testing performance 719.17
7474
}
7575

7676
expected_bandwidth = instance_bandwidth_dict.get(instance)

0 commit comments

Comments
 (0)