experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix windows gpu build by disabling nccl which currently will not ()

Browse Source 
compile/work on windows
This commit is contained in:
Guenther Schmuelling 2017-04-06 15:31:08 -07:00 committed by Rohan Jain
parent fa21d4d83d
commit f8dce81aea
1 changed files with 4 additions and 0 deletions

4
tensorflow/contrib/cmake/tf_core_kernels.cmake
View File

@ -102,6 +102,10 @@ if(WIN32)
"${tensorflow_source_dir}/tensorflow/contrib/rnn/kernels/lstm_ops.cc"
"${tensorflow_source_dir}/tensorflow/contrib/rnn/ops/gru_ops.cc"
"${tensorflow_source_dir}/tensorflow/contrib/rnn/ops/lstm_ops.cc"
# temporarily disable nccl (nccl itself needs to be ported to windows first)
"${tensorflow_source_dir}/tensorflow/contrib/nccl/kernels/nccl_manager.cc"
"${tensorflow_source_dir}/tensorflow/contrib/nccl/kernels/nccl_ops.cc"
"${tensorflow_source_dir}/tensorflow/contrib/nccl/ops/nccl_ops.cc"
)
list(REMOVE_ITEM tf_core_kernels_srcs ${tf_core_kernels_windows_exclude_srcs})
endif(WIN32)