[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