Skip to content

Commit 320f510

Browse files
amaslennamaslenn
authored
Merge pull request #776 from NVIDIA/am/container-versions
Upgrade container versions for common examples
2 parents 14e317b + f72a79a commit 320f510

File tree

10 files changed

+20
-20
lines changed

10 files changed

+20
-20
lines changed

‎conf/common/test/dse_nccl_all_gather.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,7 +19,7 @@ description = "all_gather"
1919
test_template_name = "NcclTest"
2020

2121
[cmd_args]
22-
"docker_image_url" = "nvcr.io#nvidia/pytorch:25.06-py3"
22+
"docker_image_url" = "nvcr.io#nvidia/pytorch:25.12-py3"
2323
"subtest_name" = "all_gather_perf_mpi"
2424
"ngpus" = "1"
2525
"minbytes" = "128"

‎conf/common/test/nccl_test.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,7 +19,7 @@ description = "NCCL base test configuration"
1919
test_template_name = "NcclTest"
2020

2121
[cmd_args]
22-
docker_image_url = "nvcr.io#nvidia/pytorch:25.06-py3"
22+
docker_image_url = "nvcr.io#nvidia/pytorch:25.12-py3"
2323
ngpus = 1
2424
minbytes = "128"
2525
maxbytes = "4G"

‎conf/common/test/nccl_test_all_gather.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,7 +19,7 @@ description = "all_gather"
1919
test_template_name = "NcclTest"
2020

2121
[cmd_args]
22-
"docker_image_url" = "nvcr.io#nvidia/pytorch:25.06-py3"
22+
"docker_image_url" = "nvcr.io#nvidia/pytorch:25.12-py3"
2323
"subtest_name" = "all_gather_perf_mpi"
2424
"ngpus" = "1"
2525
"minbytes" = "128"

‎conf/common/test/nemo_run_llama3_8b.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,7 +19,7 @@ description = "nemo_run_llama3_8b"
1919
test_template_name = "NeMoRun"
2020

2121
[cmd_args]
22-
docker_image_url = "nvcr.io#nvidia/nemo:25.09.00"
22+
docker_image_url = "nvcr.io#nvidia/nemo:25.11.01"
2323
task = "pretrain"
2424
recipe_name = "cloudai_llama3_8b_recipe"
2525

‎conf/common/test/osu_test.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,7 +19,7 @@ test_template_name = "OSUBench"
1919
description = "OSU Benchmark example"
2020

2121
[cmd_args]
22-
"docker_image_url" = "nvcr.io#nvidia/pytorch:25.06-py3"
22+
"docker_image_url" = "nvcr.io#nvidia/pytorch:25.12-py3"
2323
"benchmarks_dir" = "/opt/hpcx/ompi/tests/osu-micro-benchmarks"
2424
"benchmark" = "osu_allreduce"
2525
"iterations" = 10

‎conf/common/test/ucc_test.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,4 +19,4 @@ description = "Base config for UCC tests"
1919
test_template_name = "UCCTest"
2020

2121
[cmd_args]
22-
docker_image_url = "nvcr.io#nvidia/pytorch:25.06-py3"
22+
docker_image_url = "nvcr.io#nvidia/pytorch:25.12-py3"

‎conf/common/test_scenario/dse_nemo_run_llama3_8b.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -26,7 +26,7 @@ description = "dse_nemo_run_llama3_8b"
2626
test_template_name = "NeMoRun"
2727

2828
[Tests.cmd_args]
29-
docker_image_url = "nvcr.io#nvidia/nemo:25.09.00"
29+
docker_image_url = "nvcr.io#nvidia/nemo:25.11.01"
3030
task = "pretrain"
3131
recipe_name = "cloudai_llama3_8b_recipe"
3232
num_layers = 32

‎conf/common/test_scenario/nccl_test.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -68,7 +68,7 @@ description = "scatter_perf"
6868
test_template_name = "NcclTest"
6969

7070
[Tests.cmd_args]
71-
docker_image_url = "nvcr.io#nvidia/pytorch:25.06-py3"
71+
docker_image_url = "nvcr.io#nvidia/pytorch:25.12-py3"
7272
subtest_name = "scatter_perf_mpi"
7373
ngpus = 1
7474
minbytes = "128"

‎conf/common/test_scenario/slurm_container.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -26,5 +26,5 @@ description = "NCCL alltoall via SlurmContainer"
2626
test_template_name = "SlurmContainer"
2727

2828
[Tests.cmd_args]
29-
docker_image_url = "nvcr.io#nvidia/pytorch:25.06-py3"
29+
docker_image_url = "nvcr.io#nvidia/pytorch:25.12-py3"
3030
cmd = "alltoall_perf_mpi --nthreads 1 --ngpus 1 --minbytes 128 --maxbytes 4G --stepbytes 1M --op sum --datatype float --root 0 --iters 100 --warmup_iters 50 --agg_iters 1 --average 1 --parallel_init 0 --check 1 --blocking 0 --cudagraph 0 --stepfactor 2"

‎conf/common/test_scenario/ucc_generator_test.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -30,6 +30,6 @@ extra_container_mounts = [
3030
]
3131

3232
[Tests.cmd_args]
33-
docker_image_url = "nvcr.io/nvidia/pytorch:25.09-py3"
33+
docker_image_url = "nvcr.io/nvidia/pytorch:25.12-py3"
3434
collective = "alltoallv"
3535
gen = "file:name=/opt/hpcx/ucc/tools/perf/generator/input_matrices.txt"

0 commit comments

Comments
 (0)