Вычисление среднего для 2D-массива CUDA

Question

Мне нужно вычислить среднее значение 2D-массива с использованием CUDA, но я не знаю, как это сделать. Я начал с сокращения столбцов после этого, я сделаю сумму результирующего массива, и на последнем шаге я вычислим среднее значение.

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

Answer 1

Если это простое среднее арифметическое всех элементов в 2D массива вы можете использовать тяги:

int* data; 
int num; 
get_data_from_library(&data, &num); 

thrust::device_vector<int> iVec(data, data+num); 

// transfer to device and compute sum 
int sum = thrust::reduce(iVec.begin(), iVec.end(), 0, thrust::plus<int>()); 
double mean = sum/(double)num; 

Если вы хотите, чтобы написать свое собственное ядро ​​- иметь в виду, что 2D массив по существу 1D массив разделен на рядовые размеры и пример с параллельным сокращением SDK: Whitepaper