Pull requests: openxla/shardy
Pull requests list
Add a test case that GSPMD cannot handle. Shardy can handle it with explicit reshards.
#477
opened Apr 17, 2025 by
copybara-service
bot
Loading…
#sdy Optionally try and get the tensor sharding off the frontend attr on the function arguments.
#472
opened Apr 17, 2025 by
copybara-service
bot
Loading…
Remove free axes that introduce padding in
stablehlo_round_trip/shard_map_export.cc.
#465
opened Apr 11, 2025 by
copybara-service
bot
Loading…
#sdy Properly handle token types in JAX and
ManualComputationOp.
#462
opened Apr 9, 2025 by
copybara-service
bot
Loading…
[SDY+ModelExplorer][2/4] Basic SDY support in model explorer. Adding hierarchical node information to SDY ops.
#461
opened Apr 9, 2025 by
copybara-service
bot
Loading…
[SDY+ModelExplorer][1/4] Basic SDY support in model explorer. Register Shardy dialect in the mlir-to-json converter.
#458
opened Apr 8, 2025 by
copybara-service
bot
Loading…
Test commit to ensure I haven't broken the config
#427
opened Mar 19, 2025 by
copybara-service
bot
Loading…
#sdy add temporary pass in XLA SDY import to not allow meshes with different axis shapes.
#415
opened Mar 11, 2025 by
copybara-service
bot
Loading…
Don't allow sideways propagation between operands/results when propagation direction is forward/backwards.
#410
opened Mar 6, 2025 by
copybara-service
bot
Loading…
Replace std::bind with std::bind_front for PropagationDirectionAlongFactor.
#363
opened Feb 5, 2025 by
copybara-service
bot
Loading…
#sdy add sharding rules for ragged_all_to_all custom call
#338
opened Jan 23, 2025 by
copybara-service
bot
Loading…
Update to match upstream API change (NFC).
#282
opened Dec 23, 2024 by
copybara-service
bot
Loading…
#sdy fix constant splitter taking too long.
#251
opened Dec 6, 2024 by
copybara-service
bot
Loading…
#sdy Add custom-call sharding rule registry.
#50
opened Aug 8, 2024 by
copybara-service
bot
Loading…
ProTip!
Add no:assignee to see everything that’s not assigned.