Skip to content

Commit 77a2ace

Browse files
committed
TL/CUDA: Add ucc test mpi into MPI test workflow
This commit adds ucc test mpi in github workflows to test the new data type changes. Signed-off-by: Juee14Desai <jueehimalbha@nvidia.com>
1 parent d949707 commit 77a2ace

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

.ci/scripts/run_tests_ucc_nvls_mpi.sh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ EXE="/opt/nvidia/src/ucc/build/test/mpi/ucc_test_mpi"
1717
EXE+=" --set_device 2 --mtypes cuda"
1818

1919
echo "INFO: NVLS MPI tests (allreduce) ..."
20-
UCC_TL_CUDA_TUNE="allreduce:cuda:@0" $EXE -c allreduce -d float32 -o sum -m 1024:33554432
20+
UCC_TL_CUDA_TUNE="allreduce:cuda:@0" $EXE -c allreduce -d float32,int32,uint32,int64,uint64 -o sum -m 1024:33554432
2121
echo "INFO: NVLS MPI tests (allreduce) ... DONE"
2222

23-
# echo "INFO: NVLS MPI tests (reduce_scatter) ..."
24-
# UCC_TL_CUDA_TUNE="reduce_scatter:cuda:@3" $EXE -c reduce_scatter -d float32 -o sum -m 1024:33554432
25-
# echo "INFO: NVLS MPI tests (reduce_scatter) ... DONE"
23+
echo "INFO: NVLS MPI tests (reduce_scatter) ..."
24+
UCC_TL_CUDA_TUNE="reduce_scatter:cuda:@3" $EXE -c reduce_scatter -d float32,int32,uint32,int64,uint64 -o sum -m 1024:33554432
25+
echo "INFO: NVLS MPI tests (reduce_scatter) ... DONE"

0 commit comments

Comments
 (0)