cuda - CUDA 图形流捕获与thrust::reduce

Question

当我试图捕获流执行以构建 CUDA 图时，调用会thrust::reduce导致运行时错误cudaErrorStreamCaptureUnsupported: operation not permitted when stream is capturing。我已经尝试将归约结果返回给主机和设备变量，并且我正在通过thrust::cuda::par.on(stream). 有什么方法可以将thrust函数执行添加到 CUDA 图中？

Answer 1

Thrust 的归约操作是主机端的阻塞操作。我假设您正在使用缩减结果作为以下内核之一的参数。因此，当您捕获 CUDA 图形时，它无法实例化图形可执行文件，因为您依赖于主机端的变量，但在缩减内核完成执行之前不可用。作为一种解决方案，您可以尝试将主机节点添加到返回缩减结果的图表中。