cv :: cuda :: Stream выбрать поток по умолчанию поток

Question

Мы компилируем, используя -default-stream=per-thread, чтобы получить поток в потоке для CUDA. Но когда мы делаем: cv::cuda::Stream::Null(), мы получаем указатель потока, который указывает на «устаревший поток по умолчанию» (см. Подробности здесь ).

Как мы можем получить cv::cuda::Stream, который указывает на дескриптор потока cuda CU_STREAM_PER_THREAD?

В общем, как мы можем получить cv::cuda::Stream, который указываетк существующей ручке cudaStream_t?

Answer 1

Здесь есть друг cv::cuda::Stream здесь , который называется 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, представляющего поток потока по умолчанию.