Skip to content

Pull requests: openxla/shardy

New pull request New
24 Open 309 Closed
24 Open 309 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

#sdy Add more utility methods to the OpShardingRuleAttr.
#337 opened Jan 23, 2025 by copybara-service bot Loading…
@ZixuanJiang
Fixed url encoding for MD.
#336 opened Jan 23, 2025 by copybara-service bot Loading…
#sdy support ffi Python callbacks during round tripping.
#332 opened Jan 22, 2025 by copybara-service bot Loading…
@bartchr808
#sdy Add Shardy translate mesh pass.
#330 opened Jan 22, 2025 by copybara-service bot Loading…
@bartchr808
Shortcut to find common axes when the operation is unary and only one tensor is incompatible individually.
#329 opened Jan 22, 2025 by copybara-service bot Loading…
#sdy Initial stage of converting any sdy.reshard to collectives.
#323 opened Jan 21, 2025 by copybara-service bot Loading…
@tomnatan30
Support cases that the operation has factors that need replication but none is sharded.
#319 opened Jan 21, 2025 by copybara-service bot Loading…
#sdy add sdy.all_reduce
#315 opened Jan 20, 2025 by copybara-service bot Loading…
#sdy #debug support debugging tool to save the edge source shardings.
#310 opened Jan 20, 2025 by copybara-service bot Loading…
@bartchr808
Add section for jax.experimental.custom_partitioning.
#307 opened Jan 15, 2025 by copybara-service bot Loading…
@bixia1
Do not add explicit reshards for CholeskyOp.
#289 opened Jan 6, 2025 by copybara-service bot Loading…
Do not add explicit reshards for ReverseOp.
#288 opened Jan 6, 2025 by copybara-service bot Loading…
Add a unit test for insert-explicit-reshards pass for stablehlo.reverse.
#287 opened Jan 6, 2025 by copybara-service bot Loading…
Update to match upstream API change (NFC).
#282 opened Dec 23, 2024 by copybara-service bot Loading…
@jpienaar
#sdy. Fix sharding rule for SliceOp.
#267 opened Dec 13, 2024 by copybara-service bot Loading…
@ZixuanJiang
#sdy fix constant splitter taking too long.
#251 opened Dec 6, 2024 by copybara-service bot Loading…
@tomnatan30
[SDY] refactor propagation functions in basic_propagation.cc to utilize a parameter struct (to cleanup function signatures).
#192 opened Nov 10, 2024 by copybara-service bot Loading…
@Varcho
test change test change test change
#182 opened Nov 7, 2024 by copybara-service bot Loading…
@tomnatan30
Handle when JAX saves the manual axes on the CallOp of the shmap_body.
#92 opened Sep 9, 2024 by copybara-service bot Loading…
@bartchr808
Support JAX shard map lowering with Shardy enabled.
#91 opened Sep 6, 2024 by copybara-service bot Loading…
@bartchr808
failing build test
#79 opened Aug 28, 2024 by copybara-service bot Loading…
@tomnatan30
failing test test
#78 opened Aug 28, 2024 by copybara-service bot Loading…
@tomnatan30
#sdy Add custom-call sharding rule registry.
#50 opened Aug 8, 2024 by copybara-service bot Loading…
@bixia1
Fixed compilation on Windows.
#31 opened Jul 30, 2024 by eaplatanios Loading…
5
ProTip! Updated in the last three days: updated:>2025-01-20.