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 */