0

cudaMalloc 是否分配连续的内存块(即彼此相邻的物理字节)?

我有一段 CUDA 代码,它使用 32 个线程简单地将 128 个字节从全局设备内存复制到共享内存。我试图找到一种方法来保证可以在一个 128 个字节的内存事务中完成此传输。如果 cudaMalloc 分配连续的内存块,那么它可以很容易地完成。

以下是代码:

#include <iostream>

using namespace std;
#define SIZE 32 //SIZE of the array to store in shared memory                                                                                                                        
#define NUMTHREADS 32
__global__ void copy(uint* memPointer){

  extern __shared__ uint bits[];
  int tid = threadIdx.x;

  bits[tid] = memPointer[tid];

}

int main(){
  uint inputData[SIZE];
  uint* storedData;
  for(int  i=0;i<SIZE;i++){
    inputData[i] = i;
  }
  cudaError_t e1=cudaMalloc((void**) &storedData, sizeof(uint)*SIZE);
  if(e1 == cudaSuccess){
    cudaError_t e3= cudaMemcpy(storedData, inputData, sizeof(uint)*SIZE, cudaMemcpyHostToDevice);
      if(e3==cudaSuccess){
        copy<<<1,NUMTHREADS, SIZE*4>>>(storedData);
            cudaError_t e6 = cudaFree(storedData);
            if(e6==cudaSuccess){
            }
            else{
              cout << "Error freeing memory storedData" << e6 << endl;
            }
      }
      else{
        cout << "Failed to copy" << " " << e3 << endl;
      }

  }
  else{
    cout << "Failed to allocate memory" << " " << e1 << endl;

  }
  return 0;
}
4

1 回答 1

1

是的,cudaMalloc 分配连续的内存块。SDK (http://developer.nvidia.com/cuda-cc-sdk-code-samples) 中的“Matrix Transpose”示例有一个名为“copySharedMem”的内核,它几乎完全符合您的描述。

于 2012-07-02T17:11:19.917 回答