是的,可以通过 cudaStreamAddCallback 向一个 stream 添加多个回调。

添加回调的顺序与调用的顺序一致,即先添加的回调先被调用,后添加的回调后被调用。当多个回调被添加到同一个 stream 时,它们的执行顺序可以通过添加回调时指定的优先级来控制。回调的优先级值越小,优先级越高,先被调用。如果多个回调的优先级相同,则它们的调用顺序与添加顺序一致。

CUDA 10.0 Stream 回调:添加多个回调及调用顺序

原文地址: https://www.cveoy.top/t/topic/oe1s 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录