cfRod wrote: Could you also add an integration test similar to https://github.com/llvm/llvm-project/blob/main/mlir/test/Integration/Dialect/Linalg/CPU/test-conv-2d-nhwc-hwcf-call.mlir ? https://github.com/llvm/llvm-project/pull/68567