File tree Expand file tree Collapse file tree 5 files changed +19
-77
lines changed
Expand file tree Collapse file tree 5 files changed +19
-77
lines changed Original file line number Diff line number Diff line change 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 "
1919test_template_name = " UCCTest"
2020
2121[cmd_args ]
2222docker_image_url = " nvcr.io/nvidia/pytorch:24.02-py3"
23- collective = " alltoall"
Load Diff This file was deleted.
Load Diff This file was deleted.
Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -18,33 +18,46 @@ name = "ucc_test"
1818
1919[[Tests ]]
2020id = " Tests.alltoall"
21- test_name = " ucc_test_alltoall"
21+ test_name = " ucc_base_test"
22+ description = " UCC alltoall"
2223time_limit = " 00:20:00"
2324num_nodes = 2
25+ [Tests .cmd_args ]
26+ collective = " alltoall"
2427
2528[[Tests ]]
2629id = " Tests.allgather"
27- test_name = " ucc_test_allgather"
30+ test_name = " ucc_base_test"
31+ description = " UCC allgather"
2832time_limit = " 00:20:00"
2933num_nodes = 2
34+ [Tests .cmd_args ]
35+ collective = " allgather"
3036 [[Tests .dependencies ]]
3137 type = " start_post_comp"
3238 id = " Tests.alltoall"
3339
3440[[Tests ]]
3541id = " Tests.allreduce"
36- test_name = " ucc_test_allreduce"
42+ test_name = " ucc_base_test"
43+ description = " UCC allreduce"
3744time_limit = " 00:20:00"
3845num_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 ]]
4454id = " Tests.reduce_scatter"
45- test_name = " ucc_test_reduce_scatter"
55+ test_name = " ucc_base_test"
56+ description = " UCC reduce_scatter"
4657time_limit = " 00:20:00"
4758num_nodes = 2
59+ [Tests .cmd_args ]
60+ collective = " reduce_scatter"
4861 [[Tests .dependencies ]]
4962 type = " start_post_comp"
5063 id = " Tests.allreduce"
You can’t perform that action at this time.
0 commit comments