Question

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;
}

Answer 1

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

GPU上的连续内存分配

1 个答案: