Move input tensors to CPU device in `check_results`

iree-org / iree-torch

Torch Frontend for IREE

Apache License 2.0

25 stars 11 forks source link

Move input tensors to CPU device in `check_results` #38

Closed ramiro050 closed 2 years ago

ramiro050 commented 2 years ago

The op torch.allclose results in a runtime error if the two input tensors are not on the same device. This commit moves the input tensors of torch.allclose to the CPU device.