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

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.

iree-org/iree-torch,iree-org/iree-torch