blob: 93673b35791a828f3f9af9d3325afa12af2d059c [file] [log] [blame]
//
// 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.
//
#ifndef SUBHELPERS_H
#define SUBHELPERS_H
#include "testHarness.h"
#include "kernelHelpers.h"
#include "typeWrappers.h"
#include "imageHelpers.h"
#include <limits>
#include <vector>
#include <type_traits>
#define NR_OF_ACTIVE_WORK_ITEMS 4
extern MTdata gMTdata;
struct WorkGroupParams
{
WorkGroupParams(size_t gws, size_t lws,
const std::vector<std::string> &req_ext = {},
const std::vector<uint32_t> &all_wim = {})
: global_workgroup_size(gws), local_workgroup_size(lws),
required_extensions(req_ext), all_work_item_masks(all_wim)
{
subgroup_size = 0;
work_items_mask = 0;
use_core_subgroups = true;
dynsc = 0;
}
size_t global_workgroup_size;
size_t local_workgroup_size;
size_t subgroup_size;
uint32_t work_items_mask;
int dynsc;
bool use_core_subgroups;
std::vector<std::string> required_extensions;
std::vector<uint32_t> all_work_item_masks;
};
enum class SubgroupsBroadcastOp
{
broadcast,
broadcast_first,
non_uniform_broadcast
};
enum class NonUniformVoteOp
{
elect,
all,
any,
all_equal
};
enum class BallotOp
{
ballot,
inverse_ballot,
ballot_bit_extract,
ballot_bit_count,
ballot_inclusive_scan,
ballot_exclusive_scan,
ballot_find_lsb,
ballot_find_msb,
eq_mask,
ge_mask,
gt_mask,
le_mask,
lt_mask,
};
enum class ShuffleOp
{
shuffle,
shuffle_up,
shuffle_down,
shuffle_xor
};
enum class ArithmeticOp
{
add_,
max_,
min_,
mul_,
and_,
or_,
xor_,
logical_and,
logical_or,
logical_xor
};
static const char *const operation_names(ArithmeticOp operation)
{
switch (operation)
{
case ArithmeticOp::add_: return "add";
case ArithmeticOp::max_: return "max";
case ArithmeticOp::min_: return "min";
case ArithmeticOp::mul_: return "mul";
case ArithmeticOp::and_: return "and";
case ArithmeticOp::or_: return "or";
case ArithmeticOp::xor_: return "xor";
case ArithmeticOp::logical_and: return "logical_and";
case ArithmeticOp::logical_or: return "logical_or";
case ArithmeticOp::logical_xor: return "logical_xor";
default: log_error("Unknown operation request"); break;
}
return "";
}
static const char *const operation_names(BallotOp operation)
{
switch (operation)
{
case BallotOp::ballot: return "ballot";
case BallotOp::inverse_ballot: return "inverse_ballot";
case BallotOp::ballot_bit_extract: return "bit_extract";
case BallotOp::ballot_bit_count: return "bit_count";
case BallotOp::ballot_inclusive_scan: return "inclusive_scan";
case BallotOp::ballot_exclusive_scan: return "exclusive_scan";
case BallotOp::ballot_find_lsb: return "find_lsb";
case BallotOp::ballot_find_msb: return "find_msb";
case BallotOp::eq_mask: return "eq";
case BallotOp::ge_mask: return "ge";
case BallotOp::gt_mask: return "gt";
case BallotOp::le_mask: return "le";
case BallotOp::lt_mask: return "lt";
default: log_error("Unknown operation request"); break;
}
return "";
}
static const char *const operation_names(ShuffleOp operation)
{
switch (operation)
{
case ShuffleOp::shuffle: return "shuffle";
case ShuffleOp::shuffle_up: return "shuffle_up";
case ShuffleOp::shuffle_down: return "shuffle_down";
case ShuffleOp::shuffle_xor: return "shuffle_xor";
default: log_error("Unknown operation request"); break;
}
return "";
}
static const char *const operation_names(NonUniformVoteOp operation)
{
switch (operation)
{
case NonUniformVoteOp::all: return "all";
case NonUniformVoteOp::all_equal: return "all_equal";
case NonUniformVoteOp::any: return "any";
case NonUniformVoteOp::elect: return "elect";
default: log_error("Unknown operation request"); break;
}
return "";
}
static const char *const operation_names(SubgroupsBroadcastOp operation)
{
switch (operation)
{
case SubgroupsBroadcastOp::broadcast: return "broadcast";
case SubgroupsBroadcastOp::broadcast_first: return "broadcast_first";
case SubgroupsBroadcastOp::non_uniform_broadcast:
return "non_uniform_broadcast";
default: log_error("Unknown operation request"); break;
}
return "";
}
class subgroupsAPI {
public:
subgroupsAPI(cl_platform_id platform, bool use_core_subgroups)
{
static_assert(CL_KERNEL_MAX_SUB_GROUP_SIZE_FOR_NDRANGE
== CL_KERNEL_MAX_SUB_GROUP_SIZE_FOR_NDRANGE_KHR,
"Enums have to be the same");
static_assert(CL_KERNEL_SUB_GROUP_COUNT_FOR_NDRANGE
== CL_KERNEL_SUB_GROUP_COUNT_FOR_NDRANGE_KHR,
"Enums have to be the same");
if (use_core_subgroups)
{
_clGetKernelSubGroupInfo_ptr = &clGetKernelSubGroupInfo;
clGetKernelSubGroupInfo_name = "clGetKernelSubGroupInfo";
}
else
{
_clGetKernelSubGroupInfo_ptr = (clGetKernelSubGroupInfoKHR_fn)
clGetExtensionFunctionAddressForPlatform(
platform, "clGetKernelSubGroupInfoKHR");
clGetKernelSubGroupInfo_name = "clGetKernelSubGroupInfoKHR";
}
}
clGetKernelSubGroupInfoKHR_fn clGetKernelSubGroupInfo_ptr()
{
return _clGetKernelSubGroupInfo_ptr;
}
const char *clGetKernelSubGroupInfo_name;
private:
clGetKernelSubGroupInfoKHR_fn _clGetKernelSubGroupInfo_ptr;
};
// Need to defined custom type for vector size = 3 and half type. This is
// because of 3-component types are otherwise indistinguishable from the
// 4-component types, and because the half type is indistinguishable from some
// other 16-bit type (ushort)
namespace subgroups {
struct cl_char3
{
::cl_char3 data;
};
struct cl_uchar3
{
::cl_uchar3 data;
};
struct cl_short3
{
::cl_short3 data;
};
struct cl_ushort3
{
::cl_ushort3 data;
};
struct cl_int3
{
::cl_int3 data;
};
struct cl_uint3
{
::cl_uint3 data;
};
struct cl_long3
{
::cl_long3 data;
};
struct cl_ulong3
{
::cl_ulong3 data;
};
struct cl_float3
{
::cl_float3 data;
};
struct cl_double3
{
::cl_double3 data;
};
struct cl_half
{
::cl_half data;
};
struct cl_half2
{
::cl_half2 data;
};
struct cl_half3
{
::cl_half3 data;
};
struct cl_half4
{
::cl_half4 data;
};
struct cl_half8
{
::cl_half8 data;
};
struct cl_half16
{
::cl_half16 data;
};
}
static bool int64_ok(cl_device_id device)
{
char profile[128];
int error;
error = clGetDeviceInfo(device, CL_DEVICE_PROFILE, sizeof(profile),
(void *)&profile, NULL);
if (error)
{
log_info("clGetDeviceInfo failed with CL_DEVICE_PROFILE\n");
return false;
}
if (strcmp(profile, "EMBEDDED_PROFILE") == 0)
return is_extension_available(device, "cles_khr_int64");
return true;
}
static bool double_ok(cl_device_id device)
{
int error;
cl_device_fp_config c;
error = clGetDeviceInfo(device, CL_DEVICE_DOUBLE_FP_CONFIG, sizeof(c),
(void *)&c, NULL);
if (error)
{
log_info("clGetDeviceInfo failed with CL_DEVICE_DOUBLE_FP_CONFIG\n");
return false;
}
return c != 0;
}
static bool half_ok(cl_device_id device)
{
int error;
cl_device_fp_config c;
error = clGetDeviceInfo(device, CL_DEVICE_HALF_FP_CONFIG, sizeof(c),
(void *)&c, NULL);
if (error)
{
log_info("clGetDeviceInfo failed with CL_DEVICE_HALF_FP_CONFIG\n");
return false;
}
return c != 0;
}
template <typename Ty> struct CommonTypeManager
{
static const char *name() { return ""; }
static const char *add_typedef() { return "\n"; }
typedef std::false_type is_vector_type;
typedef std::false_type is_sb_vector_size3;
typedef std::false_type is_sb_vector_type;
typedef std::false_type is_sb_scalar_type;
static const bool type_supported(cl_device_id) { return true; }
static const Ty identify_limits(ArithmeticOp operation)
{
switch (operation)
{
case ArithmeticOp::add_: return (Ty)0;
case ArithmeticOp::max_: return (std::numeric_limits<Ty>::min)();
case ArithmeticOp::min_: return (std::numeric_limits<Ty>::max)();
case ArithmeticOp::mul_: return (Ty)1;
case ArithmeticOp::and_: return (Ty)~0;
case ArithmeticOp::or_: return (Ty)0;
case ArithmeticOp::xor_: return (Ty)0;
default: log_error("Unknown operation request"); break;
}
return 0;
}
};
template <typename> struct TypeManager;
template <> struct TypeManager<cl_int> : public CommonTypeManager<cl_int>
{
static const char *name() { return "int"; }
static const char *add_typedef() { return "typedef int Type;\n"; }
static cl_int identify_limits(ArithmeticOp operation)
{
switch (operation)
{
case ArithmeticOp::add_: return (cl_int)0;
case ArithmeticOp::max_:
return (std::numeric_limits<cl_int>::min)();
case ArithmeticOp::min_:
return (std::numeric_limits<cl_int>::max)();
case ArithmeticOp::mul_: return (cl_int)1;
case ArithmeticOp::and_: return (cl_int)~0;
case ArithmeticOp::or_: return (cl_int)0;
case ArithmeticOp::xor_: return (cl_int)0;
case ArithmeticOp::logical_and: return (cl_int)1;
case ArithmeticOp::logical_or: return (cl_int)0;
case ArithmeticOp::logical_xor: return (cl_int)0;
default: log_error("Unknown operation request"); break;
}
return 0;
}
};
template <> struct TypeManager<cl_int2> : public CommonTypeManager<cl_int2>
{
static const char *name() { return "int2"; }
static const char *add_typedef() { return "typedef int2 Type;\n"; }
typedef std::true_type is_vector_type;
using scalar_type = cl_int;
};
template <>
struct TypeManager<subgroups::cl_int3>
: public CommonTypeManager<subgroups::cl_int3>
{
static const char *name() { return "int3"; }
static const char *add_typedef() { return "typedef int3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_int;
};
template <> struct TypeManager<cl_int4> : public CommonTypeManager<cl_int4>
{
static const char *name() { return "int4"; }
static const char *add_typedef() { return "typedef int4 Type;\n"; }
using scalar_type = cl_int;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_int8> : public CommonTypeManager<cl_int8>
{
static const char *name() { return "int8"; }
static const char *add_typedef() { return "typedef int8 Type;\n"; }
using scalar_type = cl_int;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_int16> : public CommonTypeManager<cl_int16>
{
static const char *name() { return "int16"; }
static const char *add_typedef() { return "typedef int16 Type;\n"; }
using scalar_type = cl_int;
typedef std::true_type is_vector_type;
};
// cl_uint
template <> struct TypeManager<cl_uint> : public CommonTypeManager<cl_uint>
{
static const char *name() { return "uint"; }
static const char *add_typedef() { return "typedef uint Type;\n"; }
};
template <> struct TypeManager<cl_uint2> : public CommonTypeManager<cl_uint2>
{
static const char *name() { return "uint2"; }
static const char *add_typedef() { return "typedef uint2 Type;\n"; }
using scalar_type = cl_uint;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<subgroups::cl_uint3>
: public CommonTypeManager<subgroups::cl_uint3>
{
static const char *name() { return "uint3"; }
static const char *add_typedef() { return "typedef uint3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_uint;
};
template <> struct TypeManager<cl_uint4> : public CommonTypeManager<cl_uint4>
{
static const char *name() { return "uint4"; }
static const char *add_typedef() { return "typedef uint4 Type;\n"; }
using scalar_type = cl_uint;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_uint8> : public CommonTypeManager<cl_uint8>
{
static const char *name() { return "uint8"; }
static const char *add_typedef() { return "typedef uint8 Type;\n"; }
using scalar_type = cl_uint;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_uint16> : public CommonTypeManager<cl_uint16>
{
static const char *name() { return "uint16"; }
static const char *add_typedef() { return "typedef uint16 Type;\n"; }
using scalar_type = cl_uint;
typedef std::true_type is_vector_type;
};
// cl_short
template <> struct TypeManager<cl_short> : public CommonTypeManager<cl_short>
{
static const char *name() { return "short"; }
static const char *add_typedef() { return "typedef short Type;\n"; }
};
template <> struct TypeManager<cl_short2> : public CommonTypeManager<cl_short2>
{
static const char *name() { return "short2"; }
static const char *add_typedef() { return "typedef short2 Type;\n"; }
using scalar_type = cl_short;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<subgroups::cl_short3>
: public CommonTypeManager<subgroups::cl_short3>
{
static const char *name() { return "short3"; }
static const char *add_typedef() { return "typedef short3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_short;
};
template <> struct TypeManager<cl_short4> : public CommonTypeManager<cl_short4>
{
static const char *name() { return "short4"; }
static const char *add_typedef() { return "typedef short4 Type;\n"; }
using scalar_type = cl_short;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_short8> : public CommonTypeManager<cl_short8>
{
static const char *name() { return "short8"; }
static const char *add_typedef() { return "typedef short8 Type;\n"; }
using scalar_type = cl_short;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<cl_short16> : public CommonTypeManager<cl_short16>
{
static const char *name() { return "short16"; }
static const char *add_typedef() { return "typedef short16 Type;\n"; }
using scalar_type = cl_short;
typedef std::true_type is_vector_type;
};
// cl_ushort
template <> struct TypeManager<cl_ushort> : public CommonTypeManager<cl_ushort>
{
static const char *name() { return "ushort"; }
static const char *add_typedef() { return "typedef ushort Type;\n"; }
};
template <>
struct TypeManager<cl_ushort2> : public CommonTypeManager<cl_ushort2>
{
static const char *name() { return "ushort2"; }
static const char *add_typedef() { return "typedef ushort2 Type;\n"; }
using scalar_type = cl_ushort;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<subgroups::cl_ushort3>
: public CommonTypeManager<subgroups::cl_ushort3>
{
static const char *name() { return "ushort3"; }
static const char *add_typedef() { return "typedef ushort3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_ushort;
};
template <>
struct TypeManager<cl_ushort4> : public CommonTypeManager<cl_ushort4>
{
static const char *name() { return "ushort4"; }
static const char *add_typedef() { return "typedef ushort4 Type;\n"; }
using scalar_type = cl_ushort;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<cl_ushort8> : public CommonTypeManager<cl_ushort8>
{
static const char *name() { return "ushort8"; }
static const char *add_typedef() { return "typedef ushort8 Type;\n"; }
using scalar_type = cl_ushort;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<cl_ushort16> : public CommonTypeManager<cl_ushort16>
{
static const char *name() { return "ushort16"; }
static const char *add_typedef() { return "typedef ushort16 Type;\n"; }
using scalar_type = cl_ushort;
typedef std::true_type is_vector_type;
};
// cl_char
template <> struct TypeManager<cl_char> : public CommonTypeManager<cl_char>
{
static const char *name() { return "char"; }
static const char *add_typedef() { return "typedef char Type;\n"; }
};
template <> struct TypeManager<cl_char2> : public CommonTypeManager<cl_char2>
{
static const char *name() { return "char2"; }
static const char *add_typedef() { return "typedef char2 Type;\n"; }
using scalar_type = cl_char;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<subgroups::cl_char3>
: public CommonTypeManager<subgroups::cl_char3>
{
static const char *name() { return "char3"; }
static const char *add_typedef() { return "typedef char3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_char;
};
template <> struct TypeManager<cl_char4> : public CommonTypeManager<cl_char4>
{
static const char *name() { return "char4"; }
static const char *add_typedef() { return "typedef char4 Type;\n"; }
using scalar_type = cl_char;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_char8> : public CommonTypeManager<cl_char8>
{
static const char *name() { return "char8"; }
static const char *add_typedef() { return "typedef char8 Type;\n"; }
using scalar_type = cl_char;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_char16> : public CommonTypeManager<cl_char16>
{
static const char *name() { return "char16"; }
static const char *add_typedef() { return "typedef char16 Type;\n"; }
using scalar_type = cl_char;
typedef std::true_type is_vector_type;
};
// cl_uchar
template <> struct TypeManager<cl_uchar> : public CommonTypeManager<cl_uchar>
{
static const char *name() { return "uchar"; }
static const char *add_typedef() { return "typedef uchar Type;\n"; }
};
template <> struct TypeManager<cl_uchar2> : public CommonTypeManager<cl_uchar2>
{
static const char *name() { return "uchar2"; }
static const char *add_typedef() { return "typedef uchar2 Type;\n"; }
using scalar_type = cl_uchar;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<subgroups::cl_uchar3>
: public CommonTypeManager<subgroups::cl_char3>
{
static const char *name() { return "uchar3"; }
static const char *add_typedef() { return "typedef uchar3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_uchar;
};
template <> struct TypeManager<cl_uchar4> : public CommonTypeManager<cl_uchar4>
{
static const char *name() { return "uchar4"; }
static const char *add_typedef() { return "typedef uchar4 Type;\n"; }
using scalar_type = cl_uchar;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_uchar8> : public CommonTypeManager<cl_uchar8>
{
static const char *name() { return "uchar8"; }
static const char *add_typedef() { return "typedef uchar8 Type;\n"; }
using scalar_type = cl_uchar;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<cl_uchar16> : public CommonTypeManager<cl_uchar16>
{
static const char *name() { return "uchar16"; }
static const char *add_typedef() { return "typedef uchar16 Type;\n"; }
using scalar_type = cl_uchar;
typedef std::true_type is_vector_type;
};
// cl_long
template <> struct TypeManager<cl_long> : public CommonTypeManager<cl_long>
{
static const char *name() { return "long"; }
static const char *add_typedef() { return "typedef long Type;\n"; }
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <> struct TypeManager<cl_long2> : public CommonTypeManager<cl_long2>
{
static const char *name() { return "long2"; }
static const char *add_typedef() { return "typedef long2 Type;\n"; }
using scalar_type = cl_long;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <>
struct TypeManager<subgroups::cl_long3>
: public CommonTypeManager<subgroups::cl_long3>
{
static const char *name() { return "long3"; }
static const char *add_typedef() { return "typedef long3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_long;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <> struct TypeManager<cl_long4> : public CommonTypeManager<cl_long4>
{
static const char *name() { return "long4"; }
static const char *add_typedef() { return "typedef long4 Type;\n"; }
using scalar_type = cl_long;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <> struct TypeManager<cl_long8> : public CommonTypeManager<cl_long8>
{
static const char *name() { return "long8"; }
static const char *add_typedef() { return "typedef long8 Type;\n"; }
using scalar_type = cl_long;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <> struct TypeManager<cl_long16> : public CommonTypeManager<cl_long16>
{
static const char *name() { return "long16"; }
static const char *add_typedef() { return "typedef long16 Type;\n"; }
using scalar_type = cl_long;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
// cl_ulong
template <> struct TypeManager<cl_ulong> : public CommonTypeManager<cl_ulong>
{
static const char *name() { return "ulong"; }
static const char *add_typedef() { return "typedef ulong Type;\n"; }
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <> struct TypeManager<cl_ulong2> : public CommonTypeManager<cl_ulong2>
{
static const char *name() { return "ulong2"; }
static const char *add_typedef() { return "typedef ulong2 Type;\n"; }
using scalar_type = cl_ulong;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <>
struct TypeManager<subgroups::cl_ulong3>
: public CommonTypeManager<subgroups::cl_ulong3>
{
static const char *name() { return "ulong3"; }
static const char *add_typedef() { return "typedef ulong3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_ulong;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <> struct TypeManager<cl_ulong4> : public CommonTypeManager<cl_ulong4>
{
static const char *name() { return "ulong4"; }
static const char *add_typedef() { return "typedef ulong4 Type;\n"; }
using scalar_type = cl_ulong;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <> struct TypeManager<cl_ulong8> : public CommonTypeManager<cl_ulong8>
{
static const char *name() { return "ulong8"; }
static const char *add_typedef() { return "typedef ulong8 Type;\n"; }
using scalar_type = cl_ulong;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
template <>
struct TypeManager<cl_ulong16> : public CommonTypeManager<cl_ulong16>
{
static const char *name() { return "ulong16"; }
static const char *add_typedef() { return "typedef ulong16 Type;\n"; }
using scalar_type = cl_ulong;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return int64_ok(device);
}
};
// cl_float
template <> struct TypeManager<cl_float> : public CommonTypeManager<cl_float>
{
static const char *name() { return "float"; }
static const char *add_typedef() { return "typedef float Type;\n"; }
static cl_float identify_limits(ArithmeticOp operation)
{
switch (operation)
{
case ArithmeticOp::add_: return 0.0f;
case ArithmeticOp::max_:
return -std::numeric_limits<float>::infinity();
case ArithmeticOp::min_:
return std::numeric_limits<float>::infinity();
case ArithmeticOp::mul_: return (cl_float)1;
default: log_error("Unknown operation request"); break;
}
return 0;
}
};
template <> struct TypeManager<cl_float2> : public CommonTypeManager<cl_float2>
{
static const char *name() { return "float2"; }
static const char *add_typedef() { return "typedef float2 Type;\n"; }
using scalar_type = cl_float;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<subgroups::cl_float3>
: public CommonTypeManager<subgroups::cl_float3>
{
static const char *name() { return "float3"; }
static const char *add_typedef() { return "typedef float3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_float;
};
template <> struct TypeManager<cl_float4> : public CommonTypeManager<cl_float4>
{
static const char *name() { return "float4"; }
static const char *add_typedef() { return "typedef float4 Type;\n"; }
using scalar_type = cl_float;
typedef std::true_type is_vector_type;
};
template <> struct TypeManager<cl_float8> : public CommonTypeManager<cl_float8>
{
static const char *name() { return "float8"; }
static const char *add_typedef() { return "typedef float8 Type;\n"; }
using scalar_type = cl_float;
typedef std::true_type is_vector_type;
};
template <>
struct TypeManager<cl_float16> : public CommonTypeManager<cl_float16>
{
static const char *name() { return "float16"; }
static const char *add_typedef() { return "typedef float16 Type;\n"; }
using scalar_type = cl_float;
typedef std::true_type is_vector_type;
};
// cl_double
template <> struct TypeManager<cl_double> : public CommonTypeManager<cl_double>
{
static const char *name() { return "double"; }
static const char *add_typedef() { return "typedef double Type;\n"; }
static cl_double identify_limits(ArithmeticOp operation)
{
switch (operation)
{
case ArithmeticOp::add_: return 0.0;
case ArithmeticOp::max_:
return -std::numeric_limits<double>::infinity();
case ArithmeticOp::min_:
return std::numeric_limits<double>::infinity();
case ArithmeticOp::mul_: return (cl_double)1;
default: log_error("Unknown operation request"); break;
}
return 0;
}
static const bool type_supported(cl_device_id device)
{
return double_ok(device);
}
};
template <>
struct TypeManager<cl_double2> : public CommonTypeManager<cl_double2>
{
static const char *name() { return "double2"; }
static const char *add_typedef() { return "typedef double2 Type;\n"; }
using scalar_type = cl_double;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return double_ok(device);
}
};
template <>
struct TypeManager<subgroups::cl_double3>
: public CommonTypeManager<subgroups::cl_double3>
{
static const char *name() { return "double3"; }
static const char *add_typedef() { return "typedef double3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = cl_double;
static const bool type_supported(cl_device_id device)
{
return double_ok(device);
}
};
template <>
struct TypeManager<cl_double4> : public CommonTypeManager<cl_double4>
{
static const char *name() { return "double4"; }
static const char *add_typedef() { return "typedef double4 Type;\n"; }
using scalar_type = cl_double;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return double_ok(device);
}
};
template <>
struct TypeManager<cl_double8> : public CommonTypeManager<cl_double8>
{
static const char *name() { return "double8"; }
static const char *add_typedef() { return "typedef double8 Type;\n"; }
using scalar_type = cl_double;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return double_ok(device);
}
};
template <>
struct TypeManager<cl_double16> : public CommonTypeManager<cl_double16>
{
static const char *name() { return "double16"; }
static const char *add_typedef() { return "typedef double16 Type;\n"; }
using scalar_type = cl_double;
typedef std::true_type is_vector_type;
static const bool type_supported(cl_device_id device)
{
return double_ok(device);
}
};
// cl_half
template <>
struct TypeManager<subgroups::cl_half>
: public CommonTypeManager<subgroups::cl_half>
{
static const char *name() { return "half"; }
static const char *add_typedef() { return "typedef half Type;\n"; }
typedef std::true_type is_sb_scalar_type;
static subgroups::cl_half identify_limits(ArithmeticOp operation)
{
switch (operation)
{
case ArithmeticOp::add_: return { 0x0000 };
case ArithmeticOp::max_: return { 0xfc00 };
case ArithmeticOp::min_: return { 0x7c00 };
case ArithmeticOp::mul_: return { 0x3c00 };
default: log_error("Unknown operation request"); break;
}
return { 0 };
}
static const bool type_supported(cl_device_id device)
{
return half_ok(device);
}
};
template <>
struct TypeManager<subgroups::cl_half2>
: public CommonTypeManager<subgroups::cl_half2>
{
static const char *name() { return "half2"; }
static const char *add_typedef() { return "typedef half2 Type;\n"; }
using scalar_type = subgroups::cl_half;
typedef std::true_type is_sb_vector_type;
static const bool type_supported(cl_device_id device)
{
return half_ok(device);
}
};
template <>
struct TypeManager<subgroups::cl_half3>
: public CommonTypeManager<subgroups::cl_half3>
{
static const char *name() { return "half3"; }
static const char *add_typedef() { return "typedef half3 Type;\n"; }
typedef std::true_type is_sb_vector_size3;
using scalar_type = subgroups::cl_half;
static const bool type_supported(cl_device_id device)
{
return half_ok(device);
}
};
template <>
struct TypeManager<subgroups::cl_half4>
: public CommonTypeManager<subgroups::cl_half4>
{
static const char *name() { return "half4"; }
static const char *add_typedef() { return "typedef half4 Type;\n"; }
using scalar_type = subgroups::cl_half;
typedef std::true_type is_sb_vector_type;
static const bool type_supported(cl_device_id device)
{
return half_ok(device);
}
};
template <>
struct TypeManager<subgroups::cl_half8>
: public CommonTypeManager<subgroups::cl_half8>
{
static const char *name() { return "half8"; }
static const char *add_typedef() { return "typedef half8 Type;\n"; }
using scalar_type = subgroups::cl_half;
typedef std::true_type is_sb_vector_type;
static const bool type_supported(cl_device_id device)
{
return half_ok(device);
}
};
template <>
struct TypeManager<subgroups::cl_half16>
: public CommonTypeManager<subgroups::cl_half16>
{
static const char *name() { return "half16"; }
static const char *add_typedef() { return "typedef half16 Type;\n"; }
using scalar_type = subgroups::cl_half;
typedef std::true_type is_sb_vector_type;
static const bool type_supported(cl_device_id device)
{
return half_ok(device);
}
};
// set scalar value to vector of halfs
template <typename Ty, int N = 0>
typename std::enable_if<TypeManager<Ty>::is_sb_vector_type::value>::type
set_value(Ty &lhs, const cl_ulong &rhs)
{
const int size = sizeof(Ty) / sizeof(typename TypeManager<Ty>::scalar_type);
for (auto i = 0; i < size; ++i)
{
lhs.data.s[i] = rhs;
}
}
// set scalar value to vector
template <typename Ty>
typename std::enable_if<TypeManager<Ty>::is_vector_type::value>::type
set_value(Ty &lhs, const cl_ulong &rhs)
{
const int size = sizeof(Ty) / sizeof(typename TypeManager<Ty>::scalar_type);
for (auto i = 0; i < size; ++i)
{
lhs.s[i] = rhs;
}
}
// set vector to vector value
template <typename Ty>
typename std::enable_if<TypeManager<Ty>::is_vector_type::value>::type
set_value(Ty &lhs, const Ty &rhs)
{
lhs = rhs;
}
// set scalar value to vector size 3
template <typename Ty, int N = 0>
typename std::enable_if<TypeManager<Ty>::is_sb_vector_size3::value>::type
set_value(Ty &lhs, const cl_ulong &rhs)
{
for (auto i = 0; i < 3; ++i)
{
lhs.data.s[i] = rhs;
}
}
// set scalar value to scalar
template <typename Ty>
typename std::enable_if<std::is_scalar<Ty>::value>::type
set_value(Ty &lhs, const cl_ulong &rhs)
{
lhs = static_cast<Ty>(rhs);
}
// set scalar value to half scalar
template <typename Ty>
typename std::enable_if<TypeManager<Ty>::is_sb_scalar_type::value>::type
set_value(Ty &lhs, const cl_ulong &rhs)
{
lhs.data = rhs;
}
// compare for common vectors
template <typename Ty>
typename std::enable_if<TypeManager<Ty>::is_vector_type::value, bool>::type
compare(const Ty &lhs, const Ty &rhs)
{
const int size = sizeof(Ty) / sizeof(typename TypeManager<Ty>::scalar_type);
for (auto i = 0; i < size; ++i)
{
if (lhs.s[i] != rhs.s[i])
{
return false;
}
}
return true;
}
// compare for vectors 3
template <typename Ty>
typename std::enable_if<TypeManager<Ty>::is_sb_vector_size3::value, bool>::type
compare(const Ty &lhs, const Ty &rhs)
{
for (auto i = 0; i < 3; ++i)
{
if (lhs.data.s[i] != rhs.data.s[i])
{
return false;
}
}
return true;
}
// compare for half vectors
template <typename Ty>
typename std::enable_if<TypeManager<Ty>::is_sb_vector_type::value, bool>::type
compare(const Ty &lhs, const Ty &rhs)
{
const int size = sizeof(Ty) / sizeof(typename TypeManager<Ty>::scalar_type);
for (auto i = 0; i < size; ++i)
{
if (lhs.data.s[i] != rhs.data.s[i])
{
return false;
}
}
return true;
}
// compare for scalars
template <typename Ty>
typename std::enable_if<std::is_scalar<Ty>::value, bool>::type
compare(const Ty &lhs, const Ty &rhs)
{
return lhs == rhs;
}
// compare for scalar halfs
template <typename Ty>
typename std::enable_if<TypeManager<Ty>::is_sb_scalar_type::value, bool>::type
compare(const Ty &lhs, const Ty &rhs)
{
return lhs.data == rhs.data;
}
template <typename Ty> inline bool compare_ordered(const Ty &lhs, const Ty &rhs)
{
return lhs == rhs;
}
template <>
inline bool compare_ordered(const subgroups::cl_half &lhs,
const subgroups::cl_half &rhs)
{
return cl_half_to_float(lhs.data) == cl_half_to_float(rhs.data);
}
template <typename Ty>
inline bool compare_ordered(const subgroups::cl_half &lhs, const int &rhs)
{
return cl_half_to_float(lhs.data) == rhs;
}
// Run a test kernel to compute the result of a built-in on an input
static int run_kernel(cl_context context, cl_command_queue queue,
cl_kernel kernel, size_t global, size_t local,
void *idata, size_t isize, void *mdata, size_t msize,
void *odata, size_t osize, size_t tsize = 0)
{
clMemWrapper in;
clMemWrapper xy;
clMemWrapper out;
clMemWrapper tmp;
int error;
in = clCreateBuffer(context, CL_MEM_READ_ONLY, isize, NULL, &error);
test_error(error, "clCreateBuffer failed");
xy = clCreateBuffer(context, CL_MEM_WRITE_ONLY, msize, NULL, &error);
test_error(error, "clCreateBuffer failed");
out = clCreateBuffer(context, CL_MEM_WRITE_ONLY, osize, NULL, &error);
test_error(error, "clCreateBuffer failed");
if (tsize)
{
tmp = clCreateBuffer(context, CL_MEM_READ_WRITE | CL_MEM_HOST_NO_ACCESS,
tsize, NULL, &error);
test_error(error, "clCreateBuffer failed");
}
error = clSetKernelArg(kernel, 0, sizeof(in), (void *)&in);
test_error(error, "clSetKernelArg failed");
error = clSetKernelArg(kernel, 1, sizeof(xy), (void *)&xy);
test_error(error, "clSetKernelArg failed");
error = clSetKernelArg(kernel, 2, sizeof(out), (void *)&out);
test_error(error, "clSetKernelArg failed");
if (tsize)
{
error = clSetKernelArg(kernel, 3, sizeof(tmp), (void *)&tmp);
test_error(error, "clSetKernelArg failed");
}
error = clEnqueueWriteBuffer(queue, in, CL_FALSE, 0, isize, idata, 0, NULL,
NULL);
test_error(error, "clEnqueueWriteBuffer failed");
error = clEnqueueWriteBuffer(queue, xy, CL_FALSE, 0, msize, mdata, 0, NULL,
NULL);
test_error(error, "clEnqueueWriteBuffer failed");
error = clEnqueueNDRangeKernel(queue, kernel, 1, NULL, &global, &local, 0,
NULL, NULL);
test_error(error, "clEnqueueNDRangeKernel failed");
error = clEnqueueReadBuffer(queue, xy, CL_FALSE, 0, msize, mdata, 0, NULL,
NULL);
test_error(error, "clEnqueueReadBuffer failed");
error = clEnqueueReadBuffer(queue, out, CL_FALSE, 0, osize, odata, 0, NULL,
NULL);
test_error(error, "clEnqueueReadBuffer failed");
error = clFinish(queue);
test_error(error, "clFinish failed");
return error;
}
// Driver for testing a single built in function
template <typename Ty, typename Fns, size_t TSIZE = 0> struct test
{
static int mrun(cl_device_id device, cl_context context,
cl_command_queue queue, int num_elements, const char *kname,
const char *src, WorkGroupParams test_params)
{
int error = TEST_PASS;
for (auto &mask : test_params.all_work_item_masks)
{
test_params.work_items_mask = mask;
error |= run(device, context, queue, num_elements, kname, src,
test_params);
}
return error;
};
static int run(cl_device_id device, cl_context context,
cl_command_queue queue, int num_elements, const char *kname,
const char *src, WorkGroupParams test_params)
{
size_t tmp;
int error;
int subgroup_size, num_subgroups;
size_t realSize;
size_t global = test_params.global_workgroup_size;
size_t local = test_params.local_workgroup_size;
clProgramWrapper program;
clKernelWrapper kernel;
cl_platform_id platform;
std::vector<cl_int> sgmap;
sgmap.resize(4 * global);
std::vector<Ty> mapin;
mapin.resize(local);
std::vector<Ty> mapout;
mapout.resize(local);
std::stringstream kernel_sstr;
if (test_params.work_items_mask != 0)
{
kernel_sstr << "#define WORK_ITEMS_MASK ";
kernel_sstr << "0x" << std::hex << test_params.work_items_mask
<< "\n";
}
kernel_sstr << "#define NR_OF_ACTIVE_WORK_ITEMS ";
kernel_sstr << NR_OF_ACTIVE_WORK_ITEMS << "\n";
// Make sure a test of type Ty is supported by the device
if (!TypeManager<Ty>::type_supported(device))
{
log_info("Data type not supported : %s\n", TypeManager<Ty>::name());
return 0;
}
else
{
if (strstr(TypeManager<Ty>::name(), "double"))
{
kernel_sstr << "#pragma OPENCL EXTENSION cl_khr_fp64: enable\n";
}
else if (strstr(TypeManager<Ty>::name(), "half"))
{
kernel_sstr << "#pragma OPENCL EXTENSION cl_khr_fp16: enable\n";
}
}
for (std::string extension : test_params.required_extensions)
{
if (!is_extension_available(device, extension.c_str()))
{
log_info("The extension %s not supported on this device. SKIP "
"testing - kernel %s data type %s\n",
extension.c_str(), kname, TypeManager<Ty>::name());
return TEST_PASS;
}
kernel_sstr << "#pragma OPENCL EXTENSION " + extension
+ ": enable\n";
}
error = clGetDeviceInfo(device, CL_DEVICE_PLATFORM, sizeof(platform),
(void *)&platform, NULL);
test_error(error, "clGetDeviceInfo failed for CL_DEVICE_PLATFORM");
if (test_params.use_core_subgroups)
{
kernel_sstr
<< "#pragma OPENCL EXTENSION cl_khr_subgroups : enable\n";
}
kernel_sstr << "#define XY(M,I) M[I].x = get_sub_group_local_id(); "
"M[I].y = get_sub_group_id();\n";
kernel_sstr << TypeManager<Ty>::add_typedef();
kernel_sstr << src;
const std::string &kernel_str = kernel_sstr.str();
const char *kernel_src = kernel_str.c_str();
error = create_single_kernel_helper(context, &program, &kernel, 1,
&kernel_src, kname);
if (error != 0) return error;
// Determine some local dimensions to use for the test.
error = get_max_common_work_group_size(
context, kernel, test_params.global_workgroup_size, &local);
test_error(error, "get_max_common_work_group_size failed");
// Limit it a bit so we have muliple work groups
// Ideally this will still be large enough to give us multiple
if (local > test_params.local_workgroup_size)
local = test_params.local_workgroup_size;
// Get the sub group info
subgroupsAPI subgroupsApiSet(platform, test_params.use_core_subgroups);
clGetKernelSubGroupInfoKHR_fn clGetKernelSubGroupInfo_ptr =
subgroupsApiSet.clGetKernelSubGroupInfo_ptr();
if (clGetKernelSubGroupInfo_ptr == NULL)
{
log_error("ERROR: %s function not available",
subgroupsApiSet.clGetKernelSubGroupInfo_name);
return TEST_FAIL;
}
error = clGetKernelSubGroupInfo_ptr(
kernel, device, CL_KERNEL_MAX_SUB_GROUP_SIZE_FOR_NDRANGE,
sizeof(local), (void *)&local, sizeof(tmp), (void *)&tmp, NULL);
if (error != CL_SUCCESS)
{
log_error("ERROR: %s function error for "
"CL_KERNEL_MAX_SUB_GROUP_SIZE_FOR_NDRANGE",
subgroupsApiSet.clGetKernelSubGroupInfo_name);
return TEST_FAIL;
}
subgroup_size = (int)tmp;
error = clGetKernelSubGroupInfo_ptr(
kernel, device, CL_KERNEL_SUB_GROUP_COUNT_FOR_NDRANGE,
sizeof(local), (void *)&local, sizeof(tmp), (void *)&tmp, NULL);
if (error != CL_SUCCESS)
{
log_error("ERROR: %s function error for "
"CL_KERNEL_SUB_GROUP_COUNT_FOR_NDRANGE",
subgroupsApiSet.clGetKernelSubGroupInfo_name);
return TEST_FAIL;
}
num_subgroups = (int)tmp;
// Make sure the number of sub groups is what we expect
if (num_subgroups != (local + subgroup_size - 1) / subgroup_size)
{
log_error("ERROR: unexpected number of subgroups (%d) returned\n",
num_subgroups);
return TEST_FAIL;
}
std::vector<Ty> idata;
std::vector<Ty> odata;
size_t input_array_size = global;
size_t output_array_size = global;
int dynscl = test_params.dynsc;
if (dynscl != 0)
{
input_array_size =
(int)global / (int)local * num_subgroups * dynscl;
output_array_size = (int)global / (int)local * dynscl;
}
idata.resize(input_array_size);
odata.resize(output_array_size);
// Run the kernel once on zeroes to get the map
memset(idata.data(), 0, input_array_size * sizeof(Ty));
error = run_kernel(context, queue, kernel, global, local, idata.data(),
input_array_size * sizeof(Ty), sgmap.data(),
global * sizeof(cl_int4), odata.data(),
output_array_size * sizeof(Ty), TSIZE * sizeof(Ty));
test_error(error, "Running kernel first time failed");
// Generate the desired input for the kernel
test_params.subgroup_size = subgroup_size;
Fns::gen(idata.data(), mapin.data(), sgmap.data(), test_params);
error = run_kernel(context, queue, kernel, global, local, idata.data(),
input_array_size * sizeof(Ty), sgmap.data(),
global * sizeof(cl_int4), odata.data(),
output_array_size * sizeof(Ty), TSIZE * sizeof(Ty));
test_error(error, "Running kernel second time failed");
// Check the result
error = Fns::chk(idata.data(), odata.data(), mapin.data(),
mapout.data(), sgmap.data(), test_params);
test_error(error, "Data verification failed");
return TEST_PASS;
}
};
static void set_last_workgroup_params(int non_uniform_size,
int &number_of_subgroups,
int subgroup_size, int &workgroup_size,
int &last_subgroup_size)
{
number_of_subgroups = 1 + non_uniform_size / subgroup_size;
last_subgroup_size = non_uniform_size % subgroup_size;
workgroup_size = non_uniform_size;
}
template <typename Ty>
static void set_randomdata_for_subgroup(Ty *workgroup, int wg_offset,
int current_sbs)
{
int randomize_data = (int)(genrand_int32(gMTdata) % 3);
// Initialize data matrix indexed by local id and sub group id
switch (randomize_data)
{
case 0:
memset(&workgroup[wg_offset], 0, current_sbs * sizeof(Ty));
break;
case 1: {
memset(&workgroup[wg_offset], 0, current_sbs * sizeof(Ty));
int wi_id = (int)(genrand_int32(gMTdata) % (cl_uint)current_sbs);
set_value(workgroup[wg_offset + wi_id], 41);
}
break;
case 2:
memset(&workgroup[wg_offset], 0xff, current_sbs * sizeof(Ty));
break;
}
}
struct RunTestForType
{
RunTestForType(cl_device_id device, cl_context context,
cl_command_queue queue, int num_elements,
WorkGroupParams test_params)
: device_(device), context_(context), queue_(queue),
num_elements_(num_elements), test_params_(test_params)
{}
template <typename T, typename U>
int run_impl(const char *kernel_name, const char *source)
{
int error = TEST_PASS;
if (test_params_.all_work_item_masks.size() > 0)
{
error = test<T, U>::mrun(device_, context_, queue_, num_elements_,
kernel_name, source, test_params_);
}
else
{
error = test<T, U>::run(device_, context_, queue_, num_elements_,
kernel_name, source, test_params_);
}
return error;
}
private:
cl_device_id device_;
cl_context context_;
cl_command_queue queue_;
int num_elements_;
WorkGroupParams test_params_;
};
#endif