Skip to content

Pull requests: openxla/shardy

New pull request New
15 Open 192 Closed
15 Open 192 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

Find common axes deterministically by not depending on DenseMap iteration order to find the best factor-axes pair.
#208 opened Nov 14, 2024 by copybara-service bot Loading…
Use AxisRefsWithTail instead of ArrayRef of AxisRefs, so the tailing AxisRefAttr can be modified on factor-axis pairs.
#205 opened Nov 14, 2024 by copybara-service bot Loading…
Use DenseMap of factor-axis pairs instead of a vector whose elements correspond to factors.
#201 opened Nov 13, 2024 by copybara-service bot Loading…
Use FactorAxesAssignmentCandidate struct to simplify the logic that picks the best factor axes pair.
#198 opened Nov 12, 2024 by copybara-service bot Loading…
[SDY] refactor propagation functions in basic_propagation.cc to utilize parameter struct in attempt to cleanup 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
Add sdy.all_gather
#180 opened Nov 6, 2024 by copybara-service bot Loading…
Add basic documentation structure and pages
#175 opened Nov 4, 2024 by melissawm Loading…
117
Shardy. Fix sharding rule for sort.
#136 opened Oct 15, 2024 by copybara-service bot Loading…
@ZixuanJiang
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:>2024-11-11.