Skip to content

Commit 639951b

Browse files
committed
fix formatting
1 parent 8ff731c commit 639951b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1126,6 +1126,7 @@ def conv_3d_ncdhw_fcdhw(
11261126
],
11271127
) * TypeFn.cast_signed(U, K[D.f, D.c, D.kd, D.kh, D.kw])
11281128

1129+
11291130
@linalg_structured_op
11301131
def conv_3d_ncdhw_fcdhw_q(
11311132
I=TensorDef(
@@ -1152,7 +1153,7 @@ def conv_3d_ncdhw_fcdhw_q(
11521153
implements(ConvolutionOpInterface)
11531154
domain(D.n, D.od, D.oh, D.ow, D.f, D.kd, D.kh, D.kw, D.c)
11541155
O[D.n, D.f, D.od, D.oh, D.ow] += (
1155-
TypeFn.cast_signed(
1156+
TypeFn.cast_signed(
11561157
U,
11571158
I[
11581159
D.n,
@@ -1161,7 +1162,8 @@ def conv_3d_ncdhw_fcdhw_q(
11611162
D.oh * S.SH + D.kh * S.DH,
11621163
D.ow * S.SW + D.kw * S.DW,
11631164
],
1164-
) - TypeFn.cast_signed(U, IZp)
1165+
)
1166+
- TypeFn.cast_signed(U, IZp)
11651167
) * (
11661168
TypeFn.cast_signed(U, K[D.f, D.c, D.kd, D.kh, D.kw])
11671169
- TypeFn.cast_signed(U, KZp)

0 commit comments

Comments
 (0)