generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix for PR 2196

Browse Source
This commit is contained in:
Ilya Lavrenov 2014-01-30 01:47:25 +04:00
parent afa62c4161
commit 86353eb541
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
modules/core/src/ocl.cpp
View File

@ -2704,10 +2704,12 @@ bool Kernel::empty() const
int Kernel::set(int i, const void* value, size_t sz)
{
if (!p || !p->handle)
return -1;
CV_Assert(i >= 0);
if( i == 0 )
p->cleanupUMats();
if( !p || !p->handle || clSetKernelArg(p->handle, (cl_uint)i, sz, value) < 0 )
if( clSetKernelArg(p->handle, (cl_uint)i, sz, value) < 0 )
return -1;
return i+1;
}
@ -2725,9 +2727,9 @@ int Kernel::set(int i, const UMat& m)
int Kernel::set(int i, const KernelArg& arg)
{
CV_Assert( i >= 0 );
if( !p || !p->handle )
return -1;
CV_Assert( i >= 0 );
if( i == 0 )
p->cleanupUMats();
if( arg.m )
@ -2737,6 +2739,13 @@ int Kernel::set(int i, const KernelArg& arg)
bool ptronly = (arg.flags & KernelArg::PTR_ONLY) != 0;
cl_mem h = (cl_mem)arg.m->handle(accessFlags);
if (!h)
{
p->release();
p = 0;
return -1;
}
if (ptronly)
clSetKernelArg(p->handle, (cl_uint)i++, sizeof(h), &h);
else if( arg.m->dims <= 2 )