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

Jump to bottom

Improve separation of PyTorch dtypes and TVM dtypes in relay PyTorch frontend #5779

Closed
wants to merge 1 commit into from
Closed

Improve separation of PyTorch dtypes and TVM dtypes in relay PyTorch frontend #5779

wants to merge 1 commit into from

Commits on Jun 12, 2020

  1. Improve separation of PyTorch dtypes and TVM dtypes in relay PyTorch … 

    …frontend

Previously, we sometimes used type strings taken from PyTorch
(e.g. float) in TVM. Now we aim to convert PyTorch dtypes early
and only work with TVM dtypes afterwards.

Also fix arange/linspace type semantics.

A follow-up will then further improve type support (at which point
we will be able to convert e.g. double models).
    @t-vi
    t-vi committed Jun 12, 2020
    Configuration menu
    Copy the full SHA
    0c5601b View commit details
    Browse the repository at this point in the history