CUDA全局内存

CUDA global memory

本文关键字：内存 全局 CUDA      更新时间：2023-10-16

这是我的代码

 #include "stdafx.h"
  #include <iostream>
   using namespace std;
  #define n 10
  __device__ int glMem[n];
  __global__ void initVals()
  {
for(int i=0;i<n;i++)
    glMem[i] = 0;
 }
 __global__ void test(int *out)
{
for(int i=0;i<n;i++)
    out[i] = 10;
}
int main()
{
const size_t sz = size_t(n)*sizeof(int);
initVals<<<1,1>>>();
int *devMem;
cudaMalloc((void **)&devMem, sz);
test<<<1, 1>>>(devMem);
int *hoMem=new int[n];
cudaMemcpy(hoMem, devMem,sz, cudaMemcpyDeviceToHost);
//print
for(int i=0;i<n;i++)
    cout<<hoMem[i]<<endl;
return 0;
}

在这个代码中，我定义

glMem

如果我之前不知道尺寸，我能定义吗？？例如，我需要这样定义。

__device__ int *glMem;

它不起作用。请提供一些代码示例。。

在这种情况下，您需要将内存分配到设备中。

// size of data
unsigned int size_of_glMem = n * sizeof(int);
// allocate device memory for result
int* glMem = NULL;
cudaMalloc( (void**) &glMem, size_of_glMem );

希望得到帮助。

相关文章：