pytorch
2798f38c - Added checks for dtype and device of OpInfo's sample_inputs (#54949)

Commit
3 years ago
Added checks for dtype and device of OpInfo's sample_inputs (#54949) Summary: Currently, it's not tested whether `op.sample_inputs` actually used the provided dtype and device arguments. This PR fixes that introducing asserts in `test_supported_dtypes`. This will help to detect incorrectly generated inputs in the future. Pull Request resolved: https://github.com/pytorch/pytorch/pull/54949 Reviewed By: H-Huang Differential Revision: D27435952 Pulled By: mruberry fbshipit-source-id: 8465c459b9b0c007411a9a74340bc2755519624a
Author
Parents
Loading