mirror of
https://github.com/VectorCamp/vectorscan.git
synced 2025-06-28 16:41:01 +03:00
135 lines
3.9 KiB
CMake
135 lines
3.9 KiB
CMake
option(BUILD_AVX512 "Enabling support for AVX512" OFF)
|
|
option(BUILD_AVX512VBMI "Enabling support for AVX512VBMI" OFF)
|
|
|
|
set(SKYLAKE_FLAG "-march=skylake-avx512")
|
|
set(ICELAKE_FLAG "-march=icelake-server")
|
|
|
|
if (NOT FAT_RUNTIME)
|
|
if (BUILD_AVX512VBMI)
|
|
message (STATUS "AVX512VBMI implies AVX512, enabling BUILD_AVX512")
|
|
set(BUILD_AVX512 ON)
|
|
set(ARCH_C_FLAGS "${ICELAKE_FLAG}")
|
|
set(ARCH_CXX_FLAGS "${ICELAKE_FLAG}")
|
|
endif ()
|
|
if (BUILD_AVX512)
|
|
message (STATUS "AVX512 implies AVX2, enabling BUILD_AVX2")
|
|
set(BUILD_AVX2 ON)
|
|
set(ARCH_C_FLAGS "${SKYLAKE_FLAG}")
|
|
set(ARCH_CXX_FLAGS "${SKYLAKE_FLAG}")
|
|
endif ()
|
|
if (BUILD_AVX2)
|
|
message (STATUS "Enabling BUILD_AVX2")
|
|
set(ARCH_C_FLAGS "-mavx2")
|
|
set(ARCH_CXX_FLAGS "-mavx2")
|
|
else()
|
|
set(ARCH_C_FLAGS "-msse4.2")
|
|
set(ARCH_CXX_FLAGS "-msse4.2")
|
|
endif()
|
|
else()
|
|
set(BUILD_AVX512VBMI ON)
|
|
set(BUILD_AVX512 ON)
|
|
set(BUILD_AVX2 ON)
|
|
set(ARCH_C_FLAGS "-msse4.2")
|
|
set(ARCH_CXX_FLAGS "-msse4.2")
|
|
endif()
|
|
|
|
set(CMAKE_REQUIRED_FLAGS "${ARCH_C_FLAGS}")
|
|
CHECK_INCLUDE_FILES(intrin.h HAVE_C_INTRIN_H)
|
|
CHECK_INCLUDE_FILE_CXX(intrin.h HAVE_CXX_INTRIN_H)
|
|
CHECK_INCLUDE_FILES(x86intrin.h HAVE_C_X86INTRIN_H)
|
|
CHECK_INCLUDE_FILE_CXX(x86intrin.h HAVE_CXX_X86INTRIN_H)
|
|
|
|
if (HAVE_C_X86INTRIN_H)
|
|
set (INTRIN_INC_H "x86intrin.h")
|
|
elseif (HAVE_C_INTRIN_H)
|
|
set (INTRIN_INC_H "intrin.h")
|
|
else()
|
|
message (FATAL_ERROR "No intrinsics header found for SSE/AVX2/AVX512")
|
|
endif ()
|
|
|
|
if (BUILD_AVX512)
|
|
CHECK_C_COMPILER_FLAG(${SKYLAKE_FLAG} HAS_ARCH_SKYLAKE)
|
|
if (NOT HAS_ARCH_SKYLAKE)
|
|
message (FATAL_ERROR "AVX512 not supported by compiler")
|
|
endif ()
|
|
endif ()
|
|
|
|
if (BUILD_AVX512VBMI)
|
|
CHECK_C_COMPILER_FLAG(${ICELAKE_FLAG} HAS_ARCH_ICELAKE)
|
|
if (NOT HAS_ARCH_ICELAKE)
|
|
message (FATAL_ERROR "AVX512VBMI not supported by compiler")
|
|
endif ()
|
|
endif ()
|
|
|
|
# ensure we have the minimum of SSE4.2 - call a SSE4.2 intrinsic
|
|
CHECK_C_SOURCE_COMPILES("#include <${INTRIN_INC_H}>
|
|
int main() {
|
|
__m128i a = _mm_set1_epi8(1);
|
|
(void)_mm_shuffle_epi8(a, a);
|
|
}" HAVE_SSE42)
|
|
|
|
# now look for AVX2
|
|
set(CMAKE_REQUIRED_FLAGS "-mavx2")
|
|
CHECK_C_SOURCE_COMPILES("#include <${INTRIN_INC_H}>
|
|
#if !defined(__AVX2__)
|
|
#error no avx2
|
|
#endif
|
|
|
|
int main(){
|
|
__m256i z = _mm256_setzero_si256();
|
|
(void)_mm256_xor_si256(z, z);
|
|
}" HAVE_AVX2)
|
|
|
|
# and now for AVX512
|
|
set(CMAKE_REQUIRED_FLAGS "${SKYLAKE_FLAG}")
|
|
CHECK_C_SOURCE_COMPILES("#include <${INTRIN_INC_H}>
|
|
#if !defined(__AVX512BW__)
|
|
#error no avx512bw
|
|
#endif
|
|
|
|
int main(){
|
|
__m512i z = _mm512_setzero_si512();
|
|
(void)_mm512_abs_epi8(z);
|
|
}" HAVE_AVX512)
|
|
|
|
# and now for AVX512VBMI
|
|
set(CMAKE_REQUIRED_FLAGS "${ICELAKE_FLAG}")
|
|
CHECK_C_SOURCE_COMPILES("#include <${INTRIN_INC_H}>
|
|
#if !defined(__AVX512VBMI__)
|
|
#error no avx512vbmi
|
|
#endif
|
|
|
|
int main(){
|
|
__m512i a = _mm512_set1_epi8(0xFF);
|
|
__m512i idx = _mm512_set_epi64(3ULL, 2ULL, 1ULL, 0ULL, 7ULL, 6ULL, 5ULL, 4ULL);
|
|
(void)_mm512_permutexvar_epi8(idx, a);
|
|
}" HAVE_AVX512VBMI)
|
|
|
|
if (FAT_RUNTIME)
|
|
if (NOT HAVE_SSE42)
|
|
message(FATAL_ERROR "SSE4.2 support required to build fat runtime")
|
|
endif ()
|
|
if (BUILD_AVX2 AND NOT HAVE_AVX2)
|
|
message(FATAL_ERROR "AVX2 support required to build fat runtime")
|
|
endif ()
|
|
if (BUILD_AVX512 AND NOT HAVE_AVX512)
|
|
message(FATAL_ERROR "AVX512 support requested but not supported")
|
|
endif ()
|
|
if (BUILD_AVX512VBMI AND NOT HAVE_AVX512VBMI)
|
|
message(FATAL_ERROR "AVX512VBMI support requested but not supported")
|
|
endif ()
|
|
else (NOT FAT_RUNTIME)
|
|
if (NOT BUILD_AVX2)
|
|
message(STATUS "Building without AVX2 support")
|
|
endif ()
|
|
if (NOT HAVE_AVX512)
|
|
message(STATUS "Building without AVX512 support")
|
|
endif ()
|
|
if (NOT HAVE_AVX512VBMI)
|
|
message(STATUS "Building without AVX512VBMI support")
|
|
endif ()
|
|
if (NOT HAVE_SSE42)
|
|
message(FATAL_ERROR "A minimum of SSE4.2 compiler support is required")
|
|
endif ()
|
|
endif ()
|