CUDA Libraries and MPI+OpenMP+CUDA - Prace Training Portal

fft3Dtest.cu – Code Snippet (C++)long h_offset=0;for(int i=0; i < totalFFT; i += nGPU*nPerCall) {for(int j=0; j < nGPU; j++) {cudaSetDevice(j);cudaMemcpyAsync(d_data[j], ((char*)h_data)+h_offset,bytesPerGPU, cudaMemcpyDefault,streams[j]);cforwardFFT_(fftPlanMany[j],d_data[j], d_data[j]);cinverseFFT_(fftPlanMany[j],d_data[j], d_data[j]);}}cudaMemcpyAsync(((char*)h_data)+h_offset, d_data[j],bytesPerGPU, cudaMemcpyDefault,streams[j]);h_offset += bytesPerGPU;cudaDeviceSynchronize();cudaSetDevice(0);Courtesy of R. FarberFebruary 10, 2012 PRACE Winter School 201215