All pull requests

Pull requests

• [SDY] Guard InsertExplicitReshardsPass against external function declarations.

  #1675·copybara-service[bot] opened Oct 13, 2026Bot·

  ·
  

• Push ApplyShardingConstraints and ShardingGroupImport passes out of shardy import pipeline.

  #1666·copybara-service[bot] opened Oct 9, 2026Bot·

  ·
  

• [SDY] Enable full version of explicit reshards by default.

  #1647·copybara-service[bot] opened Oct 6, 2026Bot·

  ·
  

• Modify runShardyPartitioner to set markPartialResultWithUnreducedAxes when the partitioner stage is specified.

  #1640·copybara-service[bot] opened Oct 5, 2026Bot·

  ·
  

• Guard empty sharding group in findCommonSharding

  #1473·EylonKrause opened Jul 16, 2026·

  ·

• Fix out-of-bounds read for unscattered dims in multi-dim reduce-scatter lowering

  #1463·EylonKrause opened Jul 9, 2026·

  ·

• Diagnose non-ranked region types in ManualComputationOp instead of crashing

  #1462·EylonKrause opened Jul 9, 2026·

  ·

• Reject negative index in getValueFromEdgeValueRef

  #1454·EylonKrause opened Jul 6, 2026·

  ·
  1

• reshard: fix reverse-iterator direction in getGatheringAxes

  #1453·EylonKrause opened Jul 3, 2026·

  ·
  1

• Re-enable SplitReshardingDimensionsPass in the export pipeline

  #1444·copybara-service[bot] opened Jun 29, 2026Bot·

  ·
  

• Optimize Wiz export pipeline for shape polymorphism.

  #1437·copybara-service[bot] opened Jun 27, 2026Bot·

  ·

• docs: Add docstring to mlirRegisterAllSdyPassesAndPipelines

  #1424·Lang-Qiu opened Jun 22, 2026·

  ·
  1

• [consts] Fix handling of closed-over constants in MPMD

  #1404·copybara-service[bot] opened Jun 9, 2026Bot·

  ·

• Extend the canonicalizer to also inline manual_computation with equal global and

  #1396·copybara-service[bot] opened Jun 5, 2026Bot·

  ·
  

• [mpmd] Prefer producer fragment when merging inferred fragments to avoid false dependencies.

  #1393·copybara-service[bot] opened Jun 4, 2026Bot·

  ·

• [mpmd] Restrict MergeFragments to adjacent fragments.

  #1389·copybara-service[bot] opened Jun 2, 2026Bot·

  ·

• [mpmd] Merge inferred fragments inline in UniquifyFunctionInputsOutputsPass.

  #1382·copybara-service[bot] opened Jun 1, 2026Bot·

  ·

• [mpmd] Add utils for finding and moving fragments during fragment merging.

  #1381·copybara-service[bot] opened Jun 1, 2026Bot·

  ·

• [mpmd] Update uniquify test to chain merge-inferred-fragments pass.

  #1377·copybara-service[bot] opened Jun 1, 2026Bot·

  ·

• [mpmd] Refactor UniquifyFunctionInputsOutputsPass to use shared utilities.

  #1374·copybara-service[bot] opened May 29, 2026Bot·

  ·

• [mpmd] Simplify inline merge: always merge into last same-mesh fragment.

  #1373·copybara-service[bot] opened May 29, 2026Bot·

  ·

• [mpmd] Fix attribute preservation and add tests for inline merge in UniquifyFunctionInputsOutputsPass.

  #1369·copybara-service[bot] opened May 29, 2026Bot·

  ·

• Ignore intra-mesh broadcasts, as they are trivial ops.

  #1363·copybara-service[bot] opened May 21, 2026Bot·

  ·

• [SDY:MPMD] Remove MIN_JAXLIB_VERSION_FOR_SCHEDULE_RULES

  #1362·copybara-service[bot] opened May 21, 2026Bot·

  ·

• Inline producer-consumer merging into UniquifyFunctionInputsOutputsPass.

  #1339·copybara-service[bot] opened May 13, 2026Bot·

  ·