Skip to content

Pull requests: openxla/shardy

New pull request New
14 Open 208 Closed
14 Open 208 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

Consider all prefixes of a tensor sharding so that largest available prefix can be assigned to factors.
#223 opened Nov 18, 2024 by copybara-service bot Loading…
#sdy Add debugging tool to show the origin shardings on the MLIR module.
#221 opened Nov 18, 2024 by copybara-service bot Loading…
@bartchr808
Add an option to skip converting sharding_constraint to reshard
#220 opened Nov 18, 2024 by copybara-service bot Loading…
Count factor-axes candidates accurately for when one axes is prefix of another, and extend from existing assignment.
#218 opened Nov 18, 2024 by copybara-service bot Loading…
[SDY] refactor propagation functions in basic_propagation.cc to utilize 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
Add sdy.all_gather
#180 opened Nov 6, 2024 by copybara-service bot Loading…
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! Follow long discussions with comments:>50.