Skip to content

Pull requests: openxla/shardy

New pull request New
12 Open 226 Closed
12 Open 226 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

Rename AxesWithTail to AxisListRef and move to its own file on shardy/dialect/sdy/ir.
#238 opened Nov 26, 2024 by copybara-service bot Loading…
#sdy #debug Add test for ManualComputationOp with manual axes in source sharding debugging.
#237 opened Nov 26, 2024 by copybara-service bot Loading…
@bartchr808
Trim axes to use the largest prefix that does not overlap with the best, instead of erasing.
#225 opened Nov 19, 2024 by copybara-service bot Loading…
Add an option to skip converting sharding_constraint to reshard
#220 opened Nov 18, 2024 by copybara-service bot Loading…
[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! Type g p on any issue or pull request to go back to the pull request listing page.