| /****************************************************************** |
| Copyright (c) 2016 The Khronos Group Inc. All Rights Reserved. |
| |
| This code is protected by copyright laws and contains material proprietary to the Khronos Group, Inc. |
| This is UNPUBLISHED PROPRIETARY SOURCE CODE that may not be disclosed in whole or in part to |
| third parties, and may not be reproduced, republished, distributed, transmitted, displayed, |
| broadcast or otherwise exploited in any manner without the express prior written permission |
| of Khronos Group. The receipt or possession of this code does not convey any rights to reproduce, |
| disclose, or distribute its contents, or to manufacture, use, or sell anything that it may describe, |
| in whole or in part other than under the terms of the Khronos Adopters Agreement |
| or Khronos Conformance Test Source License Agreement as executed between Khronos and the recipient. |
| ******************************************************************/ |
| |
| #include "testBase.h" |
| #include "types.hpp" |
| |
| #include <sstream> |
| #include <string> |
| |
| |
| template<typename T> |
| int test_selection_merge(cl_device_id deviceID, |
| cl_context context, |
| cl_command_queue queue, |
| const char *name, |
| const std::vector<T> &h_in, |
| const std::vector<T> &h_ref, |
| const int rep) |
| { |
| |
| cl_int err = CL_SUCCESS; |
| int num = (int)h_ref.size(); |
| size_t bytes = num * sizeof(T); |
| size_t in_bytes = rep * bytes; |
| |
| clMemWrapper in = clCreateBuffer(context, CL_MEM_READ_ONLY, in_bytes, NULL, &err); |
| SPIRV_CHECK_ERROR(err, "Failed to create in buffer"); |
| |
| err = clEnqueueWriteBuffer(queue, in, CL_TRUE, 0, in_bytes, &h_in[0], 0, NULL, NULL); |
| SPIRV_CHECK_ERROR(err, "Failed to copy to in buffer"); |
| |
| clProgramWrapper prog; |
| err = get_program_with_il(prog, deviceID, context, name); |
| SPIRV_CHECK_ERROR(err, "Failed to build program"); |
| |
| clKernelWrapper kernel = clCreateKernel(prog, name, &err); |
| SPIRV_CHECK_ERROR(err, "Failed to create spv kernel"); |
| |
| clMemWrapper out = clCreateBuffer(context, CL_MEM_READ_WRITE, bytes, NULL, &err); |
| SPIRV_CHECK_ERROR(err, "Failed to create out buffer"); |
| |
| err = clSetKernelArg(kernel, 0, sizeof(cl_mem), &out); |
| SPIRV_CHECK_ERROR(err, "Failed to set arg 0"); |
| |
| err = clSetKernelArg(kernel, 1, sizeof(cl_mem), &in); |
| SPIRV_CHECK_ERROR(err, "Failed to set arg 1"); |
| |
| err = clSetKernelArg(kernel, 2, sizeof(int), &rep); |
| SPIRV_CHECK_ERROR(err, "Failed to set arg 2"); |
| |
| err = clSetKernelArg(kernel, 3, sizeof(int), &num); |
| SPIRV_CHECK_ERROR(err, "Failed to set arg 3"); |
| |
| size_t global = num; |
| err = clEnqueueNDRangeKernel(queue, kernel, 1, NULL, &global, NULL, 0, NULL, NULL); |
| SPIRV_CHECK_ERROR(err, "Failed to enqueue cl kernel"); |
| |
| std::vector<T> h_out(num); |
| err = clEnqueueReadBuffer(queue, out, CL_TRUE, 0, bytes, &h_out[0], 0, NULL, NULL); |
| SPIRV_CHECK_ERROR(err, "Failed to read from ref"); |
| |
| for (int i = 0; i < num; i++) { |
| if (h_out[i] != h_ref[i]) { |
| log_error("Values do not match at location %d\n", i); |
| return -1; |
| } |
| } |
| return 0; |
| } |
| |
| #define TEST_LOOP_BRANCH(control) \ |
| TEST_SPIRV_FUNC(op_loop_merge_branch_##control) \ |
| { \ |
| const int num = 1 << 10; \ |
| RandomSeed seed(gRandomSeed); \ |
| \ |
| int rep = 4; \ |
| std::vector<cl_int> in(rep * num); \ |
| std::vector<cl_int> out(num); \ |
| \ |
| for (int i = 0; i < num; i++) { \ |
| int res = 0; \ |
| for (int j = 0; j < rep; j++) { \ |
| cl_int val = genrand<cl_int>(seed) % 1024; \ |
| res += val; \ |
| in[j * num + i] = val; \ |
| } \ |
| out[i] = res; \ |
| } \ |
| \ |
| return test_selection_merge(deviceID, context, queue, \ |
| "loop_merge_branch_" #control, \ |
| in, out, rep); \ |
| } \ |
| |
| TEST_LOOP_BRANCH(none) |
| TEST_LOOP_BRANCH(unroll) |
| TEST_LOOP_BRANCH(dont_unroll) |
| |
| |
| TEST_LOOP_BRANCH(conditional_none) |
| TEST_LOOP_BRANCH(conditional_unroll) |
| TEST_LOOP_BRANCH(conditional_dont_unroll) |