Как вернуть массив (несколько значений) из функции устройства cuda? - PullRequest
0 голосов
/ 11 марта 2019

Я использую Numpy в Python, и у меня возникают проблемы с возвратом массива из функции устройства cuda. Я не могу разбить вычисление внутри функции устройства, и результатом для вычисления будет массив (с несколькими значениями). Любое предложение по возврату нескольких значений из функции устройства cuda было бы очень полезно.

1 Ответ

0 голосов
/ 12 марта 2019

Несмотря на то, что вы не знаете, с какой точки входа вы на самом деле заставляете GPU делать что-то, вы делаете в C ++ передачу указателя на объект в ядро, в которое затем ядро ​​записывает свои значения результата. Если у вас есть разные типы данных в качестве результатов, вы также можете определить свою собственную структуру / класс, который может содержать эти типы. Удачи!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...