[Mlir-commits] [mlir] Add missing `linalg.batch_vecmat` named op (PR #70218)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Wed Oct 25 08:16:04 PDT 2023


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 cc2fbc648d7babbfa612f4f5eda3160212ef6ca7..5476ede320266de91111833793ac4fcc934b45d1 mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
``````````

</details>

<details>
<summary>
View the diff from darker here.
</summary>

``````````diff
--- core_named_ops.py	2023-10-25 15:00:32.000000 +0000
+++ core_named_ops.py	2023-10-25 15:15:57.489478 +0000
@@ -513,10 +513,11 @@
     domain(D.b, D.m, D.k)
     implements(ContractionOpInterface)
     C[D.b, D.m] += TypeFn.cast_signed(U, A[D.b, D.m, D.k]) * TypeFn.cast_signed(
         U, B[D.b, D.k]
     )
+
 
 @linalg_structured_op
 def batch_vecmat(
     A=TensorDef(T1, Batch, S.K),
     B=TensorDef(T2, Batch, S.K, S.N),

``````````

</details>


https://github.com/llvm/llvm-project/pull/70218


More information about the Mlir-commits mailing list