使用 numba 将函数传递给 CUDA 块

Passing functions to CUDA blocks with numba

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

我几乎可以肯定,numba(和一般的 cuda?)不可能传递函数数组。什么是替代解决方案?如果有的话。

在此先感谢您的帮助

Numba CUDA Python 实现目前不支持内核中的任何类型的函数指针或对象。所以你想做的事是做不到的。