test PyOperationBase method call

Author: makslevental

mlir/include/mlir/Bindings/Python/IRModule.h

@@ -571,7 +571,7 @@ class PyOperationBase {
 
   /// Verify the operation. Throws `MLIRError` if verification fails, and
   /// returns `true` otherwise.
-  bool verify();
+  MLIR_CAPI_EXPORTED bool verify();
 
   /// Each must provide access to the raw Operation.
   virtual PyOperation &getOperation() = 0;

mlir/lib/Bindings/Python/DialectGPU.cpp

@@ -9,6 +9,7 @@
 #include "mlir-c/Dialect/GPU.h"
 #include "mlir-c/IR.h"
 #include "mlir-c/Support.h"
+#include "mlir/Bindings/Python/IRModule.h"
 #include "mlir/Bindings/Python/Nanobind.h"
 #include "mlir/Bindings/Python/NanobindAdaptors.h"
 
@@ -28,6 +29,7 @@ NB_MODULE(_mlirDialectsGPU, m) {
   //===-------------------------------------------------------------------===//
   // AsyncTokenType
   //===-------------------------------------------------------------------===//
+  m.def("blahblah", [](PyOperationBase &op) { op.verify(); });
 
   auto mlirGPUAsyncTokenType =
       mlir_type_subclass(m, "AsyncTokenType", mlirTypeIsAGPUAsyncTokenType);

mlir/python/CMakeLists.txt

@@ -881,3 +881,4 @@ add_mlir_python_modules(MLIRPythonModules
   COMMON_CAPI_LINK_LIBS
     MLIRPythonCAPI
 )
+add_dependencies(MLIRPythonModules.extension._mlirDialectsGPU.dso MLIRPythonModules.extension._mlir.dso)

mlir/python/mlir/dialects/gpu/__init__.py

@@ -2,6 +2,7 @@
 #  See https://llvm.org/LICENSE.txt for license information.
 #  SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
+from ..._mlir_libs import _mlir
 from .._gpu_ops_gen import *
 from .._gpu_enum_gen import *
 from ..._mlir_libs._mlirDialectsGPU import *

mlir/test/python/dialects/gpu/dialect.py

@@ -26,7 +26,8 @@ def testGPUPass():
 def testMMAElementWiseAttr():
     module = Module.create()
     with InsertionPoint(module.body):
-        gpu.BlockDimOp(gpu.Dimension.y)
+        b = gpu.BlockDimOp(gpu.Dimension.y)
+        gpu.blahblah(b)
     # CHECK: %block_dim_y = gpu.block_dim  y
     print(module)
     pass