将 pycuda 与 reikna fft 一起使用

问题描述

试图找到一种在 GPU 上计算 fft 的快速方法。对于其他计算，我使用 pycuda。 它对我有用 pyfft，但它已经过时了，我无法通过 pip 安装它。 我使用 reikna，但使用它，我无法使用 GPU 中的数据。 因此，我重新上传了它们。

fnc_coor(data_gpu,ta_gpu,fnci,Nri,Nai,s_rf_gpu,block=(BLOCK_SIZE,BLOCK_SIZE,1),grid=grid)
 
api = any_api()
thr = api.Thread.create()
fft = FFT(s_rf,axes=(1,))
fftc = fft.compile(thr)
s_rf_gpu = thr.to_device(s_rf_gpu.get())
fftc(s_rf_gpu,s_rf_gpu)

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）