Форум программистов [Powered by Invision Power Board]

Версия для печати темы
Нажмите сюда для просмотра этой темы в оригинальном формате

Форум программистов > C/C++: Общие вопросы > CUDA - суммирование чисел

Автор: Wisdom 19.4.2012, 20:12

Здравствуйте.
Пробую реализовать сложение двух массивов с помощью CUDA в Visual Studio 2010.
Тулкит поставил, драйвера стоят.
Файл Cuda.cu:

Код


#include <iostream>
#include <cuda_runtime.h>

__global__ void sum(float *A, float *B, float *C) 
{
    int n = blockDim.x * blockIdx.x + threadIdx.x;
    C[n] = A[n] + B[n];
}

void StartSum(float *A, float *B, float *C, int N)
{
    sum<<< N/64, 64 >>>(A, B, C) ;
}

Файл с инициализацией массивов и вызовом функции суммирования:

Код


#include <windows.h>
#include <cuda.h>
#include <cuda_runtime.h>
#include <cuda_runtime_api.h>
#include <iostream>

#define N 5

void StartSum(float *A, float *B, float *C, int n);

int main() 
{
    float a[N] = {1,2,3,4,5}, b[N]={-2,-4,5,7,1}, c[N] = {0,0,0,0,0};
    cudaError_t err;
    float *dev_a , *dev_b , *dev_c ;
    cudaSetDevice(0);
    cudaMalloc((void**)&dev_a , sizeof (float)*N);
    cudaMalloc((void**)&dev_b , sizeof (float)*N);
    cudaMalloc((void**)&dev_c , sizeof (float)*N);

    err = cudaMemcpy(dev_a, a, sizeof(float)*N, cudaMemcpyHostToDevice);
    err = cudaMemcpy(dev_b, b, sizeof(float)*N, cudaMemcpyHostToDevice);
    err = cudaMemcpy(dev_c, c, sizeof(float)*N, cudaMemcpyHostToDevice);

    StartSum(dev_a, dev_b, dev_c, N);
    err = cudaMemcpy(c, dev_c, sizeof(float), cudaMemcpyDeviceToHost);

    for (int i = 0; i<N; i++)
        std::cout<<c[i]<<" ";

    std::cout<<std::endl;

    system("PAUSE");
}

Вопрос в следующем: почему в массиве c всегда нули? По идее, там должна была аккумулироваться сумма..
Бьюсь уже часа два, кто чем может, помогите.
Спасибо.

Автор: Wisdom 20.4.2012, 06:23

Друзья, неужели никто не может помочь ?
Хотя бы намекните, в какую сторону копать..

Автор: Wisdom 20.4.2012, 16:06

Решил проблему.
N/64 == 0 (целая часть). А это значение блоков ГПУ, которое должно быть отличным от нуля.