2015-12-04 14:47:52 +00:00
|
|
|
/**
|
2016-09-11 20:20:15 +00:00
|
|
|
* Author......: See docs/credits.txt
|
2015-12-04 14:47:52 +00:00
|
|
|
* License.....: MIT
|
|
|
|
*/
|
|
|
|
|
2019-03-23 21:15:38 +00:00
|
|
|
#ifndef _INC_VENDOR_H
|
|
|
|
#define _INC_VENDOR_H
|
|
|
|
|
2019-04-25 12:45:17 +00:00
|
|
|
#if defined _CPU_OPENCL_EMU_H
|
|
|
|
#define IS_NATIVE
|
|
|
|
#elif defined __CUDACC__
|
|
|
|
#define IS_CUDA
|
2021-07-09 03:50:40 +00:00
|
|
|
#elif defined __HIPCC__
|
|
|
|
#define IS_HIP
|
2022-02-04 18:54:00 +00:00
|
|
|
#elif defined __METAL_MACOS__
|
|
|
|
#define IS_METAL
|
2019-04-25 12:45:17 +00:00
|
|
|
#else
|
|
|
|
#define IS_OPENCL
|
|
|
|
#endif
|
|
|
|
|
2022-02-04 18:54:00 +00:00
|
|
|
#if defined IS_METAL
|
|
|
|
#include <metal_stdlib>
|
|
|
|
|
|
|
|
using namespace metal;
|
|
|
|
#endif
|
|
|
|
|
2019-04-25 12:45:17 +00:00
|
|
|
#if defined IS_NATIVE
|
2019-05-06 12:34:16 +00:00
|
|
|
#define CONSTANT_VK
|
2019-03-22 21:27:58 +00:00
|
|
|
#define CONSTANT_AS
|
|
|
|
#define GLOBAL_AS
|
2019-05-07 07:01:32 +00:00
|
|
|
#define LOCAL_VK
|
2019-03-22 21:27:58 +00:00
|
|
|
#define LOCAL_AS
|
2022-02-04 18:54:00 +00:00
|
|
|
#define PRIVATE_AS
|
2019-03-22 21:27:58 +00:00
|
|
|
#define KERNEL_FQ
|
2021-07-11 10:38:59 +00:00
|
|
|
#elif defined IS_CUDA
|
|
|
|
#define CONSTANT_VK __constant__
|
|
|
|
#define CONSTANT_AS
|
|
|
|
#define GLOBAL_AS
|
|
|
|
#define LOCAL_VK __shared__
|
|
|
|
#define LOCAL_AS
|
2022-02-04 18:54:00 +00:00
|
|
|
#define PRIVATE_AS
|
2021-07-11 10:38:59 +00:00
|
|
|
#define KERNEL_FQ extern "C" __global__
|
|
|
|
#elif defined IS_HIP
|
2019-05-06 12:34:16 +00:00
|
|
|
#define CONSTANT_VK __constant__
|
2019-04-25 12:45:17 +00:00
|
|
|
#define CONSTANT_AS
|
|
|
|
#define GLOBAL_AS
|
2019-05-07 07:01:32 +00:00
|
|
|
#define LOCAL_VK __shared__
|
2019-04-25 12:45:17 +00:00
|
|
|
#define LOCAL_AS
|
2022-02-04 18:54:00 +00:00
|
|
|
#define PRIVATE_AS
|
2019-05-03 13:50:07 +00:00
|
|
|
#define KERNEL_FQ extern "C" __global__
|
2022-02-04 18:54:00 +00:00
|
|
|
#elif defined IS_METAL
|
|
|
|
#define CONSTANT_VK constant
|
|
|
|
#define CONSTANT_AS constant
|
|
|
|
#define GLOBAL_AS device
|
|
|
|
#define LOCAL_VK threadgroup
|
|
|
|
#define LOCAL_AS threadgroup
|
|
|
|
#define PRIVATE_AS thread
|
|
|
|
#define KERNEL_FQ kernel
|
2019-04-25 12:45:17 +00:00
|
|
|
#elif defined IS_OPENCL
|
2019-05-06 12:34:16 +00:00
|
|
|
#define CONSTANT_VK __constant
|
2019-03-22 21:27:58 +00:00
|
|
|
#define CONSTANT_AS __constant
|
|
|
|
#define GLOBAL_AS __global
|
2019-05-07 07:01:32 +00:00
|
|
|
#define LOCAL_VK __local
|
2019-03-22 21:27:58 +00:00
|
|
|
#define LOCAL_AS __local
|
2022-02-04 18:54:00 +00:00
|
|
|
#define PRIVATE_AS
|
2019-03-22 21:27:58 +00:00
|
|
|
#define KERNEL_FQ __kernel
|
|
|
|
#endif
|
|
|
|
|
2019-04-04 18:01:37 +00:00
|
|
|
#ifndef MAYBE_UNUSED
|
|
|
|
#define MAYBE_UNUSED
|
|
|
|
#endif
|
|
|
|
|
2016-05-09 19:32:12 +00:00
|
|
|
/**
|
|
|
|
* device type
|
|
|
|
*/
|
|
|
|
|
|
|
|
#define DEVICE_TYPE_CPU 2
|
|
|
|
#define DEVICE_TYPE_GPU 4
|
|
|
|
#define DEVICE_TYPE_ACCEL 8
|
|
|
|
|
|
|
|
#if DEVICE_TYPE == DEVICE_TYPE_CPU
|
|
|
|
#define IS_CPU
|
|
|
|
#elif DEVICE_TYPE == DEVICE_TYPE_GPU
|
|
|
|
#define IS_GPU
|
|
|
|
#elif DEVICE_TYPE == DEVICE_TYPE_ACCEL
|
|
|
|
#define IS_ACCEL
|
2016-04-20 19:19:15 +00:00
|
|
|
#endif
|
|
|
|
|
2015-12-04 14:47:52 +00:00
|
|
|
/**
|
|
|
|
* vendor specific
|
|
|
|
*/
|
|
|
|
|
2016-05-09 19:32:12 +00:00
|
|
|
#if VENDOR_ID == (1 << 0)
|
2015-12-04 14:47:52 +00:00
|
|
|
#define IS_AMD
|
2016-05-14 17:45:51 +00:00
|
|
|
#elif VENDOR_ID == (1 << 1)
|
|
|
|
#define IS_APPLE
|
|
|
|
#define IS_GENERIC
|
|
|
|
#elif VENDOR_ID == (1 << 2)
|
|
|
|
#define IS_INTEL_BEIGNET
|
|
|
|
#define IS_GENERIC
|
|
|
|
#elif VENDOR_ID == (1 << 3)
|
|
|
|
#define IS_INTEL_SDK
|
|
|
|
#define IS_GENERIC
|
|
|
|
#elif VENDOR_ID == (1 << 4)
|
|
|
|
#define IS_MESA
|
|
|
|
#define IS_GENERIC
|
|
|
|
#elif VENDOR_ID == (1 << 5)
|
2021-07-11 10:38:59 +00:00
|
|
|
#define IS_NV
|
2016-05-14 17:45:51 +00:00
|
|
|
#elif VENDOR_ID == (1 << 6)
|
|
|
|
#define IS_POCL
|
|
|
|
#define IS_GENERIC
|
2021-07-11 10:38:59 +00:00
|
|
|
#elif VENDOR_ID == (1 << 8)
|
|
|
|
#define IS_AMD_USE_HIP
|
2016-01-13 16:10:40 +00:00
|
|
|
#else
|
2016-01-07 19:14:34 +00:00
|
|
|
#define IS_GENERIC
|
2016-01-04 12:17:20 +00:00
|
|
|
#endif
|
|
|
|
|
2020-01-25 11:09:39 +00:00
|
|
|
#if defined IS_AMD && HAS_VPERM == 1
|
|
|
|
#define IS_ROCM
|
|
|
|
#endif
|
|
|
|
|
2018-11-20 09:06:34 +00:00
|
|
|
#define LOCAL_MEM_TYPE_LOCAL 1
|
|
|
|
#define LOCAL_MEM_TYPE_GLOBAL 2
|
|
|
|
|
|
|
|
#if LOCAL_MEM_TYPE == LOCAL_MEM_TYPE_LOCAL
|
2018-08-13 10:10:03 +00:00
|
|
|
#define REAL_SHM
|
|
|
|
#endif
|
|
|
|
|
2020-03-20 15:20:22 +00:00
|
|
|
// So far, only used by -m 22100 and only affects NVIDIA on OpenCL. CUDA seems to work fine.
|
|
|
|
#ifdef FORCE_DISABLE_SHM
|
|
|
|
#undef REAL_SHM
|
|
|
|
#endif
|
|
|
|
|
2018-08-13 10:10:03 +00:00
|
|
|
#ifdef REAL_SHM
|
2019-03-22 21:27:58 +00:00
|
|
|
#define SHM_TYPE LOCAL_AS
|
2018-08-13 10:10:03 +00:00
|
|
|
#else
|
2019-03-22 21:27:58 +00:00
|
|
|
#define SHM_TYPE CONSTANT_AS
|
2018-08-13 10:10:03 +00:00
|
|
|
#endif
|
|
|
|
|
2018-02-06 18:12:24 +00:00
|
|
|
/**
|
|
|
|
* function declarations can have a large influence depending on the opencl runtime
|
2019-04-13 16:46:19 +00:00
|
|
|
* fast but pure kernels on rocm is a good example
|
2018-02-06 18:12:24 +00:00
|
|
|
*/
|
|
|
|
|
2020-03-03 11:36:55 +00:00
|
|
|
#if defined IS_AMD && defined IS_GPU
|
2021-07-11 10:38:59 +00:00
|
|
|
#define DECLSPEC inline static
|
|
|
|
#elif defined IS_HIP
|
2021-11-02 07:12:13 +00:00
|
|
|
#define DECLSPEC __device__
|
2020-03-03 11:36:55 +00:00
|
|
|
#else
|
|
|
|
#define DECLSPEC
|
2018-02-06 21:05:15 +00:00
|
|
|
#endif
|
2018-02-06 18:12:24 +00:00
|
|
|
|
2021-12-24 15:40:43 +00:00
|
|
|
#define HC_INLINE0 __attribute__ ((noinline))
|
|
|
|
#define HC_INLINE1 __attribute__ ((inline))
|
2021-10-31 09:05:58 +00:00
|
|
|
|
2015-12-04 14:47:52 +00:00
|
|
|
/**
|
|
|
|
* AMD specific
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifdef IS_AMD
|
2018-07-22 10:20:20 +00:00
|
|
|
#if defined(cl_amd_media_ops)
|
2016-01-04 12:17:20 +00:00
|
|
|
#pragma OPENCL EXTENSION cl_amd_media_ops : enable
|
2018-07-22 10:20:20 +00:00
|
|
|
#endif
|
|
|
|
#if defined(cl_amd_media_ops2)
|
2015-12-04 14:47:52 +00:00
|
|
|
#pragma OPENCL EXTENSION cl_amd_media_ops2 : enable
|
|
|
|
#endif
|
2018-07-22 10:20:20 +00:00
|
|
|
#endif
|
2015-12-04 14:47:52 +00:00
|
|
|
|
2020-01-12 12:32:02 +00:00
|
|
|
// Whitelist some OpenCL specific functions
|
|
|
|
// This could create more stable kernels on systems with bad OpenCL drivers
|
|
|
|
|
2021-07-11 10:38:59 +00:00
|
|
|
#ifdef IS_CUDA
|
2020-01-20 08:20:12 +00:00
|
|
|
#define USE_BITSELECT
|
|
|
|
#define USE_ROTATE
|
|
|
|
#endif
|
|
|
|
|
2021-07-11 10:38:59 +00:00
|
|
|
#ifdef IS_HIP
|
2021-07-15 21:34:27 +00:00
|
|
|
#define USE_BITSELECT
|
|
|
|
#define USE_ROTATE
|
2021-07-11 10:38:59 +00:00
|
|
|
#endif
|
|
|
|
|
2020-01-25 11:09:39 +00:00
|
|
|
#ifdef IS_ROCM
|
|
|
|
#define USE_BITSELECT
|
|
|
|
#define USE_ROTATE
|
|
|
|
#endif
|
|
|
|
|
2020-03-02 15:07:13 +00:00
|
|
|
#ifdef IS_INTEL_SDK
|
2020-03-03 07:52:26 +00:00
|
|
|
#ifdef IS_CPU
|
|
|
|
//#define USE_BITSELECT
|
|
|
|
//#define USE_ROTATE
|
|
|
|
#endif
|
2020-03-02 15:07:13 +00:00
|
|
|
#endif
|
|
|
|
|
2020-01-20 08:20:12 +00:00
|
|
|
#ifdef IS_OPENCL
|
2020-01-21 21:09:56 +00:00
|
|
|
//#define USE_BITSELECT
|
|
|
|
//#define USE_ROTATE
|
|
|
|
//#define USE_SWIZZLE
|
2020-01-12 12:32:02 +00:00
|
|
|
#endif
|
|
|
|
|
2022-02-10 20:53:08 +00:00
|
|
|
#ifdef IS_METAL
|
|
|
|
#define USE_ROTATE
|
|
|
|
|
|
|
|
// Metal support max VECT_SIZE = 4
|
|
|
|
#define s0 x
|
|
|
|
#define s1 y
|
|
|
|
#define s2 z
|
|
|
|
#define s3 w
|
2019-03-23 21:15:38 +00:00
|
|
|
#endif
|
2022-02-10 20:53:08 +00:00
|
|
|
|
|
|
|
#endif // _INC_VENDOR_H
|