All pull requests

Pull requests

• [SDY] Only split axes onto batching dimensions when they saturate remaining capacity.

  #1837·copybara-service[bot] opened Nov 14, 2026Bot·

  ·
  

• [SDY][OptimalReshards][Feature-Request] Support non-divisible tensor shapes in ReshardGraph optimality test.

  #1836·copybara-service[bot] opened Nov 13, 2026Bot·

  ·
  

• [SDY] Look through sharding constraints and negative-index selects in sdy-explicit-gather-scatter-batching.

  #1835·copybara-service[bot] opened Nov 13, 2026Bot·

  ·
  

• [SDY] Skip redundant target-0 zero masking in single-source slice lowering.

  #1834·copybara-service[bot] opened Nov 13, 2026Bot·

  ·
  

• [SDY] Rewrite shard-aligned all_gather + slice [k*K : (k+1)*K] via all_reduce for any shard k.

  #1833·copybara-service[bot] opened Nov 13, 2026Bot·

  ·
  

• [SDY] Compute channel ID dynamically in sdy-resolve-permutation-factors

  #1832·copybara-service[bot] opened Nov 13, 2026Bot·

  ·
  

• Commute binary ops with all-reduce in OptimizeCollectivesPass.

  #1827·copybara-service[bot] opened Nov 12, 2026Bot·

  ·
  

• [SDY MPMD] Import MLIR from JAX in the MPMD JAX API

  #1822·i-chaochen opened Nov 11, 2026·

  ·

• [SDY] [MPMD] Parse every boolean MPMD partitioning option

  #1821·i-chaochen opened Nov 11, 2026·

  ·

• [SDY][OptimalReshards][3/3] Piggyback trailing all-gather axes onto all-to-all.

  #1797·copybara-service[bot] opened Nov 4, 2026Bot·

  ·
  

• Fix non-unit strides on sharded dimensions in Shardy stablehlo.slice HALO exchange.

  #1782·copybara-service[bot] opened Nov 2, 2026Bot·

  ·

• Support HALO exchange for sharded stablehlo.convolution operations in Shardy.

  #1776·copybara-service[bot] opened Nov 2, 2026Bot·

  ·
  

• Support HALO exchange for sharded stablehlo.reduce_window operations in Shardy.

  #1752·copybara-service[bot] opened Oct 27, 2026Bot·

  ·
  

• Propagate unreduced axes backwards through metadata-only ops in ApplyShardingConstraints.

  #1735·copybara-service[bot] opened Oct 21, 2026Bot·

  ·
  

• prototype: shardy no-flatten. drop flattener from conservative mode.

  #1726·copybara-service[bot] opened Oct 20, 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

• 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·

  ·