8000 Leverage torch_xla::OpSharding in mark_sharding · Issue #9358 · pytorch/xla · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
Leverage torch_xla::OpSharding in mark_sharding #9358
New issue
New issue
Open
Open
Leverage torch_xla::OpSharding in mark_sharding#9358
Labels
distributedSPMD and other distributed things.enhancementNew feature or request
@pgmoka

Description

@pgmoka
pgmoka
opened on Jun 13, 2025

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    distributedSPMD and other distributed things.enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions
      0