xla
Support SegmentID when doing data prallel SPMD
#8425
Merged
Overview
Commits
3
Changes
View On GitHub

Support SegmentID when doing data prallel SPMD #8425

JackCaoG merged 3 commits into master from JackCaoG/enable_segmentid_spmd_data_parallel
JackCaoG
JackCaoG Support SegmentID when doing data prallel SPMD
0453dd1e
JackCaoG lint
5e2cb30e
JackCaoG JackCaoG added tpuci
qihqi
qihqi approved these changes on 2024-11-27
JackCaoG rename segment_id for pallas kernel to make code less confusing
c4db5985
JackCaoG JackCaoG force pushed from 165da7b0 to c4db5985 1 year ago
JackCaoG JackCaoG marked this pull request as ready for review 1 year ago
JackCaoG JackCaoG merged 1c91219f into master 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum