cv::cuda::Stream 选择线程默认流

【问题标题】：cv::cuda::Stream select thread default streamcv::cuda::Stream 选择线程默认流
【发布时间】：2020-02-29 18:48:55
【问题描述】：

我们正在使用-default-stream=per-thread 进行编译以获取 CUDA 的每个线程流。但是当我们这样做时：cv::cuda::Stream::Null() 我们会得到一个指向“传统默认流”的流指针（请参阅详细信息here）。

我们如何获得指向 cuda 流句柄 CU_STREAM_PER_THREAD 的 cv::cuda::Stream？

更一般地说，我们如何获得指向现有cudaStream_t 句柄的cv::cuda::Stream？

【问题讨论】：

标签： c++ opencv cuda

【解决方案1】：

cv::cuda::Streamhere有一个朋友叫cv::cuda::StreamAccessor。这个结构有一个静态成员函数：

static Stream wrapStream (cudaStream_t stream)

可以这样使用：

#include <opencv2/core/cuda_stream_accessor.hpp>
#include <cuda.h>

...

auto cvThreadDefaultStream = 
    cv::cuda::StreamAccessor::wrapStream(CU_STREAM_PER_THREAD);

获取代表默认线程流的cv Stream。

【讨论】：