cppgrad/CMakeLists.txt at dev · edgjj/cppgrad · 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
# Copyright (c) 2023 Yegor Suslin

# This software is released under the MIT License.
# https://opensource.org/licenses/MIT

cmake_minimum_required(VERSION 3.20)

project(cppgrad VERSION 0.0.1 LANGUAGES CXX)

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED)

if (MSVC)
	add_compile_options("$<$<COMPILE_LANGUAGE:CXX>:/MP;/fp:fast;>")
else()
	add_compile_options("$<$<COMPILE_LANGUAGE:CXX>:-ffast-math>")
endif()

string(
    COMPARE EQUAL
    "${CMAKE_SOURCE_DIR}" "${CMAKE_CURRENT_SOURCE_DIR}"
    IS_TOP_PROJECT
)

set(CPPGRAD_CUDA ON CACHE BOOL "Build with CUDA backend enabled")
set(CPPGRAD_MPI OFF CACHE BOOL "Build with distributed MPI compute capatibilities")

# to be used
set(CPPGRAD_INTERNAL_LIBS "")

include (cmake/build_extras.cmake)

file (GLOB
	CPPGRAD_HDR
	include/cppgrad/*.hpp
	include/cppgrad/autograd/*.hpp
	include/cppgrad/device/*.hpp
	include/cppgrad/tensor/*.hpp
	include/cppgrad/tensor/util/*.hpp
	include/cppgrad/itertools/*.hpp
	include/cppgrad/nn/*.hpp
)

list(APPEND CPPGRAD_HDR ${CPPGRAD_DEVICE_INCLUDES})

file (GLOB
	CPPGRAD_SRC
	src/*.cpp
	src/tensor/*.cpp
	src/autograd/*.cpp
	src/nn/*.cpp
)

add_library (cppgrad
	STATIC
	${CPPGRAD_SRC}
	${CPPGRAD_HDR}
)

target_include_directories(cppgrad PUBLIC include/)
target_link_libraries(cppgrad PRIVATE ${CPPGRAD_INTERNAL_LIBS})

set_target_properties(cppgrad
    PROPERTIES LINKER_LANGUAGE CXX
)


if (IS_TOP_PROJECT)
	if (BUILD_TESTS)
		include(CTest)
		enable_testing()
		add_subdirectory(tests)
	endif()

	if (BUILD_EXAMPLES)
		add_subdirectory(examples)
	endif()
endif()