thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-24 08:32:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add target specific macros

Browse Source
This commit is contained in:
Dmitrii Kovalkov 2020-03-30 18:36:02 +02:00
parent 04bdffd9d7
commit f6a2209169
2 changed files with 107 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

106
dbms/src/Functions/DynamicTarget/Target.h Normal file
View File

@ -0,0 +1,106 @@
#pragma once
namespace DB::DynamicTarget
{
enum class TargetArch : int {
Scalar,
SSE4,
AVX,
AVX2,
AVX512,
};
#if defined(__GNUC__)
// TODO: There are lots of different AVX512 :(
# define BEGIN_AVX512_SPECIFIC_CODE \
_Pragma("GCC push_options") \
_Pragma("GCC target(\"sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native\")")
# define BEGIN_AVX2_SPECIFIC_CODE \
_Pragma("GCC push_options") \
_Pragma("GCC target(\"sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native\")")
# define BEGIN_AVX_SPECIFIC_CODE \
_Pragma("GCC push_options") \
_Pragma("GCC target(\"sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native\")")
# define BEGIN_SSE4_SPECIFIC_CODE \
_Pragma("GCC push_options") \
_Pragma("GCC target(\"sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,tune=native\")")
# define END_TARGET_SPECIFIC_CODE \
_Pragma("GCC pop_options")
#elif defined(__clang__)
// TODO: There are lots of different AVX512 :(
# define BEGIN_AVX512_SPECIFIC_CODE \
_Pragma("clang attribute push (__attribute__((target(\"sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2\"))))")
# define BEGIN_AVX2_SPECIFIC_CODE \
_Pragma("clang attribute push (__attribute__((target(\"sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2\"))))")
# define BEGIN_AVX_SPECIFIC_CODE \
_Pragma("clang attribute push (__attribute__((target(\"sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx\"))))")
# define BEGIN_SSE4_SPECIFIC_CODE \
_Pragma("clang attribute push (__attribute__((target(\"sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx\"))))")
# define END_TARGET_SPECIFIC_CODE \
_Pragma("clang attribute pop")
#else
# error "Only CLANG and GCC compilers are supported"
#endif
#define DECLARE_DEFAULT_CODE (...) \
namespace TargetSpecific::Default { \
__VA_ARGS__ \
}
#define DECLARE_SSE4_SPECIFIC_CODE (...) \
BEGIN_SSE4_SPECIFIC_CODE \
namespace TargetSpecific::SSE4 { \
__VA_ARGS__ \
} \
END_TARGET_SPECIFIC_CODE
#define DECLARE_AVX_SPECIFIC_CODE (...) \
BEGIN_AVX_SPECIFIC_CODE \
namespace TargetSpecific::AVX { \
__VA_ARGS__ \
} \
END_TARGET_SPECIFIC_CODE
#define DECLARE_AVX2_SPECIFIC_CODE (...) \
BEGIN_AVX2_SPECIFIC_CODE \
namespace TargetSpecific::AVX2 { \
__VA_ARGS__ \
} \
END_TARGET_SPECIFIC_CODE
#define DECLARE_AVX512_SPECIFIC_CODE (...) \
BEGIN_AVX512_SPECIFIC_CODE \
namespace TargetSpecific::AVX512 { \
__VA_ARGS__ \
} \
END_TARGET_SPECIFIC_CODE
#define DYNAMIC_CODE (...) \
DECLARE_DEFAULT_CODE (__VA_ARGS__) \
DECLARE_SSE4_SPECIFIC_CODE (__VA_ARGS__) \
DECLARE_AVX_SPECIFIC_CODE (__VA_ARGS__) \
DECLARE_AVX2_SPECIFIC_CODE (__VA_ARGS__) \
DECLARE_AVX512_SPECIFIC_CODE(__VA_ARGS__)
DECLARE_DEFAULT_CODE(
constexpr auto BuildArch = TargetArch::Scalar;
) // DECLARE_DEFAULT_CODE
DECLARE_SSE4_SPECIFIC_CODE(
constexpr auto BuildArch = TargetArch::SSE4;
) // DECLARE_SSE4_SPECIFIC_CODE
DECLARE_AVX_SPECIFIC_CODE(
constexpr auto BuildArch = TargetArch::AVX;
) // DECLARE_AVX_SPECIFIC_CODE
DECLARE_AVX2_SPECIFIC_CODE(
constexpr auto BuildArch = TargetArch::AVX2;
) // DECLARE_AVX2_SPECIFIC_CODE
DECLARE_AVX512_SPECIFIC_CODE(
constexpr auto BuildArch = TargetArch::AVX512;
) // DECLARE_AVX512_SPECIFIC_CODE
} // namespace DB::DynamicTarget

2
src/Functions/IFunctionImpl.h
View File

@ -5,7 +5,7 @@
/// In order to implement a new function you can choose one of two options:
/// * Implement interface for IFunction (old function interface, which is planned to be removed sometimes)
/// * Implement three interfaces for IExecutableFunctionImpl, IFunctionBaseImpl and IFunctionOverloadResolverImpl
/// Generally saying, IFunction represents a union of tree new interfaces. However, it can't be used for all cases.
/// Generally saying, IFunction represents a union of three new interfaces. However, it can't be used for all cases.
/// Examples:
/// * Function properties may depend on arguments type (e.g. toUInt32(UInt8) is globally monotonic, toUInt32(UInt64) - only on intervals)
/// * In implementation of lambda functions DataTypeFunction needs an functional object with known arguments and return type