xla
Enable `padded` option for even tensor sharding
#4156

Merged

Enable `padded` option for even tensor sharding #4156

yeounoh merged 2 commits into master from spmd_conv_debugging

[SPMD] count availalbe devices instead of xrt_world_size()

3d2aa97b

yeounoh added distributed

yeounoh requested a review from

jonb377 3 years ago

yeounoh requested a review from

JackCaoG 3 years ago

yeounoh requested a review from

steventk-g 3 years ago

yeounoh assigned

yeounoh 3 years ago

yeounoh force pushed from bededbc0 to e4bced3e 3 years ago

JackCaoG commented on 2022-11-04

yeounoh force pushed from e4bced3e to b71462ba 3 years ago

yeounoh force pushed from b71462ba to 84b4a0c4 3 years ago

JackCaoG approved these changes on 2022-11-04

jonb377 approved these changes on 2022-11-04

[SPMD] add `bool padded` to ShardingUtil::ShardTensor

90506e02

yeounoh force pushed from 84b4a0c4 to 90506e02 3 years ago

yeounoh merged 5ff192ea into master 3 years ago

Reviewers

jonb377

JackCaoG

steventk-g

Assignees

yeounoh

Labels

distributed

Milestone

No milestone