使用 numba 将函数传递给 CUDA 块 [关闭]

Question

我正在使用 numba 库在 python 中工作，想知道是否有解决方案来编写以前工作的并行版本。我有一个函数 f(X, S, F)，其中 X 和 S 是标量数组，F 是函数列表。

我几乎可以肯定，使用 numba（和一般的 cuda？）无法传递函数数组。什么是替代解决方案？如果有的话。

提前感谢您的帮助

Answer 1

Numba CUDA Python 实现目前不支持内核中的任何类型的函数指针或对象。所以你有野心去做的事情是不可能的。