xla
[SPMD] Preserve parameter sharding with output data sharding
#4721
Merged
Overview
Commits
4
Changes
View On GitHub

[SPMD] Preserve parameter sharding with output data sharding #4721

yeounoh merged 4 commits into master from new_param_sharding_fix
yeounoh
yeounoh yeounoh added distributed
yeounoh yeounoh requested a review from alanwaketan alanwaketan 3 years ago
yeounoh yeounoh requested a review from JackCaoG JackCaoG 3 years ago
yeounoh yeounoh assigned yeounoh yeounoh 3 years ago
yeounoh yeounoh marked this pull request as draft 3 years ago
yeounoh yeounoh force pushed from 54c51f1b to 420d7013 3 years ago
JackCaoG
JackCaoG commented on 2023-03-04
JackCaoG
JackCaoG commented on 2023-03-04
JackCaoG
JackCaoG commented on 2023-03-04
JackCaoG
JackCaoG commented on 2023-03-04
JackCaoG
JackCaoG commented on 2023-03-04
yeounoh yeounoh force pushed from 420d7013 to 6e4a8355 3 years ago
yeounoh yeounoh force pushed from 6e4a8355 to 0a424ef8 3 years ago
yeounoh yeounoh force pushed from 0a424ef8 to 90f90bd3 3 years ago
yeounoh yeounoh force pushed from 90f90bd3 to b68cdfd8 3 years ago
yeounoh yeounoh force pushed from b68cdfd8 to 76a70ad6 3 years ago
yeounoh yeounoh force pushed from 76a70ad6 to 00912d58 3 years ago
yeounoh yeounoh force pushed from 00912d58 to 9aa19ad4 3 years ago
yeounoh yeounoh force pushed from 9aa19ad4 to 1d9d5abf 3 years ago
yeounoh yeounoh force pushed from 1d9d5abf to e38242f8 3 years ago
yeounoh yeounoh force pushed from e38242f8 to 74c69cfe 3 years ago
yeounoh yeounoh force pushed from 74c69cfe to 555a4747 3 years ago
yeounoh yeounoh force pushed from 555a4747 to aa6ae6e1 3 years ago
yeounoh yeounoh force pushed from aa6ae6e1 to 074435fc 3 years ago
yeounoh yeounoh force pushed from 074435fc to 3eac5e6c 3 years ago
yeounoh yeounoh force pushed from 3eac5e6c to f26b3056 3 years ago
yeounoh yeounoh marked this pull request as ready for review 3 years ago
yeounoh yeounoh force pushed from f26b3056 to 5c3e631b 2 years ago
yeounoh yeounoh force pushed from 5c3e631b to 0ddee730 2 years ago
JackCaoG
JackCaoG commented on 2023-03-04
JackCaoG
JackCaoG commented on 2023-03-07
JackCaoG
JackCaoG commented on 2023-03-07
JackCaoG
JackCaoG commented on 2023-03-07
yeounoh yeounoh force pushed from a90760e0 to e45ab940 2 years ago
JackCaoG
yeounoh yeounoh force pushed from e45ab940 to 5ba829fb 2 years ago
yeounoh [SPMD] Persist tensor sharding with XLA sharding propagation.
0004c249
yeounoh Add tests
e1a7b8a4
yeounoh Add more comments
4c1a7e63
yeounoh yeounoh force pushed from 5ba829fb to 26279e3a 2 years ago
yeounoh
yeounoh yeounoh requested a review from steventk-g steventk-g 2 years ago
yeounoh yeounoh requested a review from jonb377 jonb377 2 years ago
yeounoh yeounoh changed the title Preserve parameter sharding with output data sharding [SPMD] Preserve parameter sharding with output data sharding 2 years ago
JackCaoG
JackCaoG commented on 2023-03-07
yeounoh Refactoring
8d83ef48
yeounoh yeounoh force pushed from 26279e3a to 8d83ef48 2 years ago
JackCaoG
JackCaoG approved these changes on 2023-03-07
jonb377
jonb377 approved these changes on 2023-03-07
yeounoh yeounoh merged e2abcaf4 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
distributed
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum