Я работаю над проектом, который предполагает создание ядер CUDA в Python. Numba работает достаточно хорошо (что эти ребята достигли совершенно невероятно), а также PyCUDA.Вызов функции устройства CUDA CUDA из ядра Python
Моя проблема в том, что я хочу вызвать функцию устройства C из моего ядра, сгенерированного Python. Я не мог найти способ сделать это. Numba может вызывать модули CFFI, но только в коде CPU. В PyCUDA я могу добавить свои функции устройства C в SourceModule, но я не мог понять, как включить функции, которые уже существуют в другой библиотеке.
Есть ли способ сделать это?
Вы спрашиваете о pycuda или numba pro? – talonmies
Ответ на любой из них был бы отличным. – zmbq
Я закончил с использованием PyCUDA и вызвал шаблонные функции C++, включив их. Это обошло проблему. – zmbq