feat: abstraction of xla::OpSharding proto using wrapper class#9467

Open

kvshbg-aws wants to merge 13 commits into

pytorch:masterfrom

kvshbg-aws:kvshbg-aws/local-spmd-abstraction

Commits on Aug 20, 2025

feat: abstraction of xla::OpSharding proto using wrapper class

kvshbg-aws
authored and
kvshbg-aws
committed
fix for failing ci/cd tests

kvshbg-aws
authored and
kvshbg-aws
committed
small fix to remove unwanted var

kvshbg-aws
authored and
kvshbg-aws
committed
bug fix: set value of denormalized_tile_assignment when sharding.tile_assignment() is empty

kvshbg-aws
authored and
kvshbg-aws
committed
use tensors to get denormalized_tile_assignment directly instead of po_data

kvshbg-aws
authored and
kvshbg-aws
committed
use tensors instead of paramters_data to get denormalized_tile_assignment

kvshbg-aws
authored and
kvshbg-aws
committed
linter fix

kvshbg-aws
authored and
kvshbg-aws
committed
fix: use lowering_cntxt to save/get denormalized_tile_assignment
kvshbg-aws
committed
minor fixes; split tests in test_xla_sharding.cpp for readability
kvshbg-aws
committed

Commits on Aug 26, 2025

Merge branch 'pytorch:master' into kvshbg-aws/local-spmd-abstraction
kvshbg-aws
authored

Commits on Aug 28, 2025

Commits on Sep 3, 2025

Merge branch 'pytorch:master' into kvshbg-aws/local-spmd-abstraction
kvshbg-aws
authored