如何解决将 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 举报，一经查实，本站将立刻删除。
