diff --git a/mlir/lib/Bindings/Python/IRModules.cpp b/mlir/lib/Bindings/Python/IRModules.cpp --- a/mlir/lib/Bindings/Python/IRModules.cpp +++ b/mlir/lib/Bindings/Python/IRModules.cpp @@ -9,6 +9,7 @@ #include "IRModules.h" #include "PybindUtils.h" +#include "mlir-c/Registration.h" #include "mlir-c/StandardAttributes.h" #include "mlir-c/StandardTypes.h" #include "llvm/ADT/SmallVector.h" @@ -454,6 +455,7 @@ PyMlirContext *PyMlirContext::createNewContextForInit() { MlirContext context = mlirContextCreate(); + mlirRegisterAllDialects(context); return new PyMlirContext(context); } diff --git a/mlir/lib/CAPI/IR/IR.cpp b/mlir/lib/CAPI/IR/IR.cpp --- a/mlir/lib/CAPI/IR/IR.cpp +++ b/mlir/lib/CAPI/IR/IR.cpp @@ -10,7 +10,6 @@ #include "mlir/CAPI/IR.h" #include "mlir/CAPI/Utils.h" -#include "mlir/Dialect/StandardOps/IR/Ops.h" #include "mlir/IR/Attributes.h" #include "mlir/IR/Dialect.h" #include "mlir/IR/Module.h" @@ -26,10 +25,6 @@ MlirContext mlirContextCreate() { auto *context = new MLIRContext(/*loadAllDialects=*/false); - // TODO: Come up with a story for which dialects to load into the context - // and do not expand this beyond StandardOps until done so. This is loaded - // by default here because it is hard to make progress otherwise. - context->loadDialect(); return wrap(context); }