mirror of
https://github.com/KhronosGroup/OpenCL-CTS.git
synced 2026-03-22 15:19:02 +00:00
Initial open source release of OpenCL 2.0 CTS.
This commit is contained in:
222
test_extensions/media_sharing/main.cpp
Normal file
222
test_extensions/media_sharing/main.cpp
Normal file
@@ -0,0 +1,222 @@
|
||||
//
|
||||
// Copyright (c) 2017 The Khronos Group Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
//
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "../../test_common/harness/testHarness.h"
|
||||
#include "utils.h"
|
||||
|
||||
extern int test_context_create(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements);
|
||||
extern int test_get_device_ids(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements);
|
||||
extern int test_api(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements);
|
||||
extern int test_kernel(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements);
|
||||
extern int test_other_data_types(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements);
|
||||
extern int test_memory_access(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements);
|
||||
extern int test_interop_user_sync(cl_device_id deviceID, cl_context context, cl_command_queue queue, int num_elements);
|
||||
|
||||
basefn basefn_list[] = {
|
||||
test_context_create,
|
||||
test_get_device_ids,
|
||||
test_api,
|
||||
test_kernel,
|
||||
test_other_data_types,
|
||||
test_memory_access,
|
||||
test_interop_user_sync,
|
||||
};
|
||||
|
||||
const char *basefn_names[] = {
|
||||
"test_context_create",
|
||||
"test_get_device_ids",
|
||||
"test_api",
|
||||
"test_kernel",
|
||||
"test_other_data_types",
|
||||
"test_memory_access",
|
||||
"test_interop_user_sync",
|
||||
};
|
||||
|
||||
ct_assert((sizeof(basefn_names) / sizeof(basefn_names[0])) == (sizeof(basefn_list) / sizeof(basefn_list[0])));
|
||||
|
||||
int num_fns = sizeof(basefn_names) / sizeof(char *);
|
||||
|
||||
clGetDeviceIDsFromDX9MediaAdapterKHR_fn clGetDeviceIDsFromDX9MediaAdapterKHR = NULL;
|
||||
clCreateFromDX9MediaSurfaceKHR_fn clCreateFromDX9MediaSurfaceKHR = NULL;
|
||||
clEnqueueAcquireDX9MediaSurfacesKHR_fn clEnqueueAcquireDX9MediaSurfacesKHR = NULL;
|
||||
clEnqueueReleaseDX9MediaSurfacesKHR_fn clEnqueueReleaseDX9MediaSurfacesKHR = NULL;
|
||||
|
||||
cl_platform_id gPlatformIDdetected;
|
||||
cl_device_id gDeviceIDdetected;
|
||||
cl_device_type gDeviceTypeSelected = CL_DEVICE_TYPE_DEFAULT;
|
||||
|
||||
bool MediaSurfaceSharingExtensionInit()
|
||||
{
|
||||
clGetDeviceIDsFromDX9MediaAdapterKHR = (clGetDeviceIDsFromDX9MediaAdapterKHR_fn)clGetExtensionFunctionAddressForPlatform(gPlatformIDdetected, "clGetDeviceIDsFromDX9MediaAdapterKHR");
|
||||
if (clGetDeviceIDsFromDX9MediaAdapterKHR == NULL)
|
||||
{
|
||||
log_error("clGetExtensionFunctionAddressForPlatform(clGetDeviceIDsFromDX9MediaAdapterKHR) returned NULL.\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
clCreateFromDX9MediaSurfaceKHR = (clCreateFromDX9MediaSurfaceKHR_fn)clGetExtensionFunctionAddressForPlatform(gPlatformIDdetected, "clCreateFromDX9MediaSurfaceKHR");
|
||||
if (clCreateFromDX9MediaSurfaceKHR == NULL)
|
||||
{
|
||||
log_error("clGetExtensionFunctionAddressForPlatform(clCreateFromDX9MediaSurfaceKHR) returned NULL.\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
clEnqueueAcquireDX9MediaSurfacesKHR = (clEnqueueAcquireDX9MediaSurfacesKHR_fn)clGetExtensionFunctionAddressForPlatform(gPlatformIDdetected, "clEnqueueAcquireDX9MediaSurfacesKHR");
|
||||
if (clEnqueueAcquireDX9MediaSurfacesKHR == NULL)
|
||||
{
|
||||
log_error("clGetExtensionFunctionAddressForPlatform(clEnqueueAcquireDX9MediaSurfacesKHR) returned NULL.\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
clEnqueueReleaseDX9MediaSurfacesKHR = (clEnqueueReleaseDX9MediaSurfacesKHR_fn)clGetExtensionFunctionAddressForPlatform(gPlatformIDdetected, "clEnqueueReleaseDX9MediaSurfacesKHR");
|
||||
if (clEnqueueReleaseDX9MediaSurfacesKHR == NULL)
|
||||
{
|
||||
log_error("clGetExtensionFunctionAddressForPlatform(clEnqueueReleaseDX9MediaSurfacesKHR) returned NULL.\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool DetectPlatformAndDevice()
|
||||
{
|
||||
std::vector<cl_platform_id> platforms;
|
||||
cl_uint platformsNum = 0;
|
||||
cl_int error = clGetPlatformIDs(0, 0, &platformsNum);
|
||||
if (error != CL_SUCCESS)
|
||||
{
|
||||
print_error(error, "clGetPlatformIDs failed\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
platforms.resize(platformsNum);
|
||||
error = clGetPlatformIDs(platformsNum, &platforms[0], 0);
|
||||
if (error != CL_SUCCESS)
|
||||
{
|
||||
print_error(error, "clGetPlatformIDs failed\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
bool found = false;
|
||||
for (size_t i = 0; i < platformsNum; ++i)
|
||||
{
|
||||
std::vector<cl_device_id> devices;
|
||||
cl_uint devicesNum = 0;
|
||||
error = clGetDeviceIDs(platforms[i], gDeviceTypeSelected, 0, 0, &devicesNum);
|
||||
if (error != CL_SUCCESS)
|
||||
{
|
||||
print_error(error, "clGetDeviceIDs failed\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
devices.resize(devicesNum);
|
||||
error = clGetDeviceIDs(platforms[i], gDeviceTypeSelected, devicesNum, &devices[0], 0);
|
||||
if (error != CL_SUCCESS)
|
||||
{
|
||||
print_error(error, "clGetDeviceIDs failed\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
for (size_t j = 0; j < devicesNum; ++j)
|
||||
{
|
||||
if (ExtensionCheck("cl_khr_dx9_media_sharing", devices[j]))
|
||||
{
|
||||
gPlatformIDdetected = platforms[i];
|
||||
gDeviceIDdetected = devices[j];
|
||||
found = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!found)
|
||||
{
|
||||
log_info("Test was not run, because the media surface sharing extension is not supported for any devices.\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CmdlineParse(int argc, const char *argv[])
|
||||
{
|
||||
char *env_mode = getenv( "CL_DEVICE_TYPE" );
|
||||
if( env_mode != NULL )
|
||||
{
|
||||
if(strcmp(env_mode, "gpu") == 0 || strcmp(env_mode, "CL_DEVICE_TYPE_GPU") == 0)
|
||||
gDeviceTypeSelected = CL_DEVICE_TYPE_GPU;
|
||||
else if(strcmp(env_mode, "cpu") == 0 || strcmp(env_mode, "CL_DEVICE_TYPE_CPU") == 0)
|
||||
gDeviceTypeSelected = CL_DEVICE_TYPE_CPU;
|
||||
else if(strcmp(env_mode, "accelerator") == 0 || strcmp(env_mode, "CL_DEVICE_TYPE_ACCELERATOR") == 0)
|
||||
gDeviceTypeSelected = CL_DEVICE_TYPE_ACCELERATOR;
|
||||
else if(strcmp(env_mode, "default") == 0 || strcmp(env_mode, "CL_DEVICE_TYPE_DEFAULT") == 0)
|
||||
gDeviceTypeSelected = CL_DEVICE_TYPE_DEFAULT;
|
||||
else
|
||||
{
|
||||
log_error("Unknown CL_DEVICE_TYPE env variable setting: %s.\nAborting...\n", env_mode);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
for (int i = 0; i < argc; ++i)
|
||||
{
|
||||
if(strcmp(argv[i], "gpu") == 0 || strcmp(argv[i], "CL_DEVICE_TYPE_GPU") == 0)
|
||||
{
|
||||
gDeviceTypeSelected = CL_DEVICE_TYPE_GPU;
|
||||
continue;
|
||||
}
|
||||
else if(strcmp( argv[i], "cpu") == 0 || strcmp(argv[i], "CL_DEVICE_TYPE_CPU") == 0)
|
||||
{
|
||||
gDeviceTypeSelected = CL_DEVICE_TYPE_CPU;
|
||||
continue;
|
||||
}
|
||||
else if(strcmp( argv[i], "accelerator") == 0 || strcmp(argv[i], "CL_DEVICE_TYPE_ACCELERATOR") == 0)
|
||||
{
|
||||
gDeviceTypeSelected = CL_DEVICE_TYPE_ACCELERATOR;
|
||||
continue;
|
||||
}
|
||||
else if(strcmp(argv[i], "CL_DEVICE_TYPE_DEFAULT") == 0)
|
||||
{
|
||||
gDeviceTypeSelected = CL_DEVICE_TYPE_DEFAULT;
|
||||
continue;
|
||||
}
|
||||
else if (strcmp(argv[i], "sw") == 0 || strcmp(argv[i], "software") == 0)
|
||||
{
|
||||
CDeviceWrapper::AccelerationType(CDeviceWrapper::ACCELERATION_SW);
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
int main(int argc, const char *argv[])
|
||||
{
|
||||
if (!CmdlineParse(argc, argv))
|
||||
return 2;
|
||||
|
||||
if (!DetectPlatformAndDevice())
|
||||
{
|
||||
log_info("Test was not run, because the media surface sharing extension is not supported\n");
|
||||
return TEST_NOT_SUPPORTED;
|
||||
}
|
||||
|
||||
if (!MediaSurfaceSharingExtensionInit())
|
||||
return 2;
|
||||
|
||||
return runTestHarness( argc, argv, num_fns, basefn_list, basefn_names, false, true, 0 );
|
||||
}
|
||||
Reference in New Issue
Block a user