[all-commits] [llvm/llvm-project] 4f7ab7: [mlir][mesh] add support in spmdization for incomp...

[Boian Petkantchin via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 4f7ab789bf43b49914815bdf4e4c3703f92e781d
      https://github.com/llvm/llvm-project/commit/4f7ab789bf43b49914815bdf4e4c3703f92e781d
  Author: Boian Petkantchin <boian.petkantchin at amd.com>
  Date:   2024-02-22 (Thu, 22 Feb 2024)

  Changed paths:
    M mlir/lib/Dialect/Mesh/Transforms/Spmdization.cpp
    M mlir/test/Dialect/Mesh/spmdization.mlir

  Log Message:
  -----------
  [mlir][mesh] add support in spmdization for incomplete sharding annotations (#82442)

Don't require that `mesh.shard` operations come in pairs. If there is
only a single `mesh.shard` operation we assume that the producer result
and consumer operand have the same sharding.



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