pytorch
ebd86915 - Clarifies compare_with_numpy behavior (#40064)

Commit
4 years ago
Clarifies compare_with_numpy behavior (#40064) Summary: Currently compare_with_numpy requires a device and dtype, but these arguments are ignored if a tensor is provided. This PR updates the function to only take device and dtype if a tensor-like object is given. This should prevent confusion that you could, for example, pass a CPU float tensor but provided a CUDA device and integer dtype. Several tests are updated to reflect this behavior. Pull Request resolved: https://github.com/pytorch/pytorch/pull/40064 Differential Revision: D22058072 Pulled By: mruberry fbshipit-source-id: b494bb759855977ce45b79ed3ffb0319a21c324c
Author
Mike Ruberry
Parents
Loading