commit bf475bbf7c2bbcea4c0e1efbda441d618ef07a6b
author Stella Laurenzo <laurenzo@google.com> Thu May 11 08:41:23 2023 -0700
committer GitHub <noreply@github.com> Thu May 11 08:41:23 2023 -0700
tree ae36851f7fdfa90b04754a93a85b546f23cf3589
parent 29c6e8df6e62b7b526e64149deb40b0573d27308

Add python registration support for IREE dialects. (#13549)

Unlike the previous mechanism, this will automatically register all
known dialects with any context created from python.

Fixes #13477 after #13395 removed an API that was load bearing to a
customer use case.

compiler/setup.py

diff --git a/compiler/setup.py b/compiler/setup.py
index f52b20f..df9e1e9 100644
--- a/compiler/setup.py
+++ b/compiler/setup.py
@@ -408,6 +408,7 @@
         # it also needs to be enabled on the build side.
         # CMakeExtension("iree.compiler._mlir_libs._mlirHlo"),
         CMakeExtension("iree.compiler._mlir_libs._mlirLinalgPasses"),
+        CMakeExtension("iree.compiler._mlir_libs._site_initialize_0"),
     ],
     cmdclass={
         "build": CustomBuild,