Index: mlir/python/mlir/_mlir_libs/__init__.py =================================================================== --- mlir/python/mlir/_mlir_libs/__init__.py +++ mlir/python/mlir/_mlir_libs/__init__.py @@ -54,6 +54,7 @@ import itertools import logging from ._mlir import ir + logger = logging.getLogger(__name__) registry = ir.DialectRegistry() post_init_hooks = [] @@ -63,12 +64,12 @@ except ModuleNotFoundError: return False - logging.debug("Initializing MLIR with module: %s", module_name) + logger.debug("Initializing MLIR with module: %s", module_name) if hasattr(m, "register_dialects"): - logging.debug("Registering dialects from initializer %r", m) + logger.debug("Registering dialects from initializer %r", m) m.register_dialects(registry) if hasattr(m, "context_init_hook"): - logging.debug("Adding context init hook from %r", m) + logger.debug("Adding context init hook from %r", m) post_init_hooks.append(m.context_init_hook) return True