Phenum2 945(3.0Ghz) / GTX650
cpu Time : 0.131000
gpu Time : 0.001000
일단.. 싱글 코어로 돌려서 130msec 정도
GPU로 돌려서 동일 연산이 1msec 미만으로 걸린다.. ㄷㄷㄷ
일단 32*32*128*128*4 로
총 1024개의 쓰레드를 동시에 돌리고
16,384 개의 블럭으로 연산을 한다.
67,108,864 byte 대략 64MB + 64MB = 64MB 용량의 연산을 하는 소스이다.
#include < stdio.h >
#include < stdlib.h >
#include < time.h >
#include "cuda_runtime.h"
#define BLOCK_WID 128
#define THREAD_WID 32
#define ARRAY_SIZE (THREAD_WID * THREAD_WID * BLOCK_WID * BLOCK_WID)
__global__ void kernel_test(int *a, int *b, int *c)
{
int idx = threadIdx.x +blockIdx.x * blockDim.x + (gridDim.x * blockDim.x) * (blockIdx.y * blockDim.y + threadIdx.y);
c[idx] = a[idx] + b[idx];
}
void main()
{
clock_t start_time, end_time;
int *a, *b, *c, *res;
int *dev_a,*dev_b,*dev_c;
unsigned int idx = 0;
dim3 block(BLOCK_WID,BLOCK_WID);
dim3 thread(THREAD_WID,THREAD_WID);
a = (int *)malloc(ARRAY_SIZE * sizeof(int));
b = (int *)malloc(ARRAY_SIZE * sizeof(int));
c = (int *)malloc(ARRAY_SIZE * sizeof(int));
res = (int *)malloc(ARRAY_SIZE * sizeof(int));
// initialize
srand (time(NULL));
for(idx = 0;idx < ARRAY_SIZE ; idx++)
{
a[idx] = rand() & 0xFFFF;
b[idx] = rand() & 0xFFFF;
c[idx] = 0;
}
start_time = clock();
for(idx = 0;idx < ARRAY_SIZE ; idx++)
{
res[idx] = a[idx] + b[idx];
}
end_time = clock();
printf("cpu Time : %f\n", ((double)(end_time-start_time)) / CLOCKS_PER_SEC);
cudaMalloc(&dev_a, ARRAY_SIZE * sizeof(int));
cudaMalloc(&dev_b, ARRAY_SIZE * sizeof(int));
cudaMalloc(&dev_c, ARRAY_SIZE * sizeof(int));
cudaMemcpy(dev_a, a, ARRAY_SIZE * sizeof(int), cudaMemcpyHostToDevice);
cudaMemcpy(dev_b, b, ARRAY_SIZE * sizeof(int), cudaMemcpyHostToDevice);
start_time = clock();
kernel_test<<<block,thread>>>(dev_a,dev_b,dev_c);
end_time = clock();
printf("gpu Time : %f\n", ((double)(end_time-start_time)) / CLOCKS_PER_SEC);
cudaMemcpy(c, dev_c, ARRAY_SIZE * sizeof(int), cudaMemcpyDeviceToHost);
for(idx = 0;idx < ARRAY_SIZE ; idx++)
{
if(res[idx] != c[idx])
{
printf("%5d a:%5d b:%5d c:%5d", idx, a[idx], b[idx], c[idx]);
if(res[idx] != c[idx])
printf(" != ");
else printf(" == ");
printf("r:%5d\n",res[idx]);
break;
}
}
cudaFree(dev_a);
cudaFree(dev_b);
cudaFree(dev_c);
free(a);
free(b);
free(c);
free(res);
}
