diff options
author | 2018-06-27 22:24:29 -0700 | |
---|---|---|
committer | 2018-06-27 22:28:10 -0700 | |
commit | ea5fa07713655b93733c9aafacc8ddc904484217 (patch) | |
tree | 25431c9333aff01a659fc3bf7f3b35303dc3cfda | |
parent | 6c92ef52fd15c844462635925c8bb55835964c69 (diff) |
Exclude test sources from stream executor builds.
PiperOrigin-RevId: 202423156
-rw-r--r-- | tensorflow/contrib/cmake/tf_stream_executor.cmake | 10 |
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") |