代码
#include<cuda_runtime.h>
#include<device_launch_parameters.h>
//#include<helper_cuda.h>
#include<cstdio>
__global__ void kernel(int* arr, int* sumv, const int size){
for(int index=blockDim.x*blockIdx.x+threadIdx.x;
index<size; index+=blockDim.x*gridDim.x)
{
arr[index] = index+1;
atomicAdd(sumv,arr[index]);
}
}
int main()
{
int size = 5;
int* arr = new int[size];
int* sumv = new int;
int* arrCuda;
int* sumvCuda;
cudaMalloc((void**)&arrCuda,size*sizeof(int));
cudaMalloc((void**)&sumvCuda,1*sizeof(int));
cudaMemcpy(arrCuda,arr,size*sizeof(int),cudaMemcpyHostToDevice);
cudaMemcpy(sumvCuda,sumv,1*sizeof(int),cudaMemcpyHostToDevice);
kernel<<<32,128>>>(arrCuda,sumvCuda,size);
cudaDeviceSynchronize();
cudaMemcpy(arr,arrCuda,size*sizeof(int),cudaMemcpyDeviceToHost);
cudaMemcpy(sumv,sumvCuda,1*sizeof(int),cudaMemcpyDeviceToHost);
for(int i=0;i<size;++i){
printf("arr[%d]=%d\n",i, arr[i]);
}
printf("%d\n",*sumv);
cudaFree(sumvCuda);
cudaFree(arrCuda);
delete sumv;
delete[] arr;
return 0;
}
CMakeLists.txt
CMAKE_MINIMUM_REQUIRED(VERSION 3.16)
PROJECT(MyProject LANGUAGES CUDA CXX)
include(CheckLanguage)
check_language(CUDA)
add_executable(out main.cu ${SRC})
结果