| 1 | //===- DialectLinalg.cpp - Pybind module for Linalg dialect API support --===// |
|---|
| 2 | // |
|---|
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
|---|
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
|---|
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
|---|
| 6 | // |
|---|
| 7 | //===----------------------------------------------------------------------===// |
|---|
| 8 | |
|---|
| 9 | #include "mlir-c/Dialect/Linalg.h" |
|---|
| 10 | #include "mlir-c/IR.h" |
|---|
| 11 | #include "mlir/Bindings/Python/PybindAdaptors.h" |
|---|
| 12 | |
|---|
| 13 | namespace py = pybind11; |
|---|
| 14 | |
|---|
| 15 | static void populateDialectLinalgSubmodule(py::module m) { |
|---|
| 16 | m.def( |
|---|
| 17 | "fill_builtin_region" , |
|---|
| 18 | [](MlirOperation op) { mlirLinalgFillBuiltinNamedOpRegion(op); }, |
|---|
| 19 | py::arg("op" ), |
|---|
| 20 | "Fill the region for `op`, which is assumed to be a builtin named Linalg " |
|---|
| 21 | "op." ); |
|---|
| 22 | } |
|---|
| 23 | |
|---|
| 24 | PYBIND11_MODULE(_mlirDialectsLinalg, m) { |
|---|
| 25 | m.doc() = "MLIR Linalg dialect." ; |
|---|
| 26 | |
|---|
| 27 | populateDialectLinalgSubmodule(m); |
|---|
| 28 | } |
|---|
| 29 | |
|---|
