diff options
Diffstat (limited to 'mlir/python/mlir/ir.py')
-rw-r--r-- | mlir/python/mlir/ir.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/mlir/python/mlir/ir.py b/mlir/python/mlir/ir.py index 9a6ce462..6f37266 100644 --- a/mlir/python/mlir/ir.py +++ b/mlir/python/mlir/ir.py @@ -5,7 +5,11 @@ from ._mlir_libs._mlir.ir import * from ._mlir_libs._mlir.ir import _GlobalDebug from ._mlir_libs._mlir import register_type_caster, register_value_caster -from ._mlir_libs import get_dialect_registry +from ._mlir_libs import ( + get_dialect_registry, + append_load_on_create_dialect, + get_load_on_create_dialects, +) # Convenience decorator for registering user-friendly Attribute builders. |