mirror of
https://github.com/intel/llvm.git
synced 2026-01-19 01:15:50 +08:00
[mlir][Linalg][Python] Create the body of builtin named Linalg ops
This revision adds support to properly add the body of registered builtin named linalg ops. At this time, indexing_map and iterator_type support is still missing so the op is not executable yet. Differential Revision: https://reviews.llvm.org/D99578
This commit is contained in:
@@ -10,6 +10,7 @@
|
||||
|
||||
#include "PybindUtils.h"
|
||||
|
||||
#include "DialectLinalg.h"
|
||||
#include "ExecutionEngine.h"
|
||||
#include "Globals.h"
|
||||
#include "IRModule.h"
|
||||
@@ -225,4 +226,9 @@ PYBIND11_MODULE(_mlir, m) {
|
||||
auto executionEngineModule =
|
||||
m.def_submodule("execution_engine", "MLIR JIT Execution Engine");
|
||||
populateExecutionEngineSubmodule(executionEngineModule);
|
||||
|
||||
// Define and populate Linalg submodule.
|
||||
auto dialectsModule = m.def_submodule("dialects");
|
||||
auto linalgModule = dialectsModule.def_submodule("linalg");
|
||||
populateDialectLinalgSubmodule(linalgModule);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user