Remove some extra cuda header includes.
PiperOrigin-RevId: 225110993
This commit is contained in:
parent
e4e9409b3d
commit
4b974cf1c1
@ -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",
|
||||
|
@ -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"
|
||||
|
@ -21,7 +21,6 @@ limitations under the License.
|
||||
#include <algorithm>
|
||||
|
||||
#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"
|
||||
|
@ -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 {
|
||||
|
||||
|
@ -25,7 +25,6 @@ limitations under the License.
|
||||
#include <complex>
|
||||
|
||||
#include "cuda/include/cuComplex.h"
|
||||
#include "cuda/include/cuda.h"
|
||||
|
||||
namespace stream_executor {
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user