RuntimeError: cuFFT only supports dimensions whose sizes are powers of two when computing in half precision, but got a signal size of[48, 262144]

[wakakaFF]

你好， 我在调试模型的时候可以跑通，但是在训练模型的时候就出现了这个错误，请问这是什么原因导致的呢？

[isbrycee]

您好，我查了下是数据类型的问题。fft 不支持半精度计算。

[wakakaFF]

是的，感谢您的回复。我在mamba_simple_fft.py里加上这行代码后，已经可以运行了。