CUDA SM80 이상에서 torch.fft·torch.stft를 bfloat16으로 돌리면, 지원 크기는 cuFFT 네이티브 경로로 처리됩니다. FFT 정밀도와 성능을 같이 맞출 때 먼저 확인할 포인트입니다.

핵심만 먼저 보면 다음과 같습니다.

• PyTorch 2.14는 CUDA에서 torch.fft 계열과 torch.stft에 bfloat16 입력을 받습니다

• 네이티브 cuFFT bfloat16 경로는 SM80 이상에서 지원 크기 기준으로 갑니다

• 학습·추론에서 FFT 정밀도와 성능을 같이 보고 싶을 때 참고하세요

원문 전체는 DAKER에서 확인할 수 있습니다.

https://daker.ai/community/cuda-sm80-bfloat16-torch-fft-stft-cufft-native-siz