Merge pull request open-mpi#12535 from hppritcha/fixes_for_coll_ucc

COLL/UCC: fix for reduce_scatter_block issue
mdosanjh · May 9, 2024 · 8ecda52 · 8ecda52
2 parents e44cd58 + b6a3ca5
commit 8ecda52
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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);