从Python内核调用C ++ CUDA设备函数

时间:2016-09-18 07:38:37

标签: python cuda cython numba pycuda

我正在开发一个涉及在Python中创建CUDA内核的项目。 Numba工作得很好(这些人所取得的成就令人难以置信),PyCUDA也是如此。

我的问题是我想从Python生成的内核中调用C设备函数。我无法找到实现这一目标的方法。 Numba可以调用CFFI模块,但仅限于CPU代码。在PyCUDA中,我可以将我的C设备函数添加到SourceModule,但我无法弄清楚如何包含已存在于另一个库中的函数。

有没有办法实现这个目标?

1 个答案:

答案 0 :(得分:2)

据我所知,这两种语言都不可能。既不公开必要的工具链控件来进行单独的编译,也不公开API来进行设备代码的运行时链接。