CUDA Stream 回调函数优先级设置:cudaStreamAddCallback Flags
在使用 cudaStreamAddCallback 函数时,可以通过设置参数 flags 来指定回调的优先级。flags 参数可以设置为以下值:
cudaStreamCallbackNormal:普通优先级,回调函数将在 stream 中的所有未完成的操作完成后执行。cudaStreamCallbackLow:低优先级,回调函数将在 stream 中的所有普通优先级操作完成后执行。cudaStreamCallbackHigh:高优先级,回调函数将在 stream 中的所有普通优先级和低优先级操作完成后执行。
例如,以下代码将回调函数设置为高优先级:
cudaStreamAddCallback(stream, my_callback, data, 0, cudaStreamCallbackHigh);
其中,cudaStreamCallbackHigh 即为设置回调函数的优先级为高优先级。
原文地址: https://www.cveoy.top/t/topic/oe1E 著作权归作者所有。请勿转载和采集!