mirror of
https://github.com/intel/compute-runtime.git
synced 2026-01-12 17:33:00 +08:00
55 lines
1.9 KiB
C++
55 lines
1.9 KiB
C++
/*
|
|
* Copyright (C) 2017-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "core/memory_manager/memory_manager.h"
|
|
#include "opencl/source/mem_obj/buffer.h"
|
|
#include "opencl/source/sharings/d3d/d3d_sharing.h"
|
|
|
|
namespace NEO {
|
|
class Buffer;
|
|
class Context;
|
|
|
|
template <typename D3D>
|
|
class D3DBuffer : public D3DSharing<D3D> {
|
|
typedef typename D3D::D3DBufferObj D3DBufferObj;
|
|
typedef typename D3D::D3DBufferDesc D3DBufferDesc;
|
|
|
|
public:
|
|
static Buffer *create(Context *context, D3DBufferObj *d3dBuffer, cl_mem_flags flags, cl_int *retCode) {
|
|
ErrorCodeHelper err(retCode, CL_SUCCESS);
|
|
auto sharingFcns = context->getSharing<D3DSharingFunctions<D3D>>();
|
|
void *sharedHandle = nullptr;
|
|
|
|
D3DBufferDesc bufferDesc = {};
|
|
sharingFcns->getBufferDesc(&bufferDesc, d3dBuffer);
|
|
|
|
bool sharedResource = false;
|
|
D3DBufferObj *bufferStaging = nullptr;
|
|
if (bufferDesc.MiscFlags & D3DResourceFlags::MISC_SHARED) {
|
|
bufferStaging = d3dBuffer;
|
|
sharedResource = true;
|
|
} else {
|
|
sharingFcns->createBuffer(&bufferStaging, bufferDesc.ByteWidth);
|
|
}
|
|
|
|
sharingFcns->getSharedHandle(bufferStaging, &sharedHandle);
|
|
AllocationProperties properties = {context->getDevice(0)->getRootDeviceIndex(), false, 0, GraphicsAllocation::AllocationType::SHARED_BUFFER, false};
|
|
auto alloc = context->getMemoryManager()->createGraphicsAllocationFromSharedHandle(toOsHandle(sharedHandle), properties, true);
|
|
|
|
auto d3dBufferObj = new D3DBuffer<D3D>(context, d3dBuffer, bufferStaging, sharedResource);
|
|
|
|
return Buffer::createSharedBuffer(context, flags, d3dBufferObj, alloc);
|
|
}
|
|
~D3DBuffer() override = default;
|
|
|
|
protected:
|
|
D3DBuffer(Context *context, D3DBufferObj *d3dBuffer, D3DBufferObj *bufferStaging, bool sharedResource)
|
|
: D3DSharing(context, d3dBuffer, bufferStaging, 0, sharedResource){};
|
|
};
|
|
} // namespace NEO
|