torch.fft.irfft(torch.fft.rrft) is not identity on MPS #126649
Labels
high priority
module: fft
module: mps
Related to Apple Metal Performance Shaders framework
triage review
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
馃悰 Describe the bug
On CPU, call torch.fft.irfft(torch.fft.rrft) gives back the initial signal (as expected) it is not on MPS.
gives
This was tested on both 2.3.1 and 2.4.0-rc
Versions
Collecting environment information...
PyTorch version: 2.4.0.dev20240519
Is debug build: False
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A
OS: macOS 14.5 (arm64)
GCC version: Could not collect
Clang version: 15.0.0 (clang-1500.3.9.4)
CMake version: Could not collect
Libc version: N/A
Python version: 3.10.14 (main, May 18 2024, 21:07:07) [Clang 15.0.0 (clang-1500.3.9.4)] (64-bit runtime)
Python platform: macOS-14.5-arm64-arm-64bit
Is CUDA available: False
CUDA runtime version: No CUDA
CUDA_MODULE_LOADING set to: N/A
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True
CPU:
Apple M3
Versions of relevant libraries:
[pip3] numpy==1.26.4
[pip3] torch==2.4.0.dev20240519
[pip3] torchaudio==2.2.0.dev20240519
[pip3] torchvision==0.19.0.dev20240519
[conda] Could not collect
cc @ezyang @gchanan @zou3519 @kadeng @msaroufim @mruberry @peterbell10 @kulinseth @albanD @malfet @DenisVieriu97 @jhavukainen
The text was updated successfully, but these errors were encountered: