From 4b974cf1c1c072338f9c420b9149840780907443 Mon Sep 17 00:00:00 2001 From: Yifei Feng Date: Tue, 11 Dec 2018 18:59:00 -0800 Subject: [PATCH] Remove some extra cuda header includes. PiperOrigin-RevId: 225110993 --- tensorflow/core/grappler/costs/BUILD | 3 --- tensorflow/core/grappler/costs/utils.cc | 6 ------ tensorflow/core/util/cuda_launch_config.h | 1 - tensorflow/core/util/port.cc | 3 --- tensorflow/stream_executor/cuda/cuda_helpers.h | 1 - 5 files changed, 14 deletions(-) diff --git a/tensorflow/core/grappler/costs/BUILD b/tensorflow/core/grappler/costs/BUILD index 5090e62b2cc..f8af1232f77 100644 --- a/tensorflow/core/grappler/costs/BUILD +++ b/tensorflow/core/grappler/costs/BUILD @@ -132,9 +132,6 @@ tf_cuda_library( name = "utils", srcs = ["utils.cc"], hdrs = ["utils.h"], - cuda_deps = [ - "@local_config_cuda//cuda:cudnn_header", - ], visibility = ["//visibility:public"], deps = [ "//third_party/eigen3", diff --git a/tensorflow/core/grappler/costs/utils.cc b/tensorflow/core/grappler/costs/utils.cc index 7d868a3679e..d45bb14e070 100644 --- a/tensorflow/core/grappler/costs/utils.cc +++ b/tensorflow/core/grappler/costs/utils.cc @@ -20,12 +20,6 @@ limitations under the License. #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_manager.h" #include "tensorflow/core/framework/allocation_description.pb.h" diff --git a/tensorflow/core/util/cuda_launch_config.h b/tensorflow/core/util/cuda_launch_config.h index 080d4067cec..c0ae6349f75 100644 --- a/tensorflow/core/util/cuda_launch_config.h +++ b/tensorflow/core/util/cuda_launch_config.h @@ -21,7 +21,6 @@ limitations under the License. #include #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" -#include "cuda/include/cuda.h" #include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/stream_executor.h" diff --git a/tensorflow/core/util/port.cc b/tensorflow/core/util/port.cc index e01058dff6c..7dc8ddda06a 100644 --- a/tensorflow/core/util/port.cc +++ b/tensorflow/core/util/port.cc @@ -15,9 +15,6 @@ limitations under the License. #include "tensorflow/core/util/port.h" -#if GOOGLE_CUDA -#include "cuda/include/cuda.h" -#endif namespace tensorflow { diff --git a/tensorflow/stream_executor/cuda/cuda_helpers.h b/tensorflow/stream_executor/cuda/cuda_helpers.h index d55706c66a9..dc0dc694cdc 100644 --- a/tensorflow/stream_executor/cuda/cuda_helpers.h +++ b/tensorflow/stream_executor/cuda/cuda_helpers.h @@ -25,7 +25,6 @@ limitations under the License. #include #include "cuda/include/cuComplex.h" -#include "cuda/include/cuda.h" namespace stream_executor {