CUDA __device__ типа struct

Question

Эксперты CUDA, если я определил в коде хоста новый тип:

struct float_3{
 float x;
 float y;
 float z;
};

и перенес некоторые данные этого типа на устройство, могу ли я создать __device__ вызовы этого новоготип, то есть:

__device__ float_3 foo(float_3 r,float b,int a){
}

Можем ли мы создать __device__ любого типа?Или просто int, float, dlouble, void и т. Д. И возможно ли вернуть указатель на __device__?т.е. __device__ float_3* foo(){}

Answer 1

Да, вы можете создать __device__ любого типа.Это просто спецификатор, который делает эту функцию компилируемой для запуска на устройстве и может вызываться из устройства.

И, кстати, CUDA имеет тип float3.Я никогда не использовал его, но если я правильно помню, он обеспечивает ту же функциональность, что и ваш float_3, а также поставляется с конструктором.

Answer 2

Можем ли мы создать __device__ любого типа?

Краткий ответ: да .Длинный ответ: да, если это пользовательский тип, такой как float_3, вы можете определить __device__ указатель переменной и выделить память в устройстве, используя cudaMalloc.

Можно ли вернуть указатель на устройство?

Да, вы можете это сделать.