pytorch
51c28e4d - [ROCm] enable fft tests (#51581)

Commit
4 years ago
[ROCm] enable fft tests (#51581) Summary: This PR enable some failing unit tests for fft in pytorch on ROCM. The reason these tests were failing was due to hipfft clobbering inputs causing a mismatch in tests that were checking that applying ffts and their reverse would get you back to the input. We solve this by cloning the input using an existing flag on the ROCM platform. There PR doesnot enable all fft tests. There are other issues that need to be resolved before that can happen. Pull Request resolved: https://github.com/pytorch/pytorch/pull/51581 Reviewed By: ejguan Differential Revision: D26489344 Pulled By: seemethere fbshipit-source-id: 472fce8e514adcf91e7f46a686cbbe41e91235a9
Author
Parents
Loading