[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