Skip to content

Pull requests: openxla/shardy

New pull request New
19 Open 394 Closed
19 Open 394 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

Remove todo for the case special dimensions are unsharded as this case is handled fully.
#419 opened Mar 15, 2025 by copybara-service bot Loading…
#sdy add temporary pass in XLA SDY import to not allow meshes with different axis shapes.
#415 opened Mar 11, 2025 by copybara-service bot Loading…
@bartchr808
#sdy Introduce a temporary pass that inserts explicit reshards to apply certain optimizations to fix the behavior of the partitioner.
#414 opened Mar 11, 2025 by copybara-service bot Loading…
@tomnatan30
test remove bazelrc line
#412 opened Mar 7, 2025 by copybara-service bot Loading…
@tomnatan30
Don't allow sideways propagation between operands/results when propagation direction is forward/backwards.
#410 opened Mar 6, 2025 by copybara-service bot Loading…
@tomnatan30
#sdy Add extra export_test.py tests for using different meshes..
#405 opened Mar 3, 2025 by copybara-service bot Loading…
@bartchr808
#sdy add bytecode serialization utility
#394 opened Feb 25, 2025 by copybara-service bot Loading…
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…
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
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! Add no:assignee to see everything that’s not assigned.