pytorch
fef0219f - [ROCM] Fix hipfft transform type error (#53411)

Commit
3 years ago
[ROCM] Fix hipfft transform type error (#53411) Summary: This PR enable some failing unit tests for fft in pytorch on ROCM. The reason these tests were failing was due to an error in how hipfft was executed for different transform types for float inputs causing a mismatch error when compared to baselines. We solved the problem by calling hipfft with the right config for each transformation type. There PR doesnot enable all fft tests. There are still other issues that need to be resolved before that can happen. Pull Request resolved: https://github.com/pytorch/pytorch/pull/53411 Reviewed By: albanD Differential Revision: D27008323 Pulled By: mruberry fbshipit-source-id: 649c65d0f12a889a426ec475f7d8fcc6f1d81bd3
Author
Parents
Loading