[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