panna/CMakeLists.txt at main · Cecca/panna · GitHub

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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
cmake_minimum_required(VERSION 3.30)
Include(FetchContent)

project(PANNA CXX)

if (NOT CMAKE_BUILD_TYPE AND NOT CMAKE_CONFIGURATION_TYPES)
  set(CMAKE_BUILD_TYPE Release CACHE STRING "Choose the type of build." FORCE)
  set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Debug" "Release" "MinSizeRel" "RelWithDebInfo")
endif()

set(CMAKE_CXX_STANDARD 20)
set(CMAKE_CXX_STANDARD_REQUIRED On)
set(CMAKE_CXX_EXTENSIONS Off)
set(CMAKE_BUILD_PARALLEL_LEVEL 8)

# Add warnings
add_compile_options(-Wall -Wextra -Wno-noexcept-type -Wno-implicit-fallthrough -Wno-unused-function)
# Always optimize as much as we can
# Instead of relying on -march=native (which breaks when building with nix) we
# define the flags we need here
add_compile_options(-O3 -g -mavx2 -mfma)

find_package(OpenMP)
if (OpenMP_FOUND)
    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
endif()

include_directories("include")
include_directories("external")

# FetchContent_Declare(dbg_macro GIT_REPOSITORY https://github.com/sharkdp/dbg-macro FIND_PACKAGE_ARGS)
# FetchContent_MakeAvailable(dbg_macro)

# Get the current git commit hash
if(NOT DEFINED ENV{GIT_COMMIT_HASH})
  execute_process(
    COMMAND git rev-parse --short HEAD
    WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
    OUTPUT_VARIABLE GIT_COMMIT_HASH
    OUTPUT_STRIP_TRAILING_WHITESPACE
  )
else()
  set(GIT_COMMIT_HASH "$ENV{GIT_COMMIT_HASH}")
endif()

# Configure a header file to pass the git hash to the source code
configure_file(
  ${CMAKE_SOURCE_DIR}/include/panna/git_version.hpp.in
  ${CMAKE_BINARY_DIR}/panna/git_version.hpp
  @ONLY
)

# Add the generated header directory to the include paths
include_directories(${CMAKE_BINARY_DIR})

if (NOT SKBUILD)
  # FetchContent_Declare(
  #   Catch2
  #   GIT_REPOSITORY https://github.com/catchorg/Catch2.git
  #   GIT_TAG        v3.8.0 # or a later release
  #   FIND_PACKAGE_ARGS
  # )
  # FetchContent_MakeAvailable(Catch2)
  find_package(Catch2)

  include_directories("test/include")
  add_executable(tests test/tests.cpp)
  target_link_libraries(tests PRIVATE Catch2::Catch2WithMain)

  find_package(HighFive)
  add_executable(emst examples/emst.cpp)
  target_link_libraries(emst PRIVATE HighFive)

  add_executable(emst-test test/emst_test.cpp)
  target_link_libraries(emst-test PRIVATE HighFive)

  add_executable(recall test/recall_test.cpp)
  target_link_libraries(recall PRIVATE  HighFive)

  add_executable(scalability test/scalability_test.cpp)
  target_link_libraries(scalability PRIVATE  HighFive)

  add_executable(mare test/mare_test.cpp)
  target_link_libraries(mare PRIVATE HighFive)

  add_executable(lattice_lsh_collision_probabilities scripts/lattice_lsh_collision_probabilties.cpp)
  add_executable(bench test/bench.cpp)

  # if (HighFive_FOUND)
  #   add_executable(glove examples/glove.cpp)
  #   target_link_libraries(glove HighFive dbg_macro)

  #   add_executable(fashion examples/fashion.cpp)
  #   target_link_libraries(fashion HighFive dbg_macro)
  # endif()
endif()

###############################################################################
# Python Bindings
#
# This cmake code is adapted from
# - https://nanobind.readthedocs.io/en/latest/building.html
# - https://nanobind.readthedocs.io/en/latest/packaging.html

find_package(Python 3.12
  REQUIRED COMPONENTS Interpreter Development.Module
  OPTIONAL_COMPONENTS Development.SABIModule)

# Detect the installed nanobind package and import it into CMake
execute_process(
  COMMAND "${Python_EXECUTABLE}" -m nanobind --cmake_dir
  OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE nanobind_ROOT)
find_package(nanobind CONFIG REQUIRED)

nanobind_add_module(_panna_impl NOMINSIZE NB_STATIC python/wrapper.cpp)
target_link_libraries(_panna_impl PRIVATE)

install(TARGETS _panna_impl LIBRARY DESTINATION panna)