Method to_host

include/base/memory.h:165–173

Copy the memory space back from GPU */

    /** Copy the memory space back from GPU */
    void to_host(Index copy_count = 0) {
        if (count && device_id != -1) {
            if (!copy_count)
                copy_count = count;
            CUDA_CHECK(cudaSetDevice(device_id));
            CUDA_CHECK(cudaMemcpyAsync(host_ptr, device_ptr, copy_count * sizeof(Data), cudaMemcpyDeviceToHost, stream));
            CUDA_CHECK(cudaStreamSynchronize(stream));
        }
    }

    /** Copy the memory space back from GPU (asynchronous) */
    void to_host_async(Index copy_count = 0) {