xla
Add ShardingSpec to XLATensor when it is created with a PJRTShardedData
#5489

Merged

Add ShardingSpec to XLATensor when it is created with a PJRTShardedData #5489

JackCaoG merged 2 commits into master from JackCaoG/tensor_with_xla_data_sharding

Add ShardingSpec to XLATensor when it is created with a PJRTShardedData

d7b80a4e

JackCaoG added distributed

add test

a6626ea6

JackCaoG requested a review from

yeounoh 2 years ago

JackCaoG requested a review from

alanwaketan 2 years ago

JackCaoG requested a review from

jonb377 2 years ago

JackCaoG added xrt cherry pick

JackCaoG removed xrt cherry pick

alanwaketan commented on 2023-08-24

alanwaketan approved these changes on 2023-08-24

JackCaoG merged 6270cbae into master 2 years ago

yeounoh approved these changes on 2023-08-24

jonb377 commented on 2023-08-24

Reviewers

alanwaketan

yeounoh

jonb377

Assignees

No one assigned

Labels

distributed

Milestone

No milestone