TYPE_RESOURCE_GUIDED: add minimal support

Add minimal support for MPI_COMM_TYPE_RESOURCE_GUIDED.
It can support all of the mpi_hw_resource_type values
supported by MPI_COMM_TYPE_HW_GUIDED.

No support is provided in this PR for mpi_pset_name
info key.

Related to #12077

Signed-off-by: Howard Pritchard <[email protected]>

Author: hppritcha

docs/man-openmpi/man3/MPI_Comm_split_type.3.rst

@@ -46,6 +46,16 @@ MPI_COMM_TYPE_SHARED
    This type splits the communicator into subcommunicators, each of
    which can create a shared memory region.
 
+MPI_COMM_TYPE_HW_GUIDED
+   This type splits the communicator into subcommunicators according
+   to the resource type specified by the ``mpi_hw_resource_type``
+   info key.
+
+MPI_COMM_TYPE_RESOURCE_GUIDED
+   This type splits the communicator into subcommunicators according
+   to the resource type specified by the ``mpi_hw_resource_type``
+   or ``mpi_pset_name`` info key.
+
 OMPI_COMM_TYPE_NODE
    Synonym for MPI_COMM_TYPE_SHARED.
 

ompi/communicator/comm.c

@@ -24,7 +24,7 @@
  * Copyright (c) 2015      Mellanox Technologies. All rights reserved.
  * Copyright (c) 2017-2022 IBM Corporation.  All rights reserved.
  * Copyright (c) 2021      Nanook Consulting.  All rights reserved.
- * Copyright (c) 2018-2024 Triad National Security, LLC. All rights
+ * Copyright (c) 2018-2025 Triad National Security, LLC. All rights
  *                         reserved.
  * Copyright (c) 2023-2025 Advanced Micro Devices, Inc. All rights reserved.
  * $COPYRIGHT$
@@ -1186,7 +1186,8 @@ int ompi_comm_split_type (ompi_communicator_t *comm, int split_type, int key,
     inter = OMPI_COMM_IS_INTER(comm);
 
     /* Step 0: Convert MPI_COMM_TYPE_HW_GUIDED to the internal type */
-    if (MPI_COMM_TYPE_HW_GUIDED == split_type) {
+    if ((MPI_COMM_TYPE_HW_GUIDED == split_type) ||
+        (MPI_COMM_TYPE_RESOURCE_GUIDED == split_type)) {
         opal_info_get(info, "mpi_hw_resource_type", &value, &flag);
         /* If key is not in the 'info', then return MPI_COMM_NULL.
          * This is caught at the MPI interface level, but it doesn't hurt to

ompi/include/mpi.h.in

@@ -862,7 +862,8 @@ enum {
   OMPI_COMM_TYPE_CU,
   OMPI_COMM_TYPE_CLUSTER,
   MPI_COMM_TYPE_HW_UNGUIDED,
-  MPI_COMM_TYPE_HW_GUIDED
+  MPI_COMM_TYPE_HW_GUIDED,
+  MPI_COMM_TYPE_RESOURCE_GUIDED
 };
 #define OMPI_COMM_TYPE_NODE MPI_COMM_TYPE_SHARED
 

ompi/include/mpif-values.py

@@ -338,6 +338,7 @@
     'OMPI_COMM_TYPE_CLUSTER': 11,
     'MPI_COMM_TYPE_HW_UNGUIDED': 12,
     'MPI_COMM_TYPE_HW_GUIDED': 13,
+    'MPI_COMM_TYPE_RESOURCE_GUIDED': 14,
 }
 
 # IO Constants

ompi/mpi/bindings/ompi_bindings/consts.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2024      Triad National Security, LLC. All rights
+# Copyright (c) 2024-2025 Triad National Security, LLC. All rights
 #                         reserved.
 #
 # $COPYRIGHT$
@@ -157,6 +157,7 @@
     'MPI_COMM_TYPE_SHARED',
     'MPI_COMM_TYPE_HW_UNGUIDED',
     'MPI_COMM_TYPE_HW_GUIDED',
+    'MPI_COMM_TYPE_RESOURCE_GUIDED',
 ]
 
 RESERVED_WINDOWS = [

ompi/mpi/c/comm_split_type.c.in

@@ -15,7 +15,7 @@
  * Copyright (c) 2015      Research Organization for Information Science
  *                         and Technology (RIST). All rights reserved.
  * Copyright (c) 2017-2022 IBM Corporation.  All rights reserved.
- * Copyright (c) 2024      Triad National Security, LLC. All rights
+ * Copyright (c) 2024-2025 Triad National Security, LLC. All rights
  *                         reserved.
  * $COPYRIGHT$
  *
@@ -59,6 +59,7 @@ PROTOTYPE ERROR_CLASS comm_split_type(COMM comm, INT split_type, INT key,
         if ( MPI_COMM_TYPE_SHARED != split_type && // Same as OMPI_COMM_TYPE_NODE
              MPI_COMM_TYPE_HW_UNGUIDED != split_type &&
              MPI_COMM_TYPE_HW_GUIDED != split_type &&
+             MPI_COMM_TYPE_RESOURCE_GUIDED != split_type &&
 	     OMPI_COMM_TYPE_CLUSTER != split_type &&
 	     OMPI_COMM_TYPE_CU != split_type &&
 	     OMPI_COMM_TYPE_HOST != split_type &&