xla
[Distributed] Hide token from Python for _xla_all_reduce path
#4913

Merged

[Distributed] Hide token from Python for _xla_all_reduce path #4913

alanwaketan merged 9 commits into master from alanwaketan/hide_token

alanwaketan assigned

alanwaketan 3 years ago

alanwaketan marked this pull request as draft 3 years ago

alanwaketan force pushed from 1592259a to 42c769f9 3 years ago

alanwaketan changed the base branch from master to alanwaketan/token 3 years ago

alanwaketan marked this pull request as ready for review 3 years ago

alanwaketan requested a review from

JackCaoG 3 years ago

alanwaketan force-pushed the alanwaketan/token branch from 39c00923 to 36361ed9 3 years ago

JackCaoG commented on 2023-04-21

JackCaoG approved these changes on 2023-04-21

Introduce ResetAllReduceToken

4e427948

introduce _set_all_reduce_token

097e10b4

Make token process local and thread safe

7dfc9425

Use token in C++ directly

4507a3cd

Remove token from _xla_all_reduce signatures

63a39071

Relocate token

a29188b2

Fix linters

2ba88484

Fix rebase error

cd97a3b2

Fix linters

ce26d1a7

alanwaketan force pushed from 42c769f9 to ce26d1a7 3 years ago

alanwaketan changed the base branch from alanwaketan/token to master 3 years ago

alanwaketan merged 4994597f into master 3 years ago

Reviewers

JackCaoG

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone