Merge pull request #2147 from ilya-lavrenov:tapi_sqrt

12 years ago · fdb1019b9e
parent c3bfe3f33e 09bff78d63
commit fdb1019b9e
1 changed files with 8 additions and 3 deletions
--- a/modules/core/src/mathfuncs.cpp
+++ b/modules/core/src/mathfuncs.cpp
@ -2057,10 +2057,15 @@ static bool ocl_pow(InputArray _src, double power, OutputArray _dst)
    ocl::KernelArg srcarg = ocl::KernelArg::ReadOnlyNoSize(src),
            dstarg = ocl::KernelArg::WriteOnly(dst, cn);

-    if (depth == CV_32F)
-        k.args(srcarg, dstarg, (float)power);
+    if (issqrt)
+        k.args(srcarg, dstarg);
    else
-        k.args(srcarg, dstarg, power);
+    {
+        if (depth == CV_32F)
+            k.args(srcarg, dstarg, (float)power);
+        else
+            k.args(srcarg, dstarg, power);
+    }

    size_t globalsize[2] = { dst.cols *  cn, dst.rows };
    return k.run(2, globalsize, NULL, false);