hook getTypeIDFunction

Author: tgymnich

mlir/include/mlir-c/Dialect/AMDGPU.h

@@ -24,6 +24,8 @@ MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(AMDGPU, amdgpu);
 
 MLIR_CAPI_EXPORTED bool mlirTypeIsAAMDGPUTDMBaseType(MlirType type);
 
+MLIR_CAPI_EXPORTED MlirTypeID mlirAMDGPUTDMBaseTypeGetTypeID();
+
 MLIR_CAPI_EXPORTED MlirType mlirAMDGPUTDMBaseTypeGet(MlirContext ctx,
                                                      MlirType elementType);
 
@@ -33,6 +35,8 @@ MLIR_CAPI_EXPORTED MlirType mlirAMDGPUTDMBaseTypeGet(MlirContext ctx,
 
 MLIR_CAPI_EXPORTED bool mlirTypeIsAAMDGPUTDMDescriptorType(MlirType type);
 
+MLIR_CAPI_EXPORTED MlirTypeID mlirAMDGPUTDMDescriptorTypeGetTypeID();
+
 MLIR_CAPI_EXPORTED MlirType mlirAMDGPUTDMDescriptorTypeGet(MlirContext ctx);
 
 //===---------------------------------------------------------------------===//
@@ -41,6 +45,8 @@ MLIR_CAPI_EXPORTED MlirType mlirAMDGPUTDMDescriptorTypeGet(MlirContext ctx);
 
 MLIR_CAPI_EXPORTED bool mlirTypeIsAAMDGPUTDMGatherBaseType(MlirType type);
 
+MLIR_CAPI_EXPORTED MlirTypeID mlirAMDGPUTDMGatherBaseTypeGetTypeID();
+
 MLIR_CAPI_EXPORTED MlirType mlirAMDGPUTDMGatherBaseTypeGet(MlirContext ctx,
                                                            MlirType elementType,
                                                            MlirType indexType);

mlir/lib/Bindings/Python/DialectAMDGPU.cpp

@@ -20,7 +20,8 @@ using namespace mlir::python::nanobind_adaptors;
 
 static void populateDialectAMDGPUSubmodule(const nb::module_ &m) {
   auto amdgpuTDMBaseType =
-      mlir_type_subclass(m, "TDMBaseType", mlirTypeIsAAMDGPUTDMBaseType);
+      mlir_type_subclass(m, "TDMBaseType", mlirTypeIsAAMDGPUTDMBaseType,
+                         mlirAMDGPUTDMBaseTypeGetTypeID);
 
   amdgpuTDMBaseType.def_classmethod(
       "get",
@@ -31,7 +32,8 @@ static void populateDialectAMDGPUSubmodule(const nb::module_ &m) {
       nb::arg("element_type"), nb::arg("ctx") = nb::none());
 
   auto amdgpuTDMDescriptorType = mlir_type_subclass(
-      m, "TDMDescriptorType", mlirTypeIsAAMDGPUTDMDescriptorType);
+      m, "TDMDescriptorType", mlirTypeIsAAMDGPUTDMDescriptorType,
+      mlirAMDGPUTDMDescriptorTypeGetTypeID);
 
   amdgpuTDMDescriptorType.def_classmethod(
       "get",
@@ -42,7 +44,8 @@ static void populateDialectAMDGPUSubmodule(const nb::module_ &m) {
       nb::arg("cls"), nb::arg("ctx") = nb::none());
 
   auto amdgpuTDMGatherBaseType = mlir_type_subclass(
-      m, "TDMGatherBaseType", mlirTypeIsAAMDGPUTDMGatherBaseType);
+      m, "TDMGatherBaseType", mlirTypeIsAAMDGPUTDMGatherBaseType,
+      mlirAMDGPUTDMGatherBaseTypeGetTypeID);
 
   amdgpuTDMGatherBaseType.def_classmethod(
       "get",

mlir/lib/CAPI/Dialect/AMDGPU.cpp

@@ -24,6 +24,10 @@ bool mlirTypeIsAAMDGPUTDMBaseType(MlirType type) {
   return isa<amdgpu::TDMBaseType>(unwrap(type));
 }
 
+MlirTypeID mlirAMDGPUTDMBaseTypeGetTypeID() {
+  return wrap(amdgpu::TDMBaseType::getTypeID());
+}
+
 MlirType mlirAMDGPUTDMBaseTypeGet(MlirContext ctx, MlirType elementType) {
   return wrap(amdgpu::TDMBaseType::get(unwrap(ctx), unwrap(elementType)));
 }
@@ -36,6 +40,10 @@ bool mlirTypeIsAAMDGPUTDMDescriptorType(MlirType type) {
   return isa<amdgpu::TDMDescriptorType>(unwrap(type));
 }
 
+MlirTypeID mlirAMDGPUTDMDescriptorTypeGetTypeID() {
+  return wrap(amdgpu::TDMDescriptorType::getTypeID());
+}
+
 MlirType mlirAMDGPUTDMDescriptorTypeGet(MlirContext ctx) {
   return wrap(amdgpu::TDMDescriptorType::get(unwrap(ctx)));
 }
@@ -48,6 +56,10 @@ bool mlirTypeIsAAMDGPUTDMGatherBaseType(MlirType type) {
   return isa<amdgpu::TDMGatherBaseType>(unwrap(type));
 }
 
+MlirTypeID mlirAMDGPUTDMGatherBaseTypeGetTypeID() {
+  return wrap(amdgpu::TDMGatherBaseType::getTypeID());
+}
+
 MlirType mlirAMDGPUTDMGatherBaseTypeGet(MlirContext ctx, MlirType elementType,
                                         MlirType indexType) {
   return wrap(amdgpu::TDMGatherBaseType::get(unwrap(ctx), unwrap(elementType),