Skip to content

Commit a66fcc6

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 a6f3278 commit a66fcc6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

.ci/scripts/run_tests_ucc_nvls.sh

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,9 @@ export OMPI_MCA_coll=^hcoll
77
export OMPI_MCA_coll_ucc_enable=0
88
export UCC_TLS=cuda,ucp
99
export UCC_LOG_LEVEL=info
10-
export UCC_TL_CUDA_NVLS_SM_COUNT=20
11-
export UCC_TL_CUDA_TUNE=allreduce:cuda:@0
10+
export UCC_TL_CUDA_NVLS_SM_COUNT=4
11+
export UCC_TL_CUDA_TUNE=allreduce:cuda:@0,reduce_scatter:cuda:@3
1212

1313
/opt/nvidia/bin/ucc/build/bin/ucc_perftest -c allreduce -F -m cuda -b 1k -e 32M -d bfloat16 -o sum
14+
/opt/nvidia/bin/ucc/build/bin/ucc_test_mpi -c allreduce --ops sum --dtypes int32,uint32,int64,uint64 --mtypes cuda -m 1024:32M
15+
/opt/nvidia/bin/ucc/build/bin/ucc_test_mpi -c reduce_scatter --ops sum --dtypes int32,uint32,int64,uint64 --mtypes cuda -m 1024:32M

0 commit comments

Comments
 (0)