diff options
| author |  Gunhan Gulsoy <gunan@google.com> | 2018-06-27 22:24:29 -0700 |
|---|---|---|
| committer | Gunhan Gulsoy <gunan@google.com> | 2018-06-28 21:37:43 -0700 |
| commit | 406a3609c5edf56c73ab4797bfd045cc29faf858 (patch) | |
| tree | 9c4692611540f2aa708807ef08e0affab3693b2f | |
| parent | fbdfbd8f69b67e25e92ed98c6cb0c98fb2794b79 (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") |