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

Remove some extra cuda header includes.

Browse Source 
PiperOrigin-RevId: 225110993
This commit is contained in:
Yifei Feng 2018-12-11 18:59:00 -08:00 committed by TensorFlower Gardener
parent e4e9409b3d
commit 4b974cf1c1
5 changed files with 0 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/core/grappler/costs/BUILD
View File

@ -132,9 +132,6 @@ tf_cuda_library(
name = "utils", name = "utils",
srcs = ["utils.cc"], srcs = ["utils.cc"],
hdrs = ["utils.h"], hdrs = ["utils.h"],
cuda_deps = [
"@local_config_cuda//cuda:cudnn_header",
],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [
"//third_party/eigen3", "//third_party/eigen3",

6
tensorflow/core/grappler/costs/utils.cc
View File

@ -20,12 +20,6 @@ limitations under the License.
#include "third_party/eigen3/Eigen/Core" #include "third_party/eigen3/Eigen/Core"
#if GOOGLE_CUDA
#include "cuda/include/cuda.h"
#include "cuda/include/cuda_runtime_api.h"
#include "cuda/include/cudnn.h"
#endif
#include "tensorflow/core/common_runtime/gpu/gpu_id.h" #include "tensorflow/core/common_runtime/gpu/gpu_id.h"
#include "tensorflow/core/common_runtime/gpu/gpu_id_manager.h" #include "tensorflow/core/common_runtime/gpu/gpu_id_manager.h"
#include "tensorflow/core/framework/allocation_description.pb.h" #include "tensorflow/core/framework/allocation_description.pb.h"

1
tensorflow/core/util/cuda_launch_config.h
View File

@ -21,7 +21,6 @@ limitations under the License.
#include <algorithm> #include <algorithm>
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "cuda/include/cuda.h"
#include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/op_kernel.h"
#include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/platform/stream_executor.h" #include "tensorflow/core/platform/stream_executor.h"

3
tensorflow/core/util/port.cc
View File

@ -15,9 +15,6 @@ limitations under the License.
#include "tensorflow/core/util/port.h" #include "tensorflow/core/util/port.h"
#if GOOGLE_CUDA
#include "cuda/include/cuda.h"
#endif
namespace tensorflow { namespace tensorflow {

1
tensorflow/stream_executor/cuda/cuda_helpers.h
View File

@ -25,7 +25,6 @@ limitations under the License.
#include <complex> #include <complex>
#include "cuda/include/cuComplex.h" #include "cuda/include/cuComplex.h"
#include "cuda/include/cuda.h"
namespace stream_executor { namespace stream_executor {