cuda: fix for compatibility with CUDA Toolkit >= 12.2.0

pull/24104/head
cudawarped 1 year ago
parent 6791284994
commit ab8cb6f8a9
  1. 2
      modules/dnn/src/cuda4dnn/primitives/normalize_bbox.hpp
  2. 2
      modules/dnn/src/cuda4dnn/primitives/region.hpp

@ -111,7 +111,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
* or there might be several weights * or there might be several weights
* or we don't have to scale * or we don't have to scale
*/ */
if (weight != 1.0) if (weight != static_cast<T>(1.0f))
{ {
kernels::scale1_with_bias1<T>(stream, output, input, weight, 1.0); kernels::scale1_with_bias1<T>(stream, output, input, weight, 1.0);
} }

@ -121,7 +121,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
new_coords new_coords
); );
if (nms_iou_threshold > 0) { if (nms_iou_threshold > static_cast<T>(0.0f)) {
auto output_mat = output_wrapper->getMutableHostMat(); auto output_mat = output_wrapper->getMutableHostMat();
CV_Assert(output_mat.type() == CV_32F); CV_Assert(output_mat.type() == CV_32F);
for (int i = 0; i < input.get_axis_size(0); i++) { for (int i = 0; i < input.get_axis_size(0); i++) {

Loading…
Cancel
Save