From a696348ec5662c1a1666836d6046046a1d7a1795 Mon Sep 17 00:00:00 2001 From: Julien Maille Date: Fri, 17 Jan 2020 14:21:25 +0100 Subject: [PATCH] FIX: disable dnn cuda input_shortcut on _half for CC<5.3 --- modules/dnn/src/cuda/shortcut.cu | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/dnn/src/cuda/shortcut.cu b/modules/dnn/src/cuda/shortcut.cu index e2958627ab..8b1d369bb6 100644 --- 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 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, TensorView, TensorView); }}}} /* namespace cv::dnn::cuda4dnn::kernels */