-
Notifications
You must be signed in to change notification settings - Fork 2
/
CMakeLists.txt
77 lines (55 loc) · 2.1 KB
/
CMakeLists.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
cmake_minimum_required(VERSION 3.14)
project(EMVB)
enable_testing()
if(NOT DEFINED CMAKE_CXX_STANDARD)
set(CMAKE_CXX_STANDARD 20)
endif()
if (NOT CMAKE_BUILD_TYPE)
set(CMAKE_BUILD_TYPE "Release")
endif ()
MESSAGE(STATUS "CMAKE_BUILD_TYPE: " ${CMAKE_BUILD_TYPE})
include_directories(include)
# Set compiler options
if(NOT MSVC)
# set(CMAKE_C_FLAGS "-O3 -ffast-math -march=native -mavx512f -funroll-loops -fopenmp -g ${CMAKE_C_FLAGS}")
set(CMAKE_C_FLAGS "-O3 -ffast-math -march=native -mavx512f -mavx512vl -mavx512bw -funroll-loops -fopenmp -g ${CMAKE_C_FLAGS}")
else()
include("${CMAKE_CURRENT_LIST_DIR}/msvc_simd_isa.cmake")
if(SUPPORT_MSVC_AVX512)
set(MSVC_SIMD_FLAGS "/arch:AVX512")
elseif(SUPPORT_MSVC_AVX2)
set(MSVC_SIMD_FLAGS "/arch:AVX2")
elseif(SUPPORT_MSVC_AVX)
set(MSVC_SIMD_FLAGS "/arch:AVX")
else()
set(MSVC_SIMD_FLAGS "")
endif()
set(CMAKE_C_FLAGS "/O2 /fp:fast ${MSVC_SIMD_FLAGS} ${CMAKE_C_FLAGS}")
endif()
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -ggdb -Wall -Wextra -Wno-missing-braces")
# Set Windows stack size as on Linux: 2MB on 32-bit, 8MB on 64-bit
if (WIN32)
math(EXPR stack_size "${CMAKE_SIZEOF_VOID_P}*${CMAKE_SIZEOF_VOID_P}*128*1024")
if (MSVC)
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} /STACK:${stack_size}")
else()
# compiling with clang + lld
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Xlinker /stack:${stack_size}")
endif()
endif()
set(CMAKE_CXX_FLAGS "${CMAKE_C_FLAGS} ${CMAKE_CXX_FLAGS}")
## Project-wide setup
set(CMAKE_CXX_STANDARD_REQUIRED YES)
set(CMAKE_CXX_EXTENSIONS YES)
## External projects
Include(FetchContent)
find_package(MKL CONFIG REQUIRED)
add_subdirectory(external/faiss)
include_directories(${PROJECT_SOURCE_DIR}/external/faiss)
add_subdirectory(external/cnpy)
include_directories(${PROJECT_SOURCE_DIR}/external/cnpy)
add_subdirectory(external/cmd_line_parser)
include_directories(${PROJECT_SOURCE_DIR}/external/cmd_line_parser)
add_executable(perf_emvb src/perf_emvb.cpp)
target_link_libraries(perf_emvb cnpy faiss $<LINK_ONLY:MKL::MKL>)
target_include_directories(perf_emvb PRIVATE include)