Administrator/opencv-MIRROR
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-09-13 05:42:51 -05:00
Code Issues 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
parent d85e67d3ec a696348ec5
commit 282fcb90dc
1 changed files with 2 additions and 0 deletions
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 */