如何检测内核代码中的当前流?

Question

是否可以检查cuda内核中执行的流(https://developer.download.nvidia.com/CUDA/training/StreamsAndConcurrencyWebinar.pdf) ?特别是，我有兴趣检查我是否在默认流中运行。

我认为，潜在的，这个信息可以从%envreg或%pt寄存器https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#identifiers中提取，但没有找到任何关于这个的文档。

Answer 1

目前没有任何方法可以在CUDA设备代码中检索基于主机的流。

正如评论中建议的那样，您可以通过内核参数传递所需的信息来满足您的需求。