[all-commits] [llvm/llvm-project] 3f9b96: [mlir][linalg] Support non-unit dilations in im2co...

[Zmicier Prybysh via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 3f9b96996cac75b960adf84f5722afe48c20b480
      https://github.com/llvm/llvm-project/commit/3f9b96996cac75b960adf84f5722afe48c20b480
  Author: Zmicier Prybysh <zprybysh at baylibre.com>
  Date:   2026-07-31 (Fri, 31 Jul 2026)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/ConvertConv2DToImg2Col.cpp
    M mlir/test/Dialect/Linalg/convert-conv2d-to-img2col.mlir

  Log Message:
  -----------
  [mlir][linalg] Support non-unit dilations in im2col decomposition (#208424)

The im2col patterns for conv_2d_nhwc_hwcf, conv_2d_nchw_fchw and
conv_2d_nhwc_fhwc avoided non-unit dilations, but the restriction
doesn't seem to be fundamental: dilation only affects the gather step,
which can fold it into its indexing map. Generalize the convolved index
expression from `oh * stride + fh` to `oh * stride + fh * dilation` and
drop the match failures.

Verified by comparing the results of a dilated convolution lowered
directly to loops against the im2col decomposition with mlir-runner.

Assisted-By: Claude Code (for generating tests).



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications