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

282fcb90 · Alexander Alekhin · d85e67d3 · a696348e · 282fcb90
Commit 282fcb90 authored Jan 17, 2020 by Alexander Alekhin
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

shortcut.cu modules/dnn/src/cuda/shortcut.cu +2 -0

No files found.
--- a/modules/dnn/src/cuda/shortcut.cu
+++ b/modules/dnn/src/cuda/shortcut.cu
@@ -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 */