PR #5018 [SPMD] Output param sharding

yeounoh added distributed

yeounoh requested a review from

alanwaketan 2 years ago

yeounoh requested a review from

jonb377 2 years ago

yeounoh requested a review from

JackCaoG 2 years ago

yeounoh assigned

yeounoh 2 years ago

yeounoh marked this pull request as draft 2 years ago

JackCaoG commented on 2023-05-16

yeounoh force pushed from 5ffe5ed7 to 7c3bb93c 2 years ago

JackCaoG commented on 2023-05-16

yeounoh force pushed from 7c3bb93c to 4bd5e344 2 years ago

JackCaoG commented on 2023-05-16

jonb377 commented on 2023-05-16

JackCaoG commented on 2023-05-16

yeounoh force pushed from 4bd5e344 to 552079ce 2 years ago

yeounoh force pushed from 552079ce to 1b197fe2 2 years ago

yeounoh force pushed from 1b197fe2 to 9a4be89c 2 years ago

alanwaketan commented on 2023-05-16

yeounoh force pushed from 9a4be89c to b3dc8fc0 2 years ago

yeounoh force pushed from b3dc8fc0 to 7f7a1e32 2 years ago

yeounoh force pushed from 7f7a1e32 to ef68bc73 2 years ago

yeounoh force pushed from ef68bc73 to a308885e 2 years ago

yeounoh force pushed from a308885e to f5bc6c85 2 years ago

yeounoh force pushed from f5bc6c85 to 81d709c4 2 years ago

yeounoh force pushed from 81d709c4 to c47f86ba 2 years ago

yeounoh force pushed from c47f86ba to 0fb23390 2 years ago

yeounoh marked this pull request as ready for review 2 years ago

yeounoh force pushed from 0fb23390 to 97cae931 2 years ago

yeounoh force pushed from 97cae931 to 9804e0bc 2 years ago

yeounoh force pushed from 9804e0bc to 08d7c54c 2 years ago

jonb377 approved these changes on 2023-05-17

alanwaketan approved these changes on 2023-05-17

yeounoh force pushed from 08d7c54c to 640c62c8 2 years ago

JackCaoG commented on 2023-05-17

yeounoh force pushed from 640c62c8 to f7be1128 2 years ago

yeounoh force pushed from f7be1128 to d139cffa 2 years ago

JackCaoG commented on 2023-05-18

yeounoh force pushed from 641e4dd3 to 9df3106d 2 years ago

yeounoh force pushed from 9df3106d to b52e4d96 2 years ago

yeounoh force pushed from b52e4d96 to e9032b17 2 years ago

yeounoh force pushed from a33e6039 to 6bfc15d4 2 years ago

yeounoh force pushed from 6bfc15d4 to d8dca444 2 years ago

yeounoh force pushed from d8dca444 to 0223757e 2 years ago

yeounoh force pushed from 0223757e to 043481b9 2 years ago

yeounoh force pushed from 043481b9 to e7ddc0c8 2 years ago

yeounoh force pushed from e7ddc0c8 to b931311a 2 years ago

yeounoh force pushed from b931311a to 70e4729e 2 years ago

yeounoh force pushed from 70e4729e to 574b9b85 2 years ago

yeounoh force pushed from 574b9b85 to b574a788 2 years ago

yeounoh force pushed from b574a788 to 62354ffe 2 years ago

yeounoh force pushed from 62354ffe to 6646d631 2 years ago

yeounoh force pushed from 6646d631 to 8b3b19c5 2 years ago

yeounoh force pushed from 8b3b19c5 to 53dd180e 2 years ago

yeounoh force pushed from 53dd180e to b78a89f0 2 years ago

yeounoh force pushed from b78a89f0 to aa1ad458 2 years ago

yeounoh force pushed from aa1ad458 to 754d4fe6 2 years ago

yeounoh force pushed from 754d4fe6 to f4fd8972 2 years ago

yeounoh force pushed from f4fd8972 to 3284e6f6 2 years ago

yeounoh force pushed from 3284e6f6 to acb967f2 2 years ago

yeounoh force pushed from acb967f2 to 3080bb3d 2 years ago

yeounoh force pushed from 3080bb3d to 2bb3bd8c 2 years ago

[SPMD] Allow non-replicated outputs in computation results.

6c3d968c

Remove debugging probes

2965e092

Refactoring & linting

67b90569

Add ReplicateShardedData counter

bcb4173b

Add sharding propagation test

667e75c9

yeounoh force pushed from 263cdbc5 to 71703def 2 years ago

yeounoh force pushed from 71703def to 485442f3 2 years ago

yeounoh force pushed from ba853859 to a166a359 2 years ago

yeounoh force pushed from a166a359 to 80e19d3e 2 years ago

yeounoh force pushed from 80e19d3e to 7b451fe9 2 years ago

yeounoh force pushed from 7b451fe9 to e410bb73 2 years ago

yeounoh force pushed from e410bb73 to 9d409c36 2 years ago

linter

1c9cb409

yeounoh force pushed from 9d409c36 to 1c9cb409 2 years ago

Disable teset_dynamo_spmd

25e80af1

yeounoh commented on 2023-05-20

yeounoh merged b52b4a4c into master 2 years ago

xla
[SPMD] Output param sharding
#5018

Merged

[SPMD] Output param sharding #5018

xla [SPMD] Output param sharding #5018 Merged

[SPMD] Output param sharding #5018

xla
[SPMD] Output param sharding
#5018

Merged