[PyTorch] Unable to import a simple torch.nn.Linear to Relay

Troubleshooting
4

Thanks @tqchen and @masahi. I tried the script on the following PyTorch versions:

torch 1.10.0.dev20210629+cu111 → throws error (original post)

torch 1.9.0 → throws same error as above

torch 1.8.1 → works

torch 1.7.0 → works (@masahi’s experiment)

I have a few questions:

  1. Is PyTorch 1.8.1 the currently supported version by TVM? Curious is there a CI job that test the integration I can take a look and learn more about :slight_smile:
  2. I would like to help add PyTorch 1.9.0 conversion support at least for torch.nn.Linear, curious where should I start? (I’ve been doing PyTorch internal development for several years, so any quick code pointers will be sufficient for me to get started :slight_smile: )

Thanks!