DAKER 학습 게시 요약입니다. CUDA SM80 이상에서 torch.fft·torch.stft를 bfloat16으로 돌리면 지원 크기는 cuFFT 네이티브로 갑니다

핵심

· CUDA에서 torch.fft·torch.stft가 bfloat16 입력을 받습니다.

· 네이티브 cuFFT bf16은 SM80+·2의 거듭제곱 길이에서 열립니다.

· 비지원 길이는 float32로 승격되니 길이 정책부터 고정해 보십시오.

자세히 보기: https://daker.ai/community/post-mufwpy2t-581d5941