Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor based on recent torch tensor view support #1816

Closed
wants to merge 1 commit into from
Closed

Refactor based on recent torch tensor view support #1816

wants to merge 1 commit into from

Commits on Jun 9, 2023

  1. Refactor based on recent torch tensor view support (pytorch#1816) 

    Summary:
Pull Request resolved: pytorch#1816

torch.view recently added reinterpret_cast dtype support:

https://pytorch.org/docs/stable/generated/torch.Tensor.view.html

Previously we have to rely on numpy, which introduces the extra H2D/D2H overhead.

Reviewed By: sryap

Differential Revision: D46585543

fbshipit-source-id: a1d4e792e61c15c71bb36d47a574d2c052e0a02f
    @jianyuh @facebook-github-bot
    jianyuh authored and facebook-github-bot committed Jun 9, 2023
    Configuration menu
    Copy the full SHA
    3d249b3 View commit details
    Browse the repository at this point in the history