srcarroll 431213c99d [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.

2024-06-24 13:06:31 -05:00

2.3 KiB

Raw Blame History

View Raw