mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-25 11:40:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #16376 from JulienMaille:cuda-dnn-for-older-gpus

Browse Source
This commit is contained in:
Alexander Alekhin 2020-01-17 15:42:50 +00:00
commit 282fcb90dc
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/dnn/src/cuda/shortcut.cu
View File

@ -103,7 +103,9 @@ void input_shortcut(const csl::Stream& stream, csl::TensorSpan<T> output, csl::T
}
}
#if !defined(__CUDA_ARCH__) || (__CUDA_ARCH__ >= 530)
template void input_shortcut(const Stream&, TensorSpan<__half>, TensorView<__half>, TensorView<__half>);
#endif
template void input_shortcut(const Stream&, TensorSpan<float>, TensorView<float>, TensorView<float>);
}}}} /* namespace cv::dnn::cuda4dnn::kernels */