xla
Expose replica_id in XLAShard
#5654
Merged

Expose replica_id in XLAShard #5654

jonb377 merged 4 commits into master from jonbolin-expose-rank
jonb377
jonb377 jonb377 added distributed
jonb377 jonb377 requested a review from yeounoh yeounoh 2 years ago
jonb377 jonb377 assigned jonb377 jonb377 2 years ago
jonb377 jonb377 force pushed from c9dcc233 to 2cc60993 2 years ago
jonb377 Expose shard rank in XLAShard
c32f38bf
jonb377 Improve tests and fix dist chkpt
4125cda3
jonb377 jonb377 force pushed from 2cc60993 to 4125cda3 1 year ago
yeounoh
yeounoh
yeounoh requested changes on 2023-10-02
jonb377 Change 'rank' to 'replica_id'
3d42b5af
yeounoh
yeounoh commented on 2023-10-02
jonb377 jonb377 changed the title Expose shard rank in XLAShard Expose replica_id in XLAShard 1 year ago
yeounoh
yeounoh approved these changes on 2023-10-02
jonb377 Improve documentation
92f63086
jonb377 jonb377 merged fba326e8 into master 1 year ago
jonb377 jonb377 deleted the jonbolin-expose-rank branch 1 year ago
jonb377

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone