[Mlir-commits] [mlir] [mlir][linalg] Add linalg.conv_2d_ngchw_gfchw_q to named ops (PR #92136)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Tue May 28 12:47:17 PDT 2024
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 4ecf2caf687014a63f0434a63fe9a522ec9be445...160d20bd6a8cfa371f27764ce0e9eff02ee64338 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 2024-05-14 17:53:36.000000 +0000
+++ core_named_ops.py 2024-05-28 19:46:50.047339 +0000
@@ -934,10 +934,11 @@
implements(ConvolutionOpInterface)
domain(D.n, D.g, D.fg, D.oh, D.ow, D.c, D.kh, D.kw)
O[D.n, D.g, D.fg, D.oh, D.ow] += TypeFn.cast_signed(
U, I[D.n, D.g, D.c, D.oh * S.SH + D.kh * S.DH, D.ow * S.SW + D.kw * S.DW]
) * TypeFn.cast_signed(U, K[D.g, D.fg, D.c, D.kh, D.kw])
+
@linalg_structured_op
def conv_2d_ngchw_gfchw_q(
I=TensorDef(
T1, S.N, S.G, S.C, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW
``````````
</details>
https://github.com/llvm/llvm-project/pull/92136
More information about the Mlir-commits
mailing list