NVIDIA CUDA 计算统一设备架构

中
的
个
、cuMemAllocPitch() 分
线 性 存 储 器 。 放
节
、CUDA
数
int offset = 0;
int i;
cuParamSeti(cuFunction, offset, i);
offset += sizeof(i);
float f;
cuParamSetf(cuFunction, offset, f);
offset += sizeof(f);
char data[32];
cuParamSetv(cuFunction, offset, (void*)data, sizeof(data));
offset += sizeof(data);
cuParamSetSize(cuFunction, offset);
cuFuncSetSharedSize(cuFunction, numElements * sizeof(float));
cuLaunchGrid(cuFunction, gridWidth, gridHeight);
4.5.3.6 存 储 器 管 理
或
分
释
CUdeviceptr devPtr;
cuMemAlloc(&devPtr, 256 * sizeof(float));
:
// host code
CUdeviceptr devPtr;
int pitch;
cuMemAllocPitch(&devPtr, &pitch,
因
浮 点 元 素 的 数 组
width * sizeof(float), height, 4);
cuModuleGetFunction(&cuFunction, cuModule, “myKernel”);
cuFuncSetBlockShape(cuFunction, 512, 1, 1);
cuParamSeti(cuFunction, 0, devPtr);
cuParamSetSize(cuFunction, sizeof(devPtr));
cuLaunchGrid(cuFunction, 100, 1);
// device code
__global__ void myKernel(float* devPtr)
{
for (int r = 0; r < height; ++r) {
float* row = (float*)((char*)devPtr + r * pitch);
for (int c = 0; c < width; ++c) {
float element = row[c];
}
}
}
CUDA 数
创
销
cuMemAlloc() 用 使 可
cuMemAllocPitch()
cuMemFree() 性 存 储 器 , 并 使 用 线 配
256 的 代 码 示 例 在 线 性 存 储 器 中 分 配 了 一 个 包 含 面 下
建 议 在 分 配 二 维 数 组 时 使
用 cudaMallocPitch(),
5.1.2.1 能 确 保 合 理 填 充 已 分 配 的 存 储 器 , 满 足 第 它 为
绍 的 对 齐 要 求 , 从 而 确 保 访 问 行 地 址 或 执 行 二 维 数 组 与 设 备 存 储 器 的 其 他 区 域 之 间 的 复 制 ( 使 用
cudaMemcpy2D()) 时 获 得 最 优 性 能 。 所 返 回 的 间 距 ( 或 步 幅 ) 必 须 用 于 访 问 数 组 元 素 。 以 下 代 码 示 例 将 分 介
widthxheight 个 一 配
维 浮 点 值 数 组 , 并 显 示 如 何 在 设 备 代 码 中 循 环 遍 历 数 组 元 素 : 二
。 毁
的
数
位
浮 点 组 件 : 的
CUDA_ARRAY_DESCRIPTOR desc;
desc.Format = CU_AD_FORMAT_FLOAT;
desc.NumChannels = 1;
desc.Width = width;
desc.Height = height;
CUarray cuArray;
cuArrayCreate(&cuArray, &desc);
分
组
32 CUDA 编
cuArrayCreate() 使 用 是 组
cuArrayDestroy() , 使 用 的 建
配 的 线 性 存 储 器
widthxheight 代 码 示 例 分 配 了 一 个 下 以
CUDA
32 包 含 一 个 , 组
CUDA_MEMCPY2D copyParam;
memset(&copyParam, 0, sizeof(copyParam));
存 储 器 : 性 线 的 配
copyParam.dstMemoryType = CU_MEMORYTYPE_ARRAY;
copyParam.dstArray = cuArray;
copyParam.srcMemoryType = CU_MEMORYTYPE_DEVICE;
cuMemAlloc() 手 册 列 举 了 用 于 在 考 参
2.0 南 , 版 本 指 程
CUDA 之 间 复 制 存 储 器 的 所 有 函 数 。 下 面 的 代 码 示 例 将 二 维 数 组 复 制 到 之 前 代 码 示 例 中 分 配 的 组 数