xla icon indicating copy to clipboard operation
xla copied to clipboard
verified publisher icon pytorch

Leverage torch_xla::OpSharding in mark_sharding

Open pgmoka opened this issue 6 months ago • 0 comments

Currently get_op_sharding generates an xla::OpSharding. With the new abstraction of torch_xla::OpSharding, we will want to use it instead

pgmoka avatar Jun 13 '25 23:06 pgmoka