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

Ensure custom GPU kernels are properly registered.

Browse Source 
This change ensures that GOOGLE_CUDA=1 when compiling .cc files with --config=cuda enabled (also includes other important copts like -fno-exceptions.)
Change: 129242223
This commit is contained in:
Eugene Brevdo 2016-08-03 11:06:18 -08:00 committed by TensorFlower Gardener
parent ec73b4e819
commit c5b3ea14c0
4 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/contrib/quantization/kernels/quantized_pooling_ops.cc
View File

@ -29,11 +29,6 @@ limitations under the License.
#include "tensorflow/core/util/padding.h" #include "tensorflow/core/util/padding.h"
#include "tensorflow/core/util/tensor_format.h" #include "tensorflow/core/util/tensor_format.h"
#if GOOGLE_CUDA
#include "tensorflow/core/kernels/maxpooling_op_gpu.h"
#include "tensorflow/core/kernels/pooling_ops_common_gpu.h"
#endif // GOOGLE_CUDA
namespace tensorflow { namespace tensorflow {
typedef Eigen::ThreadPoolDevice CPUDevice; typedef Eigen::ThreadPoolDevice CPUDevice;

1
tensorflow/tensorflow.bzl
View File

@ -634,6 +634,7 @@ def tf_custom_op_library(name, srcs=[], gpu_srcs=[], deps=[]):
srcs=srcs, srcs=srcs,
deps=deps + if_cuda(cuda_deps), deps=deps + if_cuda(cuda_deps),
data=[name + "_check_deps"], data=[name + "_check_deps"],
copts=tf_copts(),
linkshared=1, linkshared=1,
linkopts = select({ linkopts = select({
"//conditions:default": [ "//conditions:default": [

1
tensorflow/tf_exported_symbols.lds
View File

@ -1 +1,2 @@
*tensorflow* *tensorflow*
*perftools*gputools*

1
tensorflow/tf_version_script.lds
View File

@ -1,6 +1,7 @@
tensorflow { tensorflow {
global: global:
*tensorflow*; *tensorflow*;
*perftools*gputools*;
local: local:
*; *;
}; };