diff --git a/ompi/mca/coll/ucc/coll_ucc_module.c b/ompi/mca/coll/ucc/coll_ucc_module.c index 35bceac3f7e..8860bb0ba2c 100644 --- a/ompi/mca/coll/ucc/coll_ucc_module.c +++ b/ompi/mca/coll/ucc/coll_ucc_module.c @@ -3,6 +3,7 @@ * Copyright (c) 2022 Amazon.com, Inc. or its affiliates. * All Rights reserved. * Copyright (c) 2022-2024 NVIDIA Corporation. All rights reserved. + * Copyright (c) 2024 Triad National Security, LLC. All rights reserved. * $COPYRIGHT$ * * Additional copyrights may follow @@ -403,8 +404,8 @@ static int mca_coll_ucc_replace_coll_handlers(mca_coll_ucc_module_t *ucc_module) UCC_INSTALL_COLL_API(comm, ucc_module, GATHER, gather); UCC_INSTALL_COLL_API(comm, ucc_module, GATHERV, gatherv); - UCC_INSTALL_COLL_API(comm, ucc_module, REDUCE_SCATTER_BLOCK, reduce_scatter_block); - UCC_INSTALL_COLL_API(comm, ucc_module, REDUCE_SCATTER, reduce_scatter); + UCC_INSTALL_COLL_API(comm, ucc_module, REDUCE_SCATTER, reduce_scatter_block); + UCC_INSTALL_COLL_API(comm, ucc_module, REDUCE_SCATTERV, reduce_scatter); UCC_INSTALL_COLL_API(comm, ucc_module, SCATTER, scatter); UCC_INSTALL_COLL_API(comm, ucc_module, SCATTERV, scatterv);