dnn(ocl): fix gemm kernel scheduling
authorAlexander Alekhin <alexander.a.alekhin@gmail.com>
Fri, 26 Mar 2021 00:32:02 +0000 (00:32 +0000)
committerAlexander Alekhin <alexander.a.alekhin@gmail.com>
Fri, 26 Mar 2021 00:35:00 +0000 (00:35 +0000)
modules/dnn/src/ocl4dnn/src/math_functions.cpp

index e26a3c3f06f9e3b4d25207a4615da557d7c4f7bb..855a21e08ffefd01b6f26818dd8a318abc41a8c0 100644 (file)
@@ -112,14 +112,14 @@ ocl::Image2D ocl4dnnGEMMCopyBufferToImage(UMat buffer, int offset,
             global_copy[0] = padded_width;
             global_copy[1] = padded_height;
 
-            oclk_gemm_copy
+            bool res = oclk_gemm_copy
                 .args(
                     ocl::KernelArg::PtrReadOnly(buffer),
                     image, offset,
                     width, height,
                     ld)
                 .run(2, global_copy, NULL, false);
-            oclk_gemm_copy.run(2, global_copy, NULL, false);
+            CV_Assert(res);
         }
     }