Add PARTITIONED_COUNT type

open-mpi · Apr 1, 2024 · 01fa44e · 01fa44e
1 parent cad1e4e
commit 01fa44e
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('PARTITIONED_COUNT')
+class TypePartitionedCount(Type):
+    """Count type for partitioned communication functions."""
+
+    def type_text(self, enable_count=False):
+        return 'MPI_Count'
+
+
 @Type.add_type('DISPL_ARRAY')
 class TypeDisplArray(Type):