[all-commits] [llvm/llvm-project] e444dc: [mlir][Transform] Add classify_convolution_dims op

Quinn Dawkins via All-commits all-commits at lists.llvm.org
Mon Jul 31 13:18:14 PDT 2023


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: e444dcef7cadb29490deca726cb743e38d0412b3
      https://github.com/llvm/llvm-project/commit/e444dcef7cadb29490deca726cb743e38d0412b3
  Author: Quinn Dawkins <quinn at nod-labs.com>
  Date:   2023-07-31 (Mon, 31 Jul 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgInterfaces.h
    M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgMatchOps.td
    M mlir/lib/Dialect/Linalg/IR/LinalgInterfaces.cpp
    M mlir/lib/Dialect/Linalg/TransformOps/LinalgMatchOps.cpp
    M mlir/test/Dialect/Linalg/match-ops-interpreter.mlir

  Log Message:
  -----------
  [mlir][Transform] Add classify_convolution_dims op

Includes `inferConvolutionDims` based on the existing helper for
contractions, `inferContractionDims`. This allows matching and
identifying the relevant dims for a convolution sub-computation of
a linalg operation.

Additionally adds stride/dilations inference to the captures and
convolution interface matcher.

Differential Revision: https://reviews.llvm.org/D156080




More information about the All-commits mailing list