2
我正在研究一個涉及在Python中創建CUDA內核的項目。 Numba工作得很好(這些人所取得的成就非常不可思議),PyCUDA也是如此。從Python內核調用C++ CUDA設備函數
我的問題是我想從我的Python生成的內核調用C設備函數。我找不到一種方法來實現這一點。 Numba可以調用CFFI模塊,但只能在CPU代碼中調用。在PyCUDA中,我可以將我的C設備函數添加到SourceModule,但我無法弄清楚如何包含另一個庫中已有的函數。
有沒有辦法做到這一點?
你問關於pycuda或numba專業? – talonmies
兩者的答案都很好。 – zmbq
我最終使用PyCUDA並通過包含它們來調用模板化的C++函數。這繞過了這個問題。 – zmbq