cuda: fix for compatibility with CUDA Toolkit >= 12.2.0

This commit is contained in:
cudawarped 2023-08-01 13:02:42 +03:00
parent 6791284994
commit ab8cb6f8a9
2 changed files with 2 additions and 2 deletions

View File

@ -111,7 +111,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
* or there might be several weights
* 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);
}

View File

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