[all-commits] [llvm/llvm-project] 39ee11: [MLIR][Shard] Fold all_gather/all_slice inverse pa...

[zackc6 via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 39ee115e92152449437a644dcf491ba659d8ce4d
      https://github.com/llvm/llvm-project/commit/39ee115e92152449437a644dcf491ba659d8ce4d
  Author: zackc6 <zackchen666 at gmail.com>
  Date:   2026-05-05 (Tue, 05 May 2026)

  Changed paths:
    M mlir/lib/Dialect/Shard/Transforms/Simplify.cpp
    M mlir/test/Dialect/Shard/simplify.mlir

  Log Message:
  -----------
  [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 at cursor.com>



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications