使用numba将函数传递给CUDA块

Question

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

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

提前感谢您的帮助

Answer 1

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