xla
Support SegmentID when doing data prallel SPMD
#8425

Merged

Commits

Support SegmentID when doing data prallel SPMD

JackCaoG committed 1 year ago
lint

JackCaoG committed 1 year ago
rename segment_id for pallas kernel to make code less confusing

JackCaoG committed 1 year ago

Loading