mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-25 05:24:02 +08:00
Add support for zero-copy r/w buffer
Change-Id: Ie9f3f2211d107eb338bd97692d36e9c7d7a0feab
This commit is contained in:
@@ -45,11 +45,31 @@ cl_int CommandQueueHw<GfxFamily>::enqueueWriteBuffer(
|
||||
cl_event *event) {
|
||||
|
||||
cl_int retVal = CL_SUCCESS;
|
||||
|
||||
auto isMemTransferNeeded = buffer->checkIfMemoryTransferIsRequired(offset, ptr, CL_COMMAND_WRITE_BUFFER);
|
||||
if ((DebugManager.flags.DoCpuCopyOnWriteBuffer.get() ||
|
||||
buffer->isReadWriteOnCpuAllowed(blockingWrite, numEventsInWaitList, const_cast<void *>(ptr), size)) &&
|
||||
context->getDevice(0)->getDeviceInfo().cpuCopyAllowed) {
|
||||
if (!isMemTransferNeeded) {
|
||||
cpuDataTransferHandler(buffer,
|
||||
CL_COMMAND_MARKER,
|
||||
CL_TRUE,
|
||||
offset,
|
||||
size,
|
||||
const_cast<void *>(ptr),
|
||||
numEventsInWaitList,
|
||||
eventWaitList,
|
||||
event,
|
||||
retVal);
|
||||
if (event) {
|
||||
auto pEvent = castToObjectOrAbort<Event>(*event);
|
||||
pEvent->setCmdType(CL_COMMAND_WRITE_BUFFER);
|
||||
}
|
||||
|
||||
if (context->isProvidingPerformanceHints()) {
|
||||
context->providePerformanceHint(CL_CONTEXT_DIAGNOSTICS_LEVEL_GOOD_INTEL, CL_ENQUEUE_WRITE_BUFFER_DOESNT_REQUIRE_COPY_DATA, static_cast<cl_mem>(buffer), ptr);
|
||||
}
|
||||
return retVal;
|
||||
}
|
||||
cpuDataTransferHandler(buffer,
|
||||
CL_COMMAND_WRITE_BUFFER,
|
||||
CL_TRUE,
|
||||
@@ -62,9 +82,28 @@ cl_int CommandQueueHw<GfxFamily>::enqueueWriteBuffer(
|
||||
retVal);
|
||||
return retVal;
|
||||
}
|
||||
|
||||
MultiDispatchInfo dispatchInfo;
|
||||
if (!isMemTransferNeeded) {
|
||||
NullSurface s;
|
||||
Surface *surfaces[] = {&s};
|
||||
enqueueHandler<CL_COMMAND_MARKER>(
|
||||
surfaces,
|
||||
blockingWrite == CL_TRUE,
|
||||
dispatchInfo,
|
||||
numEventsInWaitList,
|
||||
eventWaitList,
|
||||
event);
|
||||
if (event) {
|
||||
auto pEvent = castToObjectOrAbort<Event>(*event);
|
||||
pEvent->setCmdType(CL_COMMAND_WRITE_BUFFER);
|
||||
}
|
||||
|
||||
if (context->isProvidingPerformanceHints()) {
|
||||
context->providePerformanceHint(CL_CONTEXT_DIAGNOSTICS_LEVEL_GOOD_INTEL, CL_ENQUEUE_WRITE_BUFFER_DOESNT_REQUIRE_COPY_DATA, static_cast<cl_mem>(buffer), ptr);
|
||||
}
|
||||
|
||||
return CL_SUCCESS;
|
||||
}
|
||||
auto &builder = BuiltIns::getInstance().getBuiltinDispatchInfoBuilder(EBuiltInOps::CopyBufferToBuffer,
|
||||
this->getContext(), this->getDevice());
|
||||
|
||||
|
||||
Reference in New Issue
Block a user