xla icon indicating copy to clipboard operation
xla copied to clipboard

Set ShardingSpec standard, and generalize it

Open pgmoka opened this issue 6 months ago • 0 comments

As part of the "xla::OpSharding", I found two instances where it was actually being abstracted within PytorchXLA:

  • tensor_common.h: torch_xla::ShardingSpec
  • tensor.h/cpp: ShardingSpec

Both instances do basically the same thing, and partially what the plan for this bug is. We should look into:

  1. Merging both "ShardingSpec" definitions into one to decrease redundancy
  2. Generalizing it for the rest of xla::OpSharding.

I am planning on adding more to https://github.com/pytorch/xla/issues/9181, and make it more of an RFC which will help put everything together

pgmoka avatar Jun 11 '25 00:06 pgmoka