8 Pull requests merged by 1 person

• Relax verification on manual comp with no inputs or outputs, and instead fail before propagation (and after func inlining).

  #633 merged Jul 25, 2025

• [jax:custom_partitioning] Allow factors for non-batching dimensions to

  #631 merged Jul 24, 2025

• #sdy add an enableAutoPartitioning to PropagationOptions which will enable invoking the registered auto partitioner right after propagation.

  #629 merged Jul 24, 2025

• Automated Code Change

  #611 merged Jul 23, 2025

• Add SDY_LOG and SDY_VLOG.

  #625 merged Jul 23, 2025

• Add SDY_CHECK_LE/LT/GE/GT

  #623 merged Jul 22, 2025

• Refactor getOrCreateSharding. No behavior change.

  #622 merged Jul 22, 2025

• #sdy Introduce SDY_CHECK, SDY_CHECK_EQ, SDY_CHECK_NE (similar to absl CHECK) which calls llvm::report_fatal_error under the hood.

  #619 merged Jul 22, 2025

6 Pull requests opened by 1 person

• [SDY] dump source sharding info when output shardyDir is non-empty. This occurs when `xla_dump_to` flag is set and `xla_dump_hlo_pass_re` regex matches the shardy pass (see `getShardyDirIfShouldDump`).

  #618 opened Jul 22, 2025

• Verify that the shrdings inside the `sdy.manual_computation` shares the same mesh.

  #624 opened Jul 23, 2025

• Initial version of Shardy:MPMD OSS.

  #627 opened Jul 23, 2025

• MPMD XLA bump

  #628 opened Jul 23, 2025

• #sdy set `PropagationOptions::enableAutoPartitioning` in shardy_xla_pass and cleanup legacy auto partitioner code, as it's no longer needed.

  #630 opened Jul 24, 2025

• [SDY] Add pretty print methods for sdy attrs used with explicit collective

  #632 opened Jul 25, 2025