Skip to content

Pull requests: openxla/shardy

New pull request New
18 Open 368 Closed
18 Open 368 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

Do not set the sharding of the sharding group op during import, and emit a warning instead of error if the group has a conflicting initial sharding.
#388 opened Feb 18, 2025 by copybara-service bot Loading…
@bartchr808
#sdy Add JAX backwards compatibility test.
#379 opened Feb 12, 2025 by copybara-service bot Loading…
@bartchr808
test structured bindings
#377 opened Feb 10, 2025 by copybara-service bot Loading…
@tomnatan30
Create dump directory if not exist.
#372 opened Feb 7, 2025 by yliu120 Loading…
Replace std::bind with std::bind_front for PropagationDirectionAlongFactor.
#363 opened Feb 5, 2025 by copybara-service bot Loading…
@ZixuanJiang
#sdy add sharding rules for ragged_all_to_all custom call
#338 opened Jan 23, 2025 by copybara-service bot Loading…
@ghpvnist
Fixed url encoding for MD.
#336 opened Jan 23, 2025 by copybara-service bot Loading…
#sdy Add Shardy translate mesh pass.
#330 opened Jan 22, 2025 by copybara-service bot Loading…
@bartchr808
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
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! Filter pull requests by the default branch with base:main.