[all-commits] [llvm/llvm-project] 431213: [mlir][linalg] Implement patterns for reducing ran...
srcarroll via All-commits
all-commits at lists.llvm.org
Mon Jun 24 11:06:52 PDT 2024
Branch: refs/heads/main
Home: https://github.com/llvm/llvm-project
Commit: 431213c99d7707114d8e7956073a057cf1607160
https://github.com/llvm/llvm-project/commit/431213c99d7707114d8e7956073a057cf1607160
Author: srcarroll <50210727+srcarroll at users.noreply.github.com>
Date: 2024-06-24 (Mon, 24 Jun 2024)
Changed paths:
M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
M mlir/lib/Dialect/Linalg/Transforms/DropUnitDims.cpp
A mlir/test/Dialect/Linalg/rank-reduce-contraction-ops.mlir
M mlir/test/lib/Dialect/Linalg/CMakeLists.txt
A mlir/test/lib/Dialect/Linalg/TestLinalgRankReduceContractionOps.cpp
M mlir/tools/mlir-opt/mlir-opt.cpp
Log Message:
-----------
[mlir][linalg] Implement patterns for reducing rank of named linalg contraction ops (#95710)
This patch introduces pattern rewrites for reducing the rank of named
linalg contraction ops with unit spatial dim(s) to other named
contraction ops. For example `linalg.batch_matmul` with batch size 1 ->
`linalg.matmul` and `linalg.matmul` with unit LHS spatial dim ->
`linalg.vecmat`, etc. These patterns don't support reducing the rank
along reduction dimension as those don't convert to other named
contraction ops.
To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications
More information about the All-commits
mailing list