aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Gunhan Gulsoy <gunan@google.com>2018-06-27 22:24:29 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-27 22:28:10 -0700
commitea5fa07713655b93733c9aafacc8ddc904484217 (patch)
tree25431c9333aff01a659fc3bf7f3b35303dc3cfda
parent6c92ef52fd15c844462635925c8bb55835964c69 (diff)
Exclude test sources from stream executor builds.
PiperOrigin-RevId: 202423156
-rw-r--r--tensorflow/contrib/cmake/tf_stream_executor.cmake10
1 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/contrib/cmake/tf_stream_executor.cmake b/tensorflow/contrib/cmake/tf_stream_executor.cmake
index 9a37b68119..2f70e59d54 100644
--- a/tensorflow/contrib/cmake/tf_stream_executor.cmake
+++ b/tensorflow/contrib/cmake/tf_stream_executor.cmake
@@ -76,11 +76,11 @@ if (tensorflow_ENABLE_GPU)
list(APPEND tf_stream_executor_srcs ${tf_stream_executor_gpu_srcs})
endif()
-#file(GLOB_RECURSE tf_stream_executor_test_srcs
-# "${tensorflow_source_dir}/tensorflow/stream_executor/*_test.cc"
-# "${tensorflow_source_dir}/tensorflow/stream_executor/*_test.h"
-#)
-#list(REMOVE_ITEM tf_stream_executor_srcs ${tf_stream_executor_test_srcs})
+file(GLOB_RECURSE tf_stream_executor_test_srcs
+ "${tensorflow_source_dir}/tensorflow/stream_executor/*test.cc"
+ "${tensorflow_source_dir}/tensorflow/stream_executor/lib/*test.h"
+)
+list(REMOVE_ITEM tf_stream_executor_srcs ${tf_stream_executor_test_srcs})
if (NOT WIN32)
set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -lgomp")