JCuda : 장치에서 호스트로 다차원 배열을 복사하십시오.

Question

저는 몇 달 동안 JCuda를 사용 해왔고 장치 메모리에서 호스트 메모리로 다차원 배열을 복사 할 수 없습니다. 재미있는 점은 반대 방향으로는 아무런 문제가 없다는 것입니다 (다차원 배열로 커널을 호출 할 수 있고 모든 것이 올바른 값으로 작동 함).

몇 마디로 말해, 커널의 결과를 2 차원 배열의 단락에 넣었습니다.이 배열의 첫 번째 차원은 스레드의 수로, 서로 다른 위치에 쓸 수 있습니다. 여기

예 : 내가 잘못 뭐하는 거지

CUdeviceptr pointer_dev = new CUdeviceptr(); 
cuMemAlloc(pointer_dev, Sizeof.POINTER); // in this case, as an example, it's an array with one element (one thread), but it doesn't matter 

// Invoke kernel with pointer_dev as parameter. Now it should contain some results 

CUdeviceptr[] arrayPtr = new CUdeviceptr[1]; // It will point to the result 
arrayPtr[0] = new CUdeviceptr(); 
short[] resultArray = new short[3]; // an array of 3 shorts was allocated in the kernel 

cuMemAlloc(arrayPtr[0], 3 * Sizeof.SHORT); 
cuMemcpyDtoH(Pointer.to(arrayPtr), pointer_dev, Sizeof.POINTER); // Its seems, using the debugger, that the value of arrayPtr[0] isn't changed here! 
cuMemcpyDtoH(Pointer.to(resultArray), arrayPtr[0], 3 * Sizeof.SHORT); // Not the expected values in resultArray, probably because of the previous instruction 

?

편집 : link

하나를 : 분명히

이에 명시된 (그리고 더 많은) 스레드로 장치에 할당 된 메모리가 다시 호스트로 복사 할 수 없습니다 몇 가지 제한 사항이 있습니다 해결 방법은 무엇입니까? 나는 CUDA Toolkit v5.0을 사용하고 있습니다.

Answer 1

여기서 우리는 장치에서 호스트로 int의 2 차원 배열을 복사하고 있습니다.

I) 먼저 다른 차원 배열의 크기와 동일한 [여기서는 blockSizeX] 크기의 단일 차원 배열을 만듭니다.

 CUdeviceptr hostDevicePointers[] = new CUdeviceptr[blockSizeX]; 
     for (int i = 0; i < blockSizeX; i++) { 
      hostDevicePointers[i] = new CUdeviceptr(); 
      cuMemAlloc(hostDevicePointers[i], size * Sizeof.INT); 
     } 

II) 상기 다른 배열 가리킨 호스트로부터 디바이스 배열 포인터를 복사 포인터들의 어레이에 대해 디바이스 메모리를 할당한다.

 CUdeviceptr hostDevicePointersArray = new CUdeviceptr(); 
     cuMemAlloc(hostDevicePointersArray, blockSizeX * Sizeof.POINTER); 
     cuMemcpyHtoD(hostDevicePointersArray, Pointer.to(hostDevicePointers),blockSizeX * Sizeof.POINTER); 

III) 시작 커널

kernelLauncher.call (........ hostDevicePointersArray);

  int hostOutputData[] = new int[ numberofelementsInArray * blockSizeX]; 
      cuMemcpyDtoH(Pointer.to(hostOutputData), hostDevicePointers[i], numberofelementsInArray * blockSizeX * Sizeof.INT); 
      for (int j = 0; j < size; j++) { 
       sum = sum + hostOutputData[j]; 
      }