Question

我想将部分数组从主机上的固定内存复制到CUDA设备。例如，我已经分配了大小为500的固定内存，我想将元素100-399复制到设备上大小为300的数组。

这是我的代码，

int main()
{
    const unsigned int N = 500;
    const unsigned int bytes = N * sizeof(int);
    cudaError_t status = cudaSuccess;

    int *h_a;
    int *d_a;

    status = cudaMallocHost((void**) &h_a, bytes);
    if (status != cudaSuccess)
        cout << "Error allocating pinned host memory\n";

    status = cudaMalloc((void**) &d_a, bytes);
    if (status != cudaSuccess)
        cout << "Error allocating pinned device memory\n";

    for (int i = 0; i < N; i++) {
        h_a[i] = i;
    }

    status = cudaMemcpy(d_a, h_a + 100, bytes - (200 * sizeof(int)), cudaMemcpyHostToDevice);
    if (status != cudaSuccess)
        cout << "Error copying to device: " << cudaGetErrorString(status) << "\n";
    cudaMemcpy(h_a + 100, d_a, bytes - (200 * sizeof(int)), cudaMemcpyDeviceToHost);
    if (status != cudaSuccess)
        cout << "Error copying to host: " << cudaGetErrorString(status) << "\n";       


    cudaFree(d_a);
    cudaFreeHost(h_a);

    return 0;
}

当我运行此操作时，我收到了主机到设备复制的错误，

Error copying to device: invalid argument

仅主机到设备复制失败。设备到主机副本工作正常。此外，如果我使用非固定主机内存，相同的代码可以正常工作。有没有办法使用固定内存实现这一目标？

Answer 1

上面的代码实际上编译并正常运行。也许我正在测试时使用旧的可执行文件。

复制使用cudamallocHost分配的部分内存

1 个答案: