cudaMallocPitchによって割り当てられたメモリ（バイト単位）の計算方法

Question

cudaMallocPitchによって割り当てられたバイト単位で合計メモリを取得するにはどうすればよいですか？私はベースポインタとバイト数を期待するライブラリ（NvidiaのNCCL）を通してこのデータを別のGPUに送る必要があります。私は残念ながら私が使用することができないコピーのためのcudaMemcpy2D機能を認識しています。

Answer 1

cudaMallocPitchは、割り当てのピッチをバイト単位で返します。あなたはこのようにそれを呼ばれるのであれば：

void *devPtr; 
size_t pitch; 
cudaMallocPitch (&devPtr, &pitch, width, height); 

メモリ割り当てのサイズはpitch * heightバイトです。