[Mlir-commits] [mlir] [MLIR][Linalg] Introduce Python API for linalg.batch_matmul Ops. (PR #127614)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Tue Feb 18 02:48:57 PST 2025
github-actions[bot] wrote:
<!--LLVM CODE FORMAT COMMENT: {darker}-->
:warning: Python code formatter, darker found issues in your code. :warning:
<details>
<summary>
You can test this locally with the following command:
</summary>
``````````bash
darker --check --diff -r 2b71df5a74cb5bd67f3f34277749dc920fd35105...50236998fd9de34cf5cd5f0a9f9d9a2fc98fb19e mlir/python/mlir/dialects/linalg/__init__.py mlir/test/python/dialects/linalg/ops.py
``````````
</details>
<details>
<summary>
View the diff from darker here.
</summary>
``````````diff
--- python/mlir/dialects/linalg/__init__.py 2025-02-18 09:53:17.000000 +0000
+++ python/mlir/dialects/linalg/__init__.py 2025-02-18 10:48:25.526684 +0000
@@ -192,10 +192,11 @@
cast=cast,
)
fill_builtin_region(op.operation)
return op
+
def batch_matmul(
*ins: Union[Operation, OpView, Value],
outs: Sequence[Union[Operation, OpView, Value]],
indexing_maps: Optional[Sequence[AffineMapAttr]] = None,
):
--- test/python/dialects/linalg/ops.py 2025-02-18 09:53:17.000000 +0000
+++ test/python/dialects/linalg/ops.py 2025-02-18 10:48:25.703537 +0000
@@ -465,10 +465,11 @@
indexing_maps=[a_map, b_transposed_map, c_map],
)
print(module)
+
# CHECK-LABEL: TEST: testBatchMatmulOp
@run
def testBatchMatmulOp():
with Context(), Location.unknown():
module = Module.create()
``````````
</details>
https://github.com/llvm/llvm-project/pull/127614
More information about the Mlir-commits
mailing list