M1 mps fft 사용시 Error

m1 mac으로 GitHub - neonbjb/tortoise-tts: A multi-voice TTS system trained with an emphasis on quality

voice gen을 test 중 오류를 발견하게 되었습니다.

mps는 복소수 난수 생성 에러가 아직 고쳐지지 않았습니다.

따라서 fft, mel 변환, 퓨리에 변환을 필요로하는 작업을 mps로 작업하시는 것을 추천드리지 않습니다. (ex, ASR, STT, TTS)
General MPS op coverage tracking issue · Issue #77764 · pytorch/pytorch · GitHub - mps 이슈 목록입니다.

pytorch 2.0.0
python 3.9.7

import torch
x = torch.randn(1, 16000, device="mps")
y = torch.fft.rfft(x)
y_abs = y.abs()

NotImplementedError: The operator 'aten::_fft_r2c' is not currently implemented for the MPS device. If you want this op to be added in priority during the prototype phase of this feature, please comment on https://github.com/pytorch/pytorch/issues/77764. As a temporary fix, you can set the environment variable `PYTORCH_ENABLE_MPS_FALLBACK=1` to use the CPU as a fallback for this op. WARNING: this will be slower than running natively on MPS.
1개의 좋아요

와우, 고생 많으셨습니다 :sob:

mps는 언제쯤 쓸만해질지...

공유 감사합니다! :bowing_man:

2개의 좋아요