Use CTS type wrappers for test_basic test_if. (#1540)

Signed-off-by: John Kesapides
[john.kesapides@arm.com](mailto:john.kesapides@arm.com)
This commit is contained in:
John Kesapides
2024-07-09 17:55:57 +01:00
committed by GitHub
parent cb9b8db894
commit cabdf6a5fc

View File

@@ -1,6 +1,6 @@
// //
// Copyright (c) 2017 The Khronos Group Inc. // Copyright (c) 2017 The Khronos Group Inc.
// //
// Licensed under the Apache License, Version 2.0 (the "License"); // Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. // you may not use this file except in compliance with the License.
// You may obtain a copy of the License at // You may obtain a copy of the License at
@@ -21,146 +21,119 @@
#include <sys/types.h> #include <sys/types.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <algorithm>
#include <vector>
#include "procs.h" #include "procs.h"
const char *conditional_kernel_code = namespace {
"__kernel void test_if(__global int *src, __global int *dst)\n" const char *conditional_kernel_code = R"(
"{\n" __kernel void test_if(__global int *src, __global int *dst)
" int tid = get_global_id(0);\n"
"\n"
" if (src[tid] == 0)\n"
" dst[tid] = 0x12345678;\n"
" else if (src[tid] == 1)\n"
" dst[tid] = 0x23456781;\n"
" else if (src[tid] == 2)\n"
" dst[tid] = 0x34567812;\n"
" else if (src[tid] == 3)\n"
" dst[tid] = 0x45678123;\n"
" else if (src[tid] == 4)\n"
" dst[tid] = 0x56781234;\n"
" else if (src[tid] == 5)\n"
" dst[tid] = 0x67812345;\n"
" else if (src[tid] == 6)\n"
" dst[tid] = 0x78123456;\n"
" else if (src[tid] == 7)\n"
" dst[tid] = 0x81234567;\n"
" else\n"
" dst[tid] = 0x7FFFFFFF;\n"
"\n"
"}\n";
const int results[] = {
0x12345678,
0x23456781,
0x34567812,
0x45678123,
0x56781234,
0x67812345,
0x78123456,
0x81234567,
};
int
verify_if(int *inptr, int *outptr, int n)
{ {
int r, i; int tid = get_global_id(0);
for (i=0; i<n; i++) if (src[tid] == 0)
{ dst[tid] = 0x12345678;
if (inptr[i] <= 7) else if (src[tid] == 1)
r = results[inptr[i]]; dst[tid] = 0x23456781;
else if (src[tid] == 2)
dst[tid] = 0x34567812;
else if (src[tid] == 3)
dst[tid] = 0x45678123;
else if (src[tid] == 4)
dst[tid] = 0x56781234;
else if (src[tid] == 5)
dst[tid] = 0x67812345;
else if (src[tid] == 6)
dst[tid] = 0x78123456;
else if (src[tid] == 7)
dst[tid] = 0x81234567;
else
dst[tid] = 0x7FFFFFFF;
}
)";
int verify_if(std::vector<cl_int> input, std::vector<cl_int> output)
{
const cl_int results[] = {
0x12345678, 0x23456781, 0x34567812, 0x45678123,
0x56781234, 0x67812345, 0x78123456, 0x81234567,
};
auto predicate = [&results](cl_int a, cl_int b) {
if (a <= 7)
return b == results[a];
else else
r = 0x7FFFFFFF; return b == 0x7FFFFFFF;
};
if (r != outptr[i]) if (!std::equal(input.begin(), input.end(), output.begin(), predicate))
{ {
log_error("IF test failed\n"); log_error("IF test failed\n");
return -1; return -1;
}
} }
log_info("IF test passed\n"); log_info("IF test passed\n");
return 0; return 0;
} }
int test_if(cl_device_id device, cl_context context, cl_command_queue queue, int num_elements) void generate_random_inputs(std::vector<cl_int> &v)
{ {
cl_mem streams[2]; RandomSeed seed(gRandomSeed);
cl_int *input_ptr, *output_ptr;
cl_program program; auto random_generator = [&seed]() {
cl_kernel kernel; return static_cast<cl_int>(get_random_float(0, 32, seed));
size_t threads[1]; };
int err, i;
MTdata d = init_genrand( gRandomSeed ); std::generate(v.begin(), v.end(), random_generator);
}
}
int test_if(cl_device_id device, cl_context context, cl_command_queue queue,
int num_elements)
{
clMemWrapper streams[2];
clProgramWrapper program;
clKernelWrapper kernel;
int err;
size_t length = sizeof(cl_int) * num_elements; size_t length = sizeof(cl_int) * num_elements;
input_ptr = (cl_int*)malloc(length);
output_ptr = (cl_int*)malloc(length);
streams[0] = clCreateBuffer(context, CL_MEM_READ_WRITE, length, NULL, NULL); std::vector<cl_int> input(num_elements);
if (!streams[0]) std::vector<cl_int> output(num_elements);
{
log_error("clCreateBuffer failed\n");
return -1;
}
streams[1] = clCreateBuffer(context, CL_MEM_READ_WRITE, length, NULL, NULL);
if (!streams[1])
{
log_error("clCreateBuffer failed\n");
return -1;
}
for (i=0; i<num_elements; i++)
input_ptr[i] = (int)get_random_float(0, 32, d);
free_mtdata(d); d = NULL; streams[0] =
clCreateBuffer(context, CL_MEM_READ_WRITE, length, nullptr, &err);
test_error(err, "clCreateBuffer failed.");
streams[1] =
clCreateBuffer(context, CL_MEM_READ_WRITE, length, nullptr, &err);
test_error(err, "clCreateBuffer failed.");
err = clEnqueueWriteBuffer(queue, streams[0], CL_TRUE, 0, length, input_ptr, 0, NULL, NULL); generate_random_inputs(input);
if (err != CL_SUCCESS)
{
log_error("clEnqueueWriteBuffer failed\n");
return -1;
}
err = create_single_kernel_helper(context, &program, &kernel, 1, &conditional_kernel_code, "test_if" ); err = clEnqueueWriteBuffer(queue, streams[0], CL_TRUE, 0, length,
if (err) input.data(), 0, nullptr, nullptr);
return -1; test_error(err, "clEnqueueWriteBuffer failed.");
err = clSetKernelArg(kernel, 0, sizeof streams[0], &streams[0]); err = create_single_kernel_helper(context, &program, &kernel, 1,
err |= clSetKernelArg(kernel, 1, sizeof streams[1], &streams[1]); &conditional_kernel_code, "test_if");
if (err != CL_SUCCESS) test_error(err, "create_single_kernel_helper failed.");
{
log_error("clSetKernelArgs failed\n");
return -1;
}
threads[0] = (unsigned int)num_elements; err = clSetKernelArg(kernel, 0, sizeof streams[0], &streams[0]);
err = clEnqueueNDRangeKernel(queue, kernel, 1, NULL, threads, NULL, 0, NULL, NULL); err |= clSetKernelArg(kernel, 1, sizeof streams[1], &streams[1]);
if (err != CL_SUCCESS) test_error(err, "clSetKernelArg failed.");
{
log_error("clEnqueueNDRangeKernel failed\n");
return -1;
}
err = clEnqueueReadBuffer(queue, streams[1], CL_TRUE, 0, length, output_ptr, 0, NULL, NULL); size_t threads[] = { (size_t)num_elements };
if (err != CL_SUCCESS) err = clEnqueueNDRangeKernel(queue, kernel, 1, nullptr, threads, nullptr, 0,
{ nullptr, nullptr);
log_error("clReadArray failed\n"); test_error(err, "clEnqueueNDRangeKernel failed.");
return -1;
}
err = verify_if(input_ptr, output_ptr, num_elements); err = clEnqueueReadBuffer(queue, streams[1], CL_TRUE, 0, length,
output.data(), 0, nullptr, nullptr);
test_error(err, "clEnqueueReadBuffer failed.");
// cleanup err = verify_if(input, output);
clReleaseMemObject(streams[0]);
clReleaseMemObject(streams[1]);
clReleaseKernel(kernel);
clReleaseProgram(program);
free(input_ptr);
free(output_ptr);
return err; return err;
} }