Skip to content

Commit 24fc80a

Browse files
committed
Minimize UCC configs
1 parent 5e72532 commit 24fc80a

File tree

5 files changed

+19
-77
lines changed

5 files changed

+19
-77
lines changed
Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,9 @@
1414
# See the License for the specific language governing permissions and
1515
# limitations under the License.
1616

17-
name = "ucc_test_alltoall"
18-
description = "alltoall"
17+
name = "ucc_base_test"
18+
description = "Base config for UCC tests"
1919
test_template_name = "UCCTest"
2020

2121
[cmd_args]
2222
docker_image_url = "nvcr.io/nvidia/pytorch:24.02-py3"
23-
collective = "alltoall"

conf/common/test/ucc_test_allgather.toml

Lines changed: 0 additions & 23 deletions
This file was deleted.

conf/common/test/ucc_test_allreduce.toml

Lines changed: 0 additions & 24 deletions
This file was deleted.

conf/common/test/ucc_test_reduce_scatter.toml

Lines changed: 0 additions & 23 deletions
This file was deleted.

conf/common/test_scenario/ucc_test.toml

Lines changed: 17 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,33 +18,46 @@ name = "ucc_test"
1818

1919
[[Tests]]
2020
id = "Tests.alltoall"
21-
test_name = "ucc_test_alltoall"
21+
test_name = "ucc_base_test"
22+
description = "UCC alltoall"
2223
time_limit = "00:20:00"
2324
num_nodes = 2
25+
[Tests.cmd_args]
26+
collective = "alltoall"
2427

2528
[[Tests]]
2629
id = "Tests.allgather"
27-
test_name = "ucc_test_allgather"
30+
test_name = "ucc_base_test"
31+
description = "UCC allgather"
2832
time_limit = "00:20:00"
2933
num_nodes = 2
34+
[Tests.cmd_args]
35+
collective = "allgather"
3036
[[Tests.dependencies]]
3137
type = "start_post_comp"
3238
id = "Tests.alltoall"
3339

3440
[[Tests]]
3541
id = "Tests.allreduce"
36-
test_name = "ucc_test_allreduce"
42+
test_name = "ucc_base_test"
43+
description = "UCC allreduce"
3744
time_limit = "00:20:00"
3845
num_nodes = 2
46+
[Tests.cmd_args]
47+
collective = "allreduce"
48+
e = "4G"
3949
[[Tests.dependencies]]
4050
type = "start_post_comp"
4151
id = "Tests.allgather"
4252

4353
[[Tests]]
4454
id = "Tests.reduce_scatter"
45-
test_name = "ucc_test_reduce_scatter"
55+
test_name = "ucc_base_test"
56+
description = "UCC reduce_scatter"
4657
time_limit = "00:20:00"
4758
num_nodes = 2
59+
[Tests.cmd_args]
60+
collective = "reduce_scatter"
4861
[[Tests.dependencies]]
4962
type = "start_post_comp"
5063
id = "Tests.allreduce"

0 commit comments

Comments
 (0)