From b6a3ca51d6cb349a1fbfa74aaaa6090eb9c19b4e Mon Sep 17 00:00:00 2001 From: Howard Pritchard Date: Wed, 8 May 2024 15:49:56 -0600 Subject: [PATCH] COLL/UCC: fix for reduce_scatter_block issue See #12534 Signed-off-by: Howard Pritchard --- ompi/mca/coll/ucc/coll_ucc_module.c | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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);