mirror of
https://github.com/opencv/opencv.git
synced 2024-11-24 19:20:28 +08:00
ocl: workaround for subbuffer memory leaks
This commit is contained in:
parent
be37d99567
commit
03646e7e01
@ -148,11 +148,27 @@ void openCLMallocPitchEx(Context *ctx, void **dev_ptr, size_t *pitch,
|
||||
{
|
||||
cl_int status;
|
||||
size_t size = widthInBytes * height;
|
||||
bool useSubBuffers =
|
||||
#ifndef MEMORY_CORRUPTION_GUARD
|
||||
false;
|
||||
#else
|
||||
true;
|
||||
#endif
|
||||
const DeviceInfo& devInfo = ctx->getDeviceInfo();
|
||||
if (useSubBuffers && devInfo.isIntelDevice)
|
||||
{
|
||||
useSubBuffers = false; // TODO FIXIT We observe memory leaks then we working with sub-buffers
|
||||
// on the CPU device of Intel OpenCL SDK (Linux). We will investigate this later.
|
||||
}
|
||||
if (!useSubBuffers)
|
||||
{
|
||||
*dev_ptr = clCreateBuffer(getClContext(ctx), gDevMemRWValueMap[rw_type]|gDevMemTypeValueMap[mem_type],
|
||||
size, 0, &status);
|
||||
openCLVerifyCall(status);
|
||||
#else
|
||||
}
|
||||
#ifdef MEMORY_CORRUPTION_GUARD
|
||||
else
|
||||
{
|
||||
size_t allocSize = size + __memory_corruption_guard_bytes * 2;
|
||||
cl_mem mainBuffer = clCreateBuffer(getClContext(ctx), gDevMemRWValueMap[rw_type]|gDevMemTypeValueMap[mem_type],
|
||||
allocSize, 0, &status);
|
||||
@ -177,6 +193,7 @@ void openCLMallocPitchEx(Context *ctx, void **dev_ptr, size_t *pitch,
|
||||
#endif
|
||||
CheckBuffers data(mainBuffer, size, widthInBytes, height);
|
||||
__check_buffers.insert(std::pair<cl_mem, CheckBuffers>((cl_mem)*dev_ptr, data));
|
||||
}
|
||||
#endif
|
||||
*pitch = widthInBytes;
|
||||
}
|
||||
@ -230,6 +247,7 @@ void openCLCopyBuffer2D(Context *ctx, void *dst, size_t dpitch, int dst_offset,
|
||||
|
||||
void openCLFree(void *devPtr)
|
||||
{
|
||||
openCLSafeCall(clReleaseMemObject((cl_mem)devPtr));
|
||||
#ifdef MEMORY_CORRUPTION_GUARD
|
||||
#ifdef CHECK_MEMORY_CORRUPTION
|
||||
bool failBefore = false, failAfter = false;
|
||||
@ -270,9 +288,7 @@ void openCLFree(void *devPtr)
|
||||
openCLSafeCall(clReleaseMemObject(data.mainBuffer));
|
||||
__check_buffers.erase(i);
|
||||
}
|
||||
#endif
|
||||
openCLSafeCall(clReleaseMemObject((cl_mem)devPtr));
|
||||
#if defined(MEMORY_CORRUPTION_GUARD) && defined(CHECK_MEMORY_CORRUPTION)
|
||||
#if defined(CHECK_MEMORY_CORRUPTION)
|
||||
if (failBefore)
|
||||
{
|
||||
#ifdef CHECK_MEMORY_CORRUPTION_PRINT_ERROR
|
||||
@ -291,7 +307,8 @@ void openCLFree(void *devPtr)
|
||||
CV_Error(CV_StsInternal, "Memory corruption detected: after buffer");
|
||||
#endif
|
||||
}
|
||||
#endif
|
||||
#endif // CHECK_MEMORY_CORRUPTION
|
||||
#endif // MEMORY_CORRUPTION_GUARD
|
||||
}
|
||||
|
||||
cl_kernel openCLGetKernelFromSource(const Context *ctx, const cv::ocl::ProgramEntry* source, string kernelName)
|
||||
|
Loading…
Reference in New Issue
Block a user