Link externally installed protobuf (v3.10.1) with alpha_zero_example

google-deepmind · Jul 2, 2020 · aba6e59 · aba6e59
1 parent e81a1f1
commit aba6e59
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/open_spiel/examples/CMakeLists.txt b/open_spiel/examples/CMakeLists.txt
@@ -1,7 +1,14 @@
 # alpha_zero is commented out because we don't support depending on the C++
 # TensorFlow library yet. Fixes/contributions welcome.
-# add_executable(alpha_zero_example alpha_zero_example.cc ${OPEN_SPIEL_OBJECTS})
-# add_test(alpha_zero_example_test alpha_zero_example)
+include(FindProtobuf)
+find_package(Protobuf REQUIRED)
+
+find_package(TensorflowCC REQUIRED)
+
+add_executable(alpha_zero_example alpha_zero_example.cc $<TARGET_OBJECTS:alpha_zero> ${OPEN_SPIEL_OBJECTS})
+add_test(alpha_zero_example_test alpha_zero_example)
+
+target_link_libraries(alpha_zero_example ${PROTOBUF_LIBRARIES} TensorflowCC::TensorflowCC)
 
 add_executable(benchmark_game benchmark_game.cc ${OPEN_SPIEL_OBJECTS})
 add_test(benchmark_game_test benchmark_game --game=tic_tac_toe --sims=100 --attempts=2)