Skip to content

Reshaping Fake meta tensor results in real meta tensor #136375

@tugsbayasgalan

Description

@tugsbayasgalan
with enable_python_dispatcher():
        tensor = torch.ones(4, 4, device="meta")
        f = tensor.reshape(2, 8)
        print(f) # prints tensor(..., device='meta', size=(2, 8))

cc @ezyang @chauhang @penguinwu @zou3519 @bdhirsh

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: pt2-dispatcherPT2 dispatcher-related issues (e.g., aotdispatch, functionalization, faketensor, custom-op,oncall: pt2

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions