feat: abstraction of xla::OpSharding proto using wrapper class #9467
kvshbg-aws
wants to merge
13 commits into
pytorch:master
from
kvshbg-aws:kvshbg-aws/local-spmd-abstraction
feat: abstraction of xla::OpSharding proto using wrapper class
2f04535d
fix for failing ci/cd tests
23682f97
small fix to remove unwanted var
cd417f5a
bug fix: set value of denormalized_tile_assignment when sharding.tile…
5cc2dd2e
use tensors to get denormalized_tile_assignment directly instead of p…
4bb9b5a7
use tensors instead of paramters_data to get denormalized_tile_assign…
cbd9da2c
linter fix
a5aefa47
fix: use lowering_cntxt to save/get denormalized_tile_assignment
a8458ac8
minor fixes; split tests in test_xla_sharding.cpp for readability
4500461f
Merge branch 'pytorch:master' into kvshbg-aws/local-spmd-abstraction
67c0e169
Merge branch 'master' into kvshbg-aws/local-spmd-abstraction
982bd3e7
lint fix
3ad27346
Merge branch 'pytorch:master' into kvshbg-aws/local-spmd-abstraction
c5cf3297
qihqi
approved these changes
on 2025-09-04
pgmoka
approved these changes
on 2025-09-09
Login to write a write a comment.
Login via GitHub