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> |
||
|---|---|---|
| .. | ||
| all-scatter-op-lowering.mlir | ||
| backward-sharding-propagation.mlir | ||
| canonicalization.mlir | ||
| folding.mlir | ||
| forward-backward-sharding-propagation.mlir | ||
| forward-sharding-propagation.mlir | ||
| inlining.mlir | ||
| invalid.mlir | ||
| invalid_annotated.mlir | ||
| ops.mlir | ||
| partition.mlir | ||
| process-multi-index-op-lowering.mlir | ||
| resharding-partition.mlir | ||
| sharding-propagation-failed.mlir | ||
| sharding-propagation.mlir | ||
| simplify.mlir | ||