Skip to content

Pull requests: openxla/shardy

New pull request New
16 Open 288 Closed
16 Open 288 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 introduce sdy.all_slice that slices chunks of a tensor along axes.
#308 opened Jan 17, 2025 by copybara-service bot Loading…
@tomnatan30
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! Exclude everything labeled bug with -label:bug.