Skip to content

[Torchao Experimental] Create meta tensors directly in meta kernels #1936

@metascroy

Description

@metascroy

In meta kernels in torchao/experimental/ops, we do things like:

return torch::empty({num_out, k}).to("meta");

We should create meta tensors directly if possible.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions