GPGPU编程技术-从GLSL、CUDA到OpenCL 平方和算法第一版 书中程序158页

//GPGPU编程技术-从GLSL、CUDA到OpenCL  平方和算法第一版
//书中程序158页
#include <stdio.h>
#include <iostream>
#include <cuda_runtime.h> // For the CUDA runtime routines (prefixed with "cuda_")
#include <DEVICE_LAUNCH_PARAMETERS.h> //我在查询中找到的头文件 有可能有别的表达方式

#define DATA_SIZE 1048567   //4MB 的数据
using namespace std;
int anData[DATA_SIZE];

//生成随机数据量
void GenerateNumber(int *pnNumber, int nSize)
{
    for (int i = 0; i < nSize; i++)  pnNumber[i] = rand();
}

//全局函数计算平方和内核：在主机上调用，在设备上执行
__global__ static void sumofSquares(int *pnNum, int* pnResult)
{
    int nSum = 0;
    for (unsigned i = 0; i < DATA_SIZE; i++)
    {
        nSum += pnNum[i] * pnNum[i];
        *pnResult = nSum;
    }
}

void main()
{
    GenerateNumber(anData, DATA_SIZE); //生成随机数据量
    int *pnGpuData, *pnResult;
    cudaMalloc ((void**)&pnGpuData, sizeof(int) * DATA_SIZE);
    cudaMalloc ((void**)&pnResult , sizeof(int) );
    cudaMemcpy(pnGpuData, anData, sizeof(int)*DATA_SIZE, cudaMemcpyHostToDevice);

    //本版本是基础版本 在此基础上学习逐渐内存与内核并行优化
    //所以这里只用了一个内核
    sumofSquares <<< 1,1 >>> (pnGpuData, pnResult);

    int nSum = 0;
    cudaMemcpy(&nSum, pnResult, sizeof(int), cudaMemcpyDeviceToHost);
    cudaFree(pnGpuData);
    cudaFree(pnResult);
    printf("SuM = %d\n", nSum);
    system("pause");

    //return 0;

}

结果截屏