[nnkit/moco/tf] Build moco_test_tf only when TensorFlow is found (#3465)

Build moco_test_tf only when TensorFlow is found

Signed-off-by: Ivan Vagin <ivan.vagin@samsung.com>

diff --git a/contrib/moco/test/tf/CMakeLists.txt b/contrib/moco/test/tf/CMakeLists.txt
index 1dcc5c6..c0d72f4 100644 (file)
--- a/contrib/moco/test/tf/CMakeLists.txt
+++ b/contrib/moco/test/tf/CMakeLists.txt
@@ -57,6 +57,12 @@ foreach(PBTXTFILE IN ITEMS ${PBTXTFILES})
 
 endforeach(PBTXTFILE)
 
+nncc_find_package(TensorFlow QUIET)
+if(NOT TensorFlow_FOUND)
+  message(STATUS "moco: Skip adding test as TensorFlow is not found")
+  return()
+endif(NOT TensorFlow_FOUND)
+
 # Run tests
 add_test(NAME moco_test_tf
          COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/runall"