Skip to content

Pull requests: openxla/shardy

New pull request New
22 Open 371 Closed
22 Open 371 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 resize the sharding group map to the max of it's current size or the new group ID index.
#398 opened Feb 27, 2025 by copybara-service bot Loading…
@bartchr808
Use local spawn strategy for faster builds
#397 opened Feb 26, 2025 by apivovarov Loading…
1
#sdy Add test for the cloned round_trip_import pipeline in Shardy github.
#396 opened Feb 26, 2025 by copybara-service bot Loading…
@bixia1
#sdy Duplicate addSdyRoundTripImportPipeline and
#395 opened Feb 25, 2025 by copybara-service bot Loading…
@bixia1
#sdy add bytecode serialization utility
#394 opened Feb 25, 2025 by copybara-service bot Loading…
#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! Adding no:label will show everything without a label.