Add ELEMENT_COUNT for MPI_Get_elements_x

open-mpi · Apr 17, 2024 · 5127029 · 5127029
1 parent fc151a8
commit 5127029
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/ompi/mpi/bindings/ompi_bindings/c_type.py b/ompi/mpi/bindings/ompi_bindings/c_type.py
@@ -126,6 +126,14 @@ def parameter(self, enable_count=False):
         return f'const {count_type} {self.name}[]'
 
 
+@Type.add_type('ELEMENT_COUNT')
+class ElementCountType(Type):
+    """Special count type for MPI_Get_element_x"""
+
+    def type_text(self, enable_count=False):
+        return 'MPI_Count *'
+
+
 @Type.add_type('PARTITIONED_COUNT')
 class TypePartitionedCount(Type):
     """Count type for partitioned communication functions."""