[Mlir-commits] [mlir] [mlir][linalg] Add quantized conv2d operator with FCHW, NCHW order (PR #107740)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Sun Sep 8 00:21:40 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 8d908b8cc5f4b3aeb2303437a9c2d35654279fd9...7732ecaafa2d87af6c20afdc1d43fc96a05005e3 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-09-08 07:08:06.000000 +0000
+++ core_named_ops.py 2024-09-08 07:21:14.254451 +0000
@@ -872,10 +872,11 @@
TypeFn.cast_signed(
U, I[D.n, D.oh * S.SH + D.kh * S.DH, D.ow * S.SW + D.kw * S.DW, D.c]
)
- TypeFn.cast_signed(U, IZp)
) * (TypeFn.cast_signed(U, K[D.f, D.kh, D.kw, D.c]) - TypeFn.cast_signed(U, KZp))
+
@linalg_structured_op
def conv_2d_nchw_fchw_q(
I=TensorDef(T1, S.N, S.C, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW),
K=TensorDef(T2, S.F, S.C, S.KH, S.KW),
``````````
</details>
https://github.com/llvm/llvm-project/pull/107740
More information about the Mlir-commits
mailing list