CUDA编程12 - 使用OpenMP控制多个GPU示例

一：概述

        本文以 cuda_samples 中的 cudaOpenMP 为例，介绍下如何在多GPU系统上使用CPU线程（OpenMP）同时控制多个GPU，处理一段数组数据。

        这个例子做了以下事情： 一个是将CPU线程关联到GPU，每个CPU线程控制一个GPU，处理自己那部分数据。二是数据划分，数组被均分到多个GPU上并行处理。三是核函数（kernel）执行，每个GPU执行简单的加分核函数。四是检查结果，计算完成后回传CPU，验证正确性。 

二：示例

        1. 核函数 

#include <helper_cuda.h>  // NVIDIA CUDA 工具函数
#include <omp.h>          // OpenMP 多线程
#include <stdio.h>        // C 标准 IO 函数using namespace std;// CUDA 核函数：每个数组元素加上常量 b
__global__ void kernelAddConstant(int *g_a, const int b) {int idx = blockIdx.x * blockDim.x + threadIdx.x;g_a[idx] += b;
}// 检查结果是否正确：每个元素是否等于 i + b
int correctResult(int *data, const int n,