-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCMakeLists.txt
More file actions
65 lines (57 loc) · 1.77 KB
/
CMakeLists.txt
File metadata and controls
65 lines (57 loc) · 1.77 KB
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
cmake_minimum_required(VERSION 3.16)
project(philipchess VERSION 1.0)
set(CMAKE_CXX_STANDARD 20)
set(CMAKE_CXX_STANDARD_REQUIRED True)
set(LIBNVTOOLSEXT "pthread" CACHE FILEPATH "CUDA 13 workaround")
add_library(CUDA::nvToolsExt INTERFACE IMPORTED)
set(TORCH_CUDA_ARCH_LIST "7.5;8.0;8.6;8.9;9.0")
find_package(Torch REQUIRED)
find_package(cppzmq REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
include(FetchContent)
FetchContent_Declare(
chess
GIT_REPOSITORY https://github.com/Disservin/chess-library.git
GIT_TAG master
)
FetchContent_MakeAvailable(chess)
# ------------------------------------------------
# Executable 1: The Self-Play Data Generator
add_executable(philipchess_trainer
src/trainer_main.cpp
src/chess_env.cpp
src/mcts.cpp
src/training.cpp
src/nn.cpp
src/zmq_client.cpp
)
target_include_directories(philipchess_trainer PRIVATE
${CMAKE_CURRENT_SOURCE_DIR}/include
${chess_SOURCE_DIR}/include
)
target_link_libraries(philipchess_trainer ${TORCH_LIBRARIES} cppzmq)
# Executable 2: The GPU Inference Server
add_executable(philipchess_server
src/server_main.cpp
src/chess_env.cpp
src/nn.cpp
src/zmq_server.cpp
)
target_include_directories(philipchess_server PRIVATE
${CMAKE_CURRENT_SOURCE_DIR}/include
${chess_SOURCE_DIR}/include
)
target_link_libraries(philipchess_server ${TORCH_LIBRARIES} cppzmq)
# Executable 3: The Core Search Engine (UCI / Web / Puzzles)
add_executable(philipchess_engine
src/engine_main.cpp
src/chess_env.cpp
src/mcts.cpp
src/nn.cpp
src/zmq_client.cpp
)
target_include_directories(philipchess_engine PRIVATE
${CMAKE_CURRENT_SOURCE_DIR}/include
${chess_SOURCE_DIR}/include
)
target_link_libraries(philipchess_engine ${TORCH_LIBRARIES} cppzmq)