add support for typeclass

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

Author: hppritcha

ompi/mpi/bindings/ompi_bindings/c.py

@@ -547,9 +547,6 @@ def generate_combiner_convert_fn_intern_to_abi(self):
     def generate_typeclass_convert_fn(self):
         self.generic_convert(ConvertFuncs.TYPECLASS, 'typeclass', 'int', consts.TYPECLASS_VALUES)
 
-    def generate_typeclass_convert_fn(self):
-        self.generic_convert(ConvertFuncs.TYPECLASS, 'typeclass', 'int', consts.TYPECLASS_VALUES)
-
     def generate_win_lock_convert_fn(self):
         self.generic_convert(ConvertFuncs.WIN_LOCK, 'lock_assert', 'int', consts.WIN_LOCK_VALUES)
 
@@ -749,6 +746,7 @@ def dump_code(self):
         self.generate_t_cb_safety_convert_fn()
         self.generate_win_lock_convert_fn()
         self.generate_whence_convert_fn()
+        self.generate_typeclass_convert_fn()
 
         #
         # the following only need intern to abi converters

ompi/mpi/bindings/ompi_bindings/c_type.py

@@ -3212,4 +3212,21 @@ def type_text(self, enable_count=False):
     def argument(self):
         return f'{self.name}'
 
+@Type.add_type('TYPECLASS', abi_type=['ompi'])
+class TypeClass(Type):
+
+    def type_text(self, enable_count=False):
+        return 'int'
+
+@Type.add_type('TYPECLASS', abi_type=['standard'])
+class TypeClassStandard(StandardABIType):
+
+    @property
+    def init_code(self):
+        return [f'int {self.tmpname} = {ConvertFuncs.TYPECLASS}({self.name});']
+
+    def type_text(self, enable_count=False):
+        return 'int'
+
+
 

ompi/mpi/c/type_match_size.c.in

@@ -12,7 +12,7 @@
  * Copyright (c) 2009      Sun Microsystems, Inc. All rights reserved.
  * Copyright (c) 2015      Research Organization for Information Science
  *                         and Technology (RIST). All rights reserved.
- * Copyright (c) 2024      Triad National Security, LLC. All rights
+ * Copyright (c) 2024-2025 Triad National Security, LLC. All rights
  *                         reserved.
  * $COPYRIGHT$
  *
@@ -30,7 +30,7 @@
 #include "ompi/datatype/ompi_datatype.h"
 #include "ompi/datatype/ompi_datatype_internal.h"
 
-PROTOTYPE ERROR_CLASS type_match_size(INT typeclass, INT size, DATATYPE_OUT type)
+PROTOTYPE ERROR_CLASS type_match_size(TYPECLASS typeclass, INT size, DATATYPE_OUT type)
 {
     if (MPI_PARAM_CHECK) {
         OMPI_ERR_INIT_FINALIZE(FUNC_NAME);