masahi edited a comment on issue #6268:
URL: https://github.com/apache/incubator-tvm/issues/6268#issuecomment-690044821


   oh I've just tried the above script (reproduced below) on torch 1.6, and it 
seems they fixed it:
   
   ```
   import torch
   import numpy as np
   
   lhs = torch.zeros((), dtype=torch.int64)
   rhs = 5 * np.ones([]).astype("int64")  # what prim::NumToTensor(5) above 
converts to in our frontend
   
   print(torch.result_type(lhs, 5))
   print(torch.result_type(lhs, rhs))
   ```
   Output
   ```
   torch.int64
   torch.int64
   ```
   
   @nolanliou so for both `torch.result_type` (fixed in 1.6) and integer div 
(no longer supported), I suggest updating your code to avoid integer div and 
upgrade to PyTorch 1.6


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to