xla
Introduce `torch_xla.runtime.use_spmd()`
#5474
Merged
Overview
Commits
7
Changes
View On GitHub

Introduce `torch_xla.runtime.use_spmd()` #5474

yeounoh merged 7 commits into master from spmd_new_api
yeounoh
yeounoh yeounoh added distributed
yeounoh yeounoh assigned yeounoh yeounoh 2 years ago
yeounoh yeounoh marked this pull request as draft 2 years ago
yeounoh yeounoh added DO_NOT_MERGE_YET
yeounoh
yeounoh yeounoh force pushed from e7cea87c to 27e8a8d1 2 years ago
yeounoh yeounoh force pushed from 27e8a8d1 to dbcca6f1 2 years ago
yeounoh yeounoh force pushed from dbcca6f1 to 64e3f4ea 2 years ago
yeounoh yeounoh requested a review from miladm miladm 2 years ago
yeounoh yeounoh requested a review from alanwaketan alanwaketan 2 years ago
yeounoh yeounoh requested a review from jonb377 jonb377 2 years ago
yeounoh yeounoh requested a review from JackCaoG JackCaoG 2 years ago
yeounoh yeounoh requested a review from khatwanimohit khatwanimohit 2 years ago
yeounoh yeounoh removed DO_NOT_MERGE_YET
yeounoh
yeounoh yeounoh marked this pull request as ready for review 2 years ago
alanwaketan
alanwaketan commented on 2023-08-21
yeounoh yeounoh force pushed from 35c286aa to 14155c36 2 years ago
yeounoh yeounoh force pushed from 14155c36 to eac936d9 2 years ago
yeounoh Introduce torch_xla.runtime.set_use_spmd()
0cbf71f4
yeounoh linter fix
5c3d0716
yeounoh Declare lock_spmd_config in device.cpp
f5be2203
yeounoh yeounoh force pushed from eac936d9 to d3a1b137 2 years ago
yeounoh yeounoh changed the title Introduce `torch_xla.runtime.set_use_spmd()` Introduce `torch_xla.runtime.use_spmd()` 2 years ago
yeounoh yeounoh force pushed from d3a1b137 to 64664e8a 2 years ago
yeounoh yeounoh force pushed from 64664e8a to 5ff89246 2 years ago
yeounoh yeounoh force pushed from 5ff89246 to d1d61ca5 2 years ago
yeounoh yeounoh force pushed from d1d61ca5 to e81f49a1 2 years ago
yeounoh Rename set_use_spmd to use_spmd
2f9f3622
yeounoh yeounoh force pushed from e81f49a1 to 2f9f3622 2 years ago
JackCaoG
JackCaoG commented on 2023-08-22
yeounoh Add a getter, is_spmd
8afb28b5
yeounoh yeounoh force pushed from 5e96da64 to 8afb28b5 2 years ago
yeounoh yeounoh force pushed from 6c1b8ffc to 611432e1 2 years ago
yeounoh yeounoh force pushed from 611432e1 to eabe6cf5 2 years ago
yeounoh add test_runtime_spmd_api
64b83f14
yeounoh yeounoh force pushed from eabe6cf5 to 64b83f14 2 years ago
JackCaoG
JackCaoG commented on 2023-08-22
JackCaoG
JackCaoG approved these changes on 2023-08-22
alanwaketan
alanwaketan approved these changes on 2023-08-22
jonb377
jonb377 approved these changes on 2023-08-22
yeounoh change lock_spmd_config to spmd_config_is_locked
e0b748a8
yeounoh yeounoh merged 462cb0f4 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