mirror-llvm-project/mlir/test/Dialect/Shard
zackc6 39ee115e92
[MLIR][Shard] Fold all_gather/all_slice inverse pairs (#193906)
Add a simplify pattern that replaces all_gather(all_slice(x))/all_slice(all_gather(x))
with x when grid, grid axes, and gather/slice axis match.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-05-05 11:45:47 +02:00
..
all-scatter-op-lowering.mlir [mlir][shard, mpi] Adding Shard/MPI reduce_scatter and simplification (#184189) 2026-03-03 17:34:36 +01:00
backward-sharding-propagation.mlir
canonicalization.mlir [mlir][shard, mpi] Adding Shard/MPI reduce_scatter and simplification (#184189) 2026-03-03 17:34:36 +01:00
folding.mlir [mlir][shard, mpi] Adding Shard/MPI reduce_scatter and simplification (#184189) 2026-03-03 17:34:36 +01:00
forward-backward-sharding-propagation.mlir
forward-sharding-propagation.mlir
inlining.mlir
invalid.mlir [mlir][shard, mpi] Adding Shard/MPI reduce_scatter and simplification (#184189) 2026-03-03 17:34:36 +01:00
invalid_annotated.mlir
ops.mlir [mlir][shard, mpi] Adding Shard/MPI reduce_scatter and simplification (#184189) 2026-03-03 17:34:36 +01:00
partition.mlir [mlir][shard] Empowering resharding (#180962) 2026-02-19 11:56:05 +01:00
process-multi-index-op-lowering.mlir
resharding-partition.mlir [MLIR][Shard] Fix three bugs in ND mesh resharding in Partition pass (#189241) 2026-04-13 19:57:26 +02:00
sharding-propagation-failed.mlir [mlir][shard] Small fixes to partition pass (#185050) 2026-03-09 10:47:45 +01:00
sharding-propagation.mlir
simplify.mlir [MLIR][Shard] Fold all_gather/all_slice inverse pairs (#193906) 2026-05-05 11:45:47 +02:00