Commit 34e2c78c authored by Vladislav Vinogradov's avatar Vladislav Vinogradov

minor fix

parent 5fe8beac
...@@ -454,7 +454,7 @@ GpuMat& GpuMat::setTo(const Scalar& s, const GpuMat& mask) ...@@ -454,7 +454,7 @@ GpuMat& GpuMat::setTo(const Scalar& s, const GpuMat& mask)
if (cn == 1 || (cn == 2 && s[0] == s[1]) || (cn == 3 && s[0] == s[1] && s[0] == s[2]) || (cn == 4 && s[0] == s[1] && s[0] == s[2] && s[0] == s[3])) if (cn == 1 || (cn == 2 && s[0] == s[1]) || (cn == 3 && s[0] == s[1] && s[0] == s[2]) || (cn == 4 && s[0] == s[1] && s[0] == s[2] && s[0] == s[3]))
{ {
int val = saturate_cast<uchar>(s[0]); int val = saturate_cast<uchar>(s[0]);
cudaSafeCall( cudaMemset2D(data, step, val, cols, rows) ); cudaSafeCall( cudaMemset2D(data, step, val, cols * elemSize(), rows) );
return *this; return *this;
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment