FIX: disable dnn cuda input_shortcut on _half for CC<5.3

a696348e · Julien Maille · d85e67d3 · a696348e
Commit a696348e authored Jan 17, 2020 by Julien Maille
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 */