From d86d019d0e3231fb20dcbcf42745c4853e3dd25a Mon Sep 17 00:00:00 2001 From: Nirvedh Date: Tue, 1 Apr 2025 10:53:46 -0500 Subject: [PATCH] [NFC][mlir] Update generate script for conv_3d_ncdhw_fcdhw Signed-off-by: Nirvedh --- mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py b/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py index 040663c882a08..48e724d80c926 100644 --- a/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py +++ b/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py @@ -1140,7 +1140,7 @@ def conv_3d_ncdhw_fcdhw( them to the same data type as the accumulator/output. """ implements(ConvolutionOpInterface) - domain(D.n, D.od, D.oh, D.ow, D.f, D.kd, D.kh, D.kw, D.c) + domain(D.n, D.f, D.od, D.oh, D.ow, D.c, D.kd, D.kh, D.kw) O[D.n, D.f, D.od, D.oh, D.ow] += TypeFn.cast_signed( U, I[