All pull requests

Pull requests

• [SDY] Support partitioning single-dim size-1 stablehlo.dynamic_slice with stablehlo.collective_broadcast.

  #1652·copybara-service[bot] opened Oct 7, 2026Bot·

  ·
  

• Check-fail on unknown mesh references in Shardy passes and utilities.

  #1651·copybara-service[bot] opened Oct 7, 2026Bot·

  ·

• Add patterns for sdy.constant and stablehlo.constant.

  #1650·copybara-service[bot] opened Oct 7, 2026Bot·

  ·
  

• Fix reshard-to-collectives to not remove common prefix for dimension d when dimsTounshard[d] is true.

  #1649·copybara-service[bot] opened Oct 7, 2026Bot·

  ·
  

• Add sdy.collective_permute to generic op.

  #1648·copybara-service[bot] opened Oct 7, 2026Bot·

  ·
  

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

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

  ·
  

• Replace # pytype: disable supprressions with # pyrefly: ignore

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

  ·

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

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

  ·
  

• [SDY][Refactor][1/2] Rename maximal to single-device in Shardy.

  #1604·copybara-service[bot] opened Sep 15, 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·

  ·