diff options
Diffstat (limited to 'tensorflow/contrib/cmake/tf_python.cmake')
-rw-r--r-- | tensorflow/contrib/cmake/tf_python.cmake | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/cmake/tf_python.cmake b/tensorflow/contrib/cmake/tf_python.cmake index d7d3d54003..6fe83bf83e 100644 --- a/tensorflow/contrib/cmake/tf_python.cmake +++ b/tensorflow/contrib/cmake/tf_python.cmake @@ -584,6 +584,7 @@ add_library(pywrap_tensorflow SHARED $<TARGET_OBJECTS:tf_core_direct_session> $<$<BOOL:${tensorflow_ENABLE_GRPC_SUPPORT}>:$<TARGET_OBJECTS:tf_core_distributed_runtime>> $<TARGET_OBJECTS:tf_core_kernels> + $<$<BOOL:${tensorflow_ENABLE_GPU}>:$<TARGET_OBJECTS:tf_core_kernels_cpu_only>> $<$<BOOL:${tensorflow_ENABLE_GPU}>:$<TARGET_OBJECTS:tf_stream_executor>> ) target_include_directories(pywrap_tensorflow PUBLIC |