compute-runtime/runtime/os_interface/windows/api.cpp

437 lines
20 KiB
C++

/*
* Copyright (c) 2017, Intel Corporation
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included
* in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
#include "runtime/api/dispatch.h"
#include "runtime/api/api.h"
#include "runtime/platform/platform.h"
#include "runtime/device/device.h"
#include "runtime/command_queue/command_queue.h"
#include "runtime/context/context.h"
#include "runtime/helpers/get_info.h"
#include "runtime/sharings/d3d/d3d_buffer.h"
#include "runtime/sharings/d3d/d3d_texture.h"
#include "runtime/sharings/d3d/d3d_surface.h"
#include "runtime/mem_obj/image.h"
using namespace OCLRT;
void OCLRT::MemObj::getOsSpecificMemObjectInfo(const cl_mem_info &paramName, size_t *srcParamSize, void **srcParam) {
switch (paramName) {
case CL_MEM_D3D10_RESOURCE_KHR:
*srcParamSize = sizeof(ID3D10Resource *);
*srcParam = static_cast<D3DSharing<D3DTypesHelper::D3D10> *>(peekSharingHandler())->getResourceHandler();
break;
case CL_MEM_D3D11_RESOURCE_KHR:
*srcParamSize = sizeof(ID3D11Resource *);
*srcParam = static_cast<D3DSharing<D3DTypesHelper::D3D11> *>(peekSharingHandler())->getResourceHandler();
break;
case CL_MEM_DX9_MEDIA_SURFACE_INFO_KHR:
*srcParamSize = sizeof(cl_dx9_surface_info_khr);
*srcParam = &static_cast<D3DSurface *>(peekSharingHandler())->getSurfaceInfo();
break;
case CL_MEM_DX9_MEDIA_ADAPTER_TYPE_KHR:
*srcParamSize = sizeof(cl_dx9_media_adapter_type_khr);
*srcParam = &static_cast<D3DSurface *>(peekSharingHandler())->getAdapterType();
break;
case CL_MEM_DX9_RESOURCE_INTEL:
*srcParamSize = sizeof(IDirect3DSurface9 *);
*srcParam = &static_cast<D3DSurface *>(peekSharingHandler())->getSurfaceInfo().resource;
break;
case CL_MEM_DX9_SHARED_HANDLE_INTEL:
*srcParamSize = sizeof(HANDLE);
*srcParam = &static_cast<D3DSurface *>(peekSharingHandler())->getSurfaceInfo().shared_handle;
break;
}
}
void OCLRT::Image::getOsSpecificImageInfo(const cl_mem_info &paramName, size_t *srcParamSize, void **srcParam) {
switch (paramName) {
case CL_IMAGE_D3D10_SUBRESOURCE_KHR:
*srcParamSize = sizeof(unsigned int);
*srcParam = &static_cast<D3DSharing<D3DTypesHelper::D3D10> *>(peekSharingHandler())->getSubresource();
break;
case CL_IMAGE_D3D11_SUBRESOURCE_KHR:
*srcParamSize = sizeof(unsigned int);
*srcParam = &static_cast<D3DSharing<D3DTypesHelper::D3D11> *>(peekSharingHandler())->getSubresource();
break;
case CL_IMAGE_DX9_MEDIA_PLANE_KHR:
case CL_IMAGE_DX9_PLANE_INTEL:
*srcParamSize = sizeof(cl_uint);
*srcParam = &static_cast<D3DSurface *>(peekSharingHandler())->getPlane();
break;
}
}
cl_int OCLRT::Context::createContextOsProperties(cl_context_properties &propertyType, cl_context_properties &propertyValue) {
return CL_INVALID_PROPERTY;
}
void *OCLRT::Context::getOsContextInfo(cl_context_info &paramName, size_t *srcParamSize) {
switch (paramName) {
case CL_CONTEXT_D3D10_PREFER_SHARED_RESOURCES_KHR:
case CL_CONTEXT_D3D11_PREFER_SHARED_RESOURCES_KHR:
*srcParamSize = sizeof(cl_bool);
return &preferD3dSharedResources;
default:
break;
}
return nullptr;
}
cl_int CL_API_CALL clGetDeviceIDsFromDX9INTEL(cl_platform_id platform, cl_dx9_device_source_intel dx9DeviceSource, void *dx9Object,
cl_dx9_device_set_intel dx9DeviceSet, cl_uint numEntries, cl_device_id *devices, cl_uint *numDevices) {
cl_device_id device = castToObject<Platform>(platform)->getDevice(0);
GetInfoHelper::set(devices, device);
GetInfoHelper::set(numDevices, 1u);
return CL_SUCCESS;
}
cl_mem CL_API_CALL clCreateFromDX9MediaSurfaceINTEL(cl_context context, cl_mem_flags flags, IDirect3DSurface9 *resource,
HANDLE sharedHandle, UINT plane, cl_int *errcodeRet) {
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
cl_mem_flags validFlags = CL_MEM_READ_WRITE | CL_MEM_WRITE_ONLY | CL_MEM_READ_ONLY;
if ((flags & (~validFlags)) != 0) {
err.set(CL_INVALID_VALUE);
return nullptr;
}
if (!resource) {
err.set(CL_INVALID_DX9_RESOURCE_INTEL);
return nullptr;
}
cl_dx9_surface_info_khr surfaceInfo = {resource, sharedHandle};
auto ctx = castToObject<Context>(context);
return D3DSurface::create(ctx, &surfaceInfo, flags, 0, plane, errcodeRet);
}
cl_int CL_API_CALL clEnqueueAcquireDX9ObjectsINTEL(cl_command_queue commandQueue, cl_uint numObjects, const cl_mem *memObjects,
cl_uint numEventsInWaitList, const cl_event *eventWaitList, cl_event *event) {
auto cmdQ = castToObject<CommandQueue>(commandQueue);
return cmdQ->enqueueAcquireSharedObjects(numObjects, memObjects, numEventsInWaitList,
eventWaitList, event, CL_COMMAND_ACQUIRE_DX9_OBJECTS_INTEL);
}
cl_int CL_API_CALL clEnqueueReleaseDX9ObjectsINTEL(cl_command_queue commandQueue, cl_uint numObjects, cl_mem *memObjects,
cl_uint numEventsInWaitList, const cl_event *eventWaitList, cl_event *event) {
auto cmdQ = castToObject<CommandQueue>(commandQueue);
for (unsigned int object = 0; object < numObjects; object++) {
auto memObject = castToObject<MemObj>(memObjects[object]);
if (!static_cast<D3DSharing<D3DTypesHelper::D3D9> *>(memObject->peekSharingHandler())->isSharedResource()) {
cmdQ->finish(true);
break;
}
}
auto retVal = cmdQ->enqueueReleaseSharedObjects(numObjects, memObjects, numEventsInWaitList,
eventWaitList, event, CL_COMMAND_RELEASE_DX9_OBJECTS_INTEL);
if (!cmdQ->getContext().getInteropUserSyncEnabled()) {
cmdQ->finish(true);
}
return retVal;
}
cl_int CL_API_CALL clGetDeviceIDsFromDX9MediaAdapterKHR(cl_platform_id platform, cl_uint numMediaAdapters,
cl_dx9_media_adapter_type_khr *mediaAdapterType, void *mediaAdapters,
cl_dx9_media_adapter_set_khr mediaAdapterSet, cl_uint numEntries,
cl_device_id *devices, cl_uint *numDevices) {
cl_device_id device = castToObject<Platform>(platform)->getDevice(0);
GetInfoHelper::set(devices, device);
GetInfoHelper::set(numDevices, 1u);
return CL_SUCCESS;
}
cl_mem CL_API_CALL clCreateFromDX9MediaSurfaceKHR(cl_context context, cl_mem_flags flags, cl_dx9_media_adapter_type_khr adapterType,
void *surfaceInfo, cl_uint plane, cl_int *errcodeRet) {
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
auto localSurfaceInfo = (cl_dx9_surface_info_khr *)surfaceInfo;
auto ctx = castToObject<Context>(context);
return D3DSurface::create(ctx, localSurfaceInfo, flags, adapterType, plane, errcodeRet);
}
cl_int CL_API_CALL clEnqueueAcquireDX9MediaSurfacesKHR(cl_command_queue commandQueue, cl_uint numObjects, const cl_mem *memObjects,
cl_uint numEventsInWaitList, const cl_event *eventWaitList, cl_event *event) {
auto cmdQ = castToObject<CommandQueue>(commandQueue);
return cmdQ->enqueueAcquireSharedObjects(numObjects, memObjects, numEventsInWaitList,
eventWaitList, event, CL_COMMAND_ACQUIRE_DX9_MEDIA_SURFACES_KHR);
}
cl_int CL_API_CALL clEnqueueReleaseDX9MediaSurfacesKHR(cl_command_queue commandQueue, cl_uint numObjects, const cl_mem *memObjects,
cl_uint numEventsInWaitList, const cl_event *eventWaitList, cl_event *event) {
auto cmdQ = castToObject<CommandQueue>(commandQueue);
for (unsigned int object = 0; object < numObjects; object++) {
auto memObject = castToObject<MemObj>(memObjects[object]);
if (!static_cast<D3DSharing<D3DTypesHelper::D3D9> *>(memObject->peekSharingHandler())->isSharedResource()) {
cmdQ->finish(true);
break;
}
}
auto retVal = cmdQ->enqueueReleaseSharedObjects(numObjects, memObjects, numEventsInWaitList,
eventWaitList, event, CL_COMMAND_RELEASE_DX9_MEDIA_SURFACES_KHR);
if (!cmdQ->getContext().getInteropUserSyncEnabled()) {
cmdQ->finish(true);
}
return retVal;
}
cl_int CL_API_CALL clGetDeviceIDsFromD3D10KHR(cl_platform_id platform, cl_d3d10_device_source_khr d3dDeviceSource,
void *d3dObject, cl_d3d10_device_set_khr d3dDeviceSet, cl_uint numEntries,
cl_device_id *devices, cl_uint *numDevices) {
DXGI_ADAPTER_DESC dxgiDesc = {{0}};
IDXGIAdapter *dxgiAdapter = nullptr;
ID3D10Device *d3dDevice = nullptr;
D3DSharingFunctions<D3DTypesHelper::D3D10> sharingFcns((ID3D10Device *)nullptr);
cl_uint localNumDevices = 0;
cl_int retCode = CL_SUCCESS;
if (DebugManager.injectFcn) {
sharingFcns.getDxgiDescFcn = (D3DSharingFunctions<D3DTypesHelper::D3D10>::GetDxgiDescFcn)DebugManager.injectFcn;
}
cl_device_id device = castToObject<Platform>(platform)->getDevice(0);
switch (d3dDeviceSource) {
case CL_D3D10_DEVICE_KHR:
d3dDevice = (ID3D10Device *)d3dObject;
break;
case CL_D3D10_DXGI_ADAPTER_KHR:
dxgiAdapter = (IDXGIAdapter *)d3dObject;
break;
default:
GetInfoHelper::set(numDevices, localNumDevices);
return CL_INVALID_VALUE;
break;
}
sharingFcns.getDxgiDescFcn(&dxgiDesc, dxgiAdapter, d3dDevice);
if (dxgiDesc.VendorId != INTEL_VENDOR_ID) {
GetInfoHelper::set(numDevices, localNumDevices);
return CL_DEVICE_NOT_FOUND;
}
switch (d3dDeviceSet) {
case CL_PREFERRED_DEVICES_FOR_D3D10_KHR:
case CL_ALL_DEVICES_FOR_D3D10_KHR:
GetInfoHelper::set(devices, device);
localNumDevices = 1;
break;
default:
retCode = CL_INVALID_VALUE;
break;
}
GetInfoHelper::set(numDevices, localNumDevices);
return retCode;
}
cl_mem CL_API_CALL clCreateFromD3D10BufferKHR(cl_context context, cl_mem_flags flags, ID3D10Buffer *resource, cl_int *errcodeRet) {
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
auto ctx = castToObject<Context>(context);
if (ctx->getSharing<D3DSharingFunctions<D3DTypesHelper::D3D10>>()->isTracked(resource, 0)) {
err.set(CL_INVALID_D3D10_RESOURCE_KHR);
return nullptr;
}
return D3DBuffer<D3DTypesHelper::D3D10>::create(ctx, resource, flags, errcodeRet);
}
cl_mem CL_API_CALL clCreateFromD3D10Texture2DKHR(cl_context context, cl_mem_flags flags, ID3D10Texture2D *resource,
UINT subresource, cl_int *errcodeRet) {
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
auto ctx = castToObject<Context>(context);
if (ctx->getSharing<D3DSharingFunctions<D3DTypesHelper::D3D10>>()->isTracked(resource, subresource)) {
err.set(CL_INVALID_D3D10_RESOURCE_KHR);
return nullptr;
}
return D3DTexture<D3DTypesHelper::D3D10>::create2d(ctx, resource, flags, subresource, errcodeRet);
}
cl_mem CL_API_CALL clCreateFromD3D10Texture3DKHR(cl_context context, cl_mem_flags flags, ID3D10Texture3D *resource,
UINT subresource, cl_int *errcodeRet) {
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
auto ctx = castToObject<Context>(context);
if (ctx->getSharing<D3DSharingFunctions<D3DTypesHelper::D3D10>>()->isTracked(resource, subresource)) {
err.set(CL_INVALID_D3D10_RESOURCE_KHR);
return nullptr;
}
return D3DTexture<D3DTypesHelper::D3D10>::create3d(ctx, resource, flags, subresource, errcodeRet);
}
cl_int CL_API_CALL clEnqueueAcquireD3D10ObjectsKHR(cl_command_queue commandQueue, cl_uint numObjects, const cl_mem *memObjects,
cl_uint numEventsInWaitList, const cl_event *eventWaitList, cl_event *event) {
auto cmdQ = castToObject<CommandQueue>(commandQueue);
for (unsigned int object = 0; object < numObjects; object++) {
if (castToObject<MemObj>(memObjects[object])->acquireCount >= 1) {
return CL_D3D10_RESOURCE_ALREADY_ACQUIRED_KHR;
}
}
return cmdQ->enqueueAcquireSharedObjects(numObjects, memObjects, numEventsInWaitList,
eventWaitList, event, CL_COMMAND_ACQUIRE_D3D10_OBJECTS_KHR);
}
cl_int CL_API_CALL clEnqueueReleaseD3D10ObjectsKHR(cl_command_queue commandQueue, cl_uint numObjects, const cl_mem *memObjects,
cl_uint numEventsInWaitList, const cl_event *eventWaitList, cl_event *event) {
auto cmdQ = castToObject<CommandQueue>(commandQueue);
for (unsigned int object = 0; object < numObjects; object++) {
auto memObject = castToObject<MemObj>(memObjects[object]);
if (memObject->acquireCount == 0) {
return CL_D3D10_RESOURCE_NOT_ACQUIRED_KHR;
}
if (!static_cast<D3DSharing<D3DTypesHelper::D3D10> *>(memObject->peekSharingHandler())->isSharedResource()) {
cmdQ->finish(true);
break;
}
}
auto retVal = cmdQ->enqueueReleaseSharedObjects(numObjects, memObjects, numEventsInWaitList,
eventWaitList, event, CL_COMMAND_RELEASE_D3D10_OBJECTS_KHR);
if (!cmdQ->getContext().getInteropUserSyncEnabled()) {
cmdQ->finish(true);
}
return retVal;
}
cl_int CL_API_CALL clGetDeviceIDsFromD3D11KHR(cl_platform_id platform, cl_d3d11_device_source_khr d3dDeviceSource,
void *d3dObject, cl_d3d11_device_set_khr d3dDeviceSet,
cl_uint numEntries, cl_device_id *devices, cl_uint *numDevices) {
DXGI_ADAPTER_DESC dxgiDesc = {{0}};
IDXGIAdapter *dxgiAdapter = nullptr;
ID3D11Device *d3dDevice = nullptr;
D3DSharingFunctions<D3DTypesHelper::D3D11> sharingFcns((ID3D11Device *)nullptr);
cl_uint localNumDevices = 0;
cl_int retCode = CL_SUCCESS;
if (DebugManager.injectFcn) {
sharingFcns.getDxgiDescFcn = (D3DSharingFunctions<D3DTypesHelper::D3D11>::GetDxgiDescFcn)DebugManager.injectFcn;
}
cl_device_id device = castToObject<Platform>(platform)->getDevice(0);
switch (d3dDeviceSource) {
case CL_D3D11_DEVICE_KHR:
d3dDevice = (ID3D11Device *)d3dObject;
break;
case CL_D3D11_DXGI_ADAPTER_KHR:
dxgiAdapter = (IDXGIAdapter *)d3dObject;
break;
default:
GetInfoHelper::set(numDevices, localNumDevices);
return CL_INVALID_VALUE;
break;
}
sharingFcns.getDxgiDescFcn(&dxgiDesc, dxgiAdapter, d3dDevice);
if (dxgiDesc.VendorId != INTEL_VENDOR_ID) {
GetInfoHelper::set(numDevices, localNumDevices);
return CL_DEVICE_NOT_FOUND;
}
switch (d3dDeviceSet) {
case CL_PREFERRED_DEVICES_FOR_D3D11_KHR:
case CL_ALL_DEVICES_FOR_D3D11_KHR:
GetInfoHelper::set(devices, device);
localNumDevices = 1;
break;
default:
retCode = CL_INVALID_VALUE;
break;
}
GetInfoHelper::set(numDevices, localNumDevices);
return retCode;
}
cl_mem CL_API_CALL clCreateFromD3D11BufferKHR(cl_context context, cl_mem_flags flags, ID3D11Buffer *resource, cl_int *errcodeRet) {
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
auto ctx = castToObject<Context>(context);
if (ctx->getSharing<D3DSharingFunctions<D3DTypesHelper::D3D11>>()->isTracked(resource, 0)) {
err.set(CL_INVALID_D3D11_RESOURCE_KHR);
return nullptr;
}
return D3DBuffer<D3DTypesHelper::D3D11>::create(ctx, resource, flags, errcodeRet);
}
cl_mem CL_API_CALL clCreateFromD3D11Texture2DKHR(cl_context context, cl_mem_flags flags, ID3D11Texture2D *resource,
UINT subresource, cl_int *errcodeRet) {
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
auto ctx = castToObject<Context>(context);
if (ctx->getSharing<D3DSharingFunctions<D3DTypesHelper::D3D11>>()->isTracked(resource, subresource)) {
err.set(CL_INVALID_D3D11_RESOURCE_KHR);
return nullptr;
}
return D3DTexture<D3DTypesHelper::D3D11>::create2d(ctx, resource, flags, subresource, errcodeRet);
}
cl_mem CL_API_CALL clCreateFromD3D11Texture3DKHR(cl_context context, cl_mem_flags flags, ID3D11Texture3D *resource,
UINT subresource, cl_int *errcodeRet) {
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
auto ctx = castToObject<Context>(context);
if (ctx->getSharing<D3DSharingFunctions<D3DTypesHelper::D3D11>>()->isTracked(resource, subresource)) {
err.set(CL_INVALID_D3D11_RESOURCE_KHR);
return nullptr;
}
return D3DTexture<D3DTypesHelper::D3D11>::create3d(ctx, resource, flags, subresource, errcodeRet);
}
cl_int CL_API_CALL clEnqueueAcquireD3D11ObjectsKHR(cl_command_queue commandQueue, cl_uint numObjects, const cl_mem *memObjects,
cl_uint numEventsInWaitList, const cl_event *eventWaitList, cl_event *event) {
auto cmdQ = castToObject<CommandQueue>(commandQueue);
for (unsigned int object = 0; object < numObjects; object++) {
if (castToObject<MemObj>(memObjects[object])->acquireCount >= 1) {
return CL_D3D11_RESOURCE_ALREADY_ACQUIRED_KHR;
}
}
return cmdQ->enqueueAcquireSharedObjects(numObjects, memObjects, numEventsInWaitList,
eventWaitList, event, CL_COMMAND_ACQUIRE_D3D11_OBJECTS_KHR);
}
cl_int CL_API_CALL clEnqueueReleaseD3D11ObjectsKHR(cl_command_queue commandQueue, cl_uint numObjects, const cl_mem *memObjects,
cl_uint numEventsInWaitList, const cl_event *eventWaitList, cl_event *event) {
auto cmdQ = castToObject<CommandQueue>(commandQueue);
for (unsigned int object = 0; object < numObjects; object++) {
auto memObject = castToObject<MemObj>(memObjects[object]);
if (memObject->acquireCount == 0) {
return CL_D3D11_RESOURCE_NOT_ACQUIRED_KHR;
}
if (!static_cast<D3DSharing<D3DTypesHelper::D3D11> *>(memObject->peekSharingHandler())->isSharedResource()) {
cmdQ->finish(true);
break;
}
}
auto retVal = cmdQ->enqueueReleaseSharedObjects(numObjects, memObjects, numEventsInWaitList,
eventWaitList, event, CL_COMMAND_RELEASE_D3D11_OBJECTS_KHR);
if (!cmdQ->getContext().getInteropUserSyncEnabled()) {
cmdQ->finish(true);
}
return retVal;
}