[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