[Mlir-commits] [mlir] [mlir][python] allow upstream dialect registration (PR #74252)

Maksim Levental llvmlistbot at llvm.org
Mon Dec 4 04:46:35 PST 2023


================
@@ -423,7 +423,30 @@ declare_mlir_python_extension(MLIRPythonExtension.Core
     MLIRCAPIInterfaces
 
     # Dialects
+    MLIRCAPIAMDGPU
+    MLIRCAPIArith
+    MLIRCAPIAsync
+    MLIRCAPIControlFlow
     MLIRCAPIFunc
+    MLIRCAPIGPU
+    MLIRCAPILLVM
+    MLIRCAPILinalg
+    MLIRCAPIMLProgram
+    MLIRCAPIMath
+    MLIRCAPIMemRef
+    MLIRCAPINVGPU
+    MLIRCAPINVVM
+    MLIRCAPIOpenMP
+    MLIRCAPIPDL
+    MLIRCAPIQuant
+    MLIRCAPIROCDL
+    MLIRCAPISCF
+    MLIRCAPIShape
+    MLIRCAPISparseTensor
+    MLIRCAPITensor
+    MLIRCAPITransformDialect
+    MLIRCAPIVector
+    MLIRCAPIRemainingDialects
----------------
makslevental wrote:

The issue is that right now most of dialects don't have their own pybind modules. I was trying to avoid introducing 20 new modules that just have `register_dialect` in them. I also couldn't put these in `RegisterEverything` because the mere presence of that module triggers the omnibus loading.

https://github.com/llvm/llvm-project/pull/74252


More information about the Mlir-commits mailing list