diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index a158f686575..ab9d3e11607 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -232,8 +232,8 @@ tf_kernel_library( "concat_lib_gpu_impl.cu.cc", "concat_lib.h", "concat_lib_gpu.h", - "cuda_device_array.h", - "cuda_device_array_gpu.h", + "gpu_device_array.h", + "gpu_device_array_gpu.h", ], deps = [ ":bounds_check", @@ -632,7 +632,7 @@ tf_kernel_library( "split_lib_gpu.h", ], deps = [ - ":cuda_device_array", + ":gpu_device_array", "//tensorflow/core:framework", "//third_party/eigen3", ], @@ -684,10 +684,10 @@ cc_header_only_library( ) cc_library( - name = "cuda_device_array", + name = "gpu_device_array", hdrs = [ - "cuda_device_array.h", - "cuda_device_array_gpu.h", + "gpu_device_array.h", + "gpu_device_array_gpu.h", ], visibility = ["//tensorflow:__subpackages__"], deps = [ @@ -1118,14 +1118,14 @@ tf_kernel_library( tf_kernel_library( name = "split_op", - gpu_srcs = ["cuda_device_array.h"], + gpu_srcs = ["gpu_device_array.h"], prefix = "split_op", deps = ARRAY_DEPS + [":split_lib"], ) tf_kernel_library( name = "split_v_op", - gpu_srcs = ["cuda_device_array.h"], + gpu_srcs = ["gpu_device_array.h"], prefix = "split_v_op", deps = ARRAY_DEPS + [":split_lib"], ) @@ -2160,8 +2160,8 @@ tf_kernel_library( tf_kernel_library( name = "dynamic_stitch_op", gpu_srcs = [ - "cuda_device_array.h", - "cuda_device_array_gpu.h", + "gpu_device_array.h", + "gpu_device_array_gpu.h", ], prefix = "dynamic_stitch_op", deps = DYNAMIC_DEPS, @@ -3309,7 +3309,7 @@ tf_kernel_library( tf_kernel_library( name = "bucketize_op", - gpu_srcs = ["cuda_device_array.h"], + gpu_srcs = ["gpu_device_array.h"], prefix = "bucketize_op", deps = ARRAY_DEPS, ) diff --git a/tensorflow/core/kernels/bucketize_op_gpu.cu.cc b/tensorflow/core/kernels/bucketize_op_gpu.cu.cc index a0f6b755a86..31c73fadf50 100644 --- a/tensorflow/core/kernels/bucketize_op_gpu.cu.cc +++ b/tensorflow/core/kernels/bucketize_op_gpu.cu.cc @@ -24,7 +24,7 @@ limitations under the License. #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/kernels/bucketize_op.h" -#include "tensorflow/core/kernels/cuda_device_array.h" +#include "tensorflow/core/kernels/gpu_device_array.h" #include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/types.h" #include "tensorflow/core/util/cuda_kernel_helper.h" diff --git a/tensorflow/core/kernels/concat_lib_gpu.cc b/tensorflow/core/kernels/concat_lib_gpu.cc index f4f16a6bb17..3108558a220 100644 --- a/tensorflow/core/kernels/concat_lib_gpu.cc +++ b/tensorflow/core/kernels/concat_lib_gpu.cc @@ -27,7 +27,7 @@ limitations under the License. #if GOOGLE_CUDA #include "tensorflow/core/kernels/concat_lib_gpu.h" -#include "tensorflow/core/kernels/cuda_device_array.h" +#include "tensorflow/core/kernels/gpu_device_array.h" namespace tensorflow { namespace { diff --git a/tensorflow/core/kernels/concat_lib_gpu.h b/tensorflow/core/kernels/concat_lib_gpu.h index f8898e6537b..3fcecd754fe 100644 --- a/tensorflow/core/kernels/concat_lib_gpu.h +++ b/tensorflow/core/kernels/concat_lib_gpu.h @@ -25,7 +25,7 @@ limitations under the License. #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "tensorflow/core/framework/register_types.h" #include "tensorflow/core/kernels/concat_lib.h" -#include "tensorflow/core/kernels/cuda_device_array_gpu.h" +#include "tensorflow/core/kernels/gpu_device_array_gpu.h" namespace tensorflow { diff --git a/tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc b/tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc index 8727fb736ab..36f2a8ec3b7 100644 --- a/tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc +++ b/tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc @@ -24,7 +24,7 @@ limitations under the License. #include "tensorflow/core/framework/register_types.h" #include "tensorflow/core/framework/tensor_types.h" #include "tensorflow/core/kernels/concat_lib_gpu.h" -#include "tensorflow/core/kernels/cuda_device_array_gpu.h" +#include "tensorflow/core/kernels/gpu_device_array_gpu.h" #include "tensorflow/core/util/cuda_kernel_helper.h" namespace tensorflow { diff --git a/tensorflow/core/kernels/dynamic_stitch_op.cc b/tensorflow/core/kernels/dynamic_stitch_op.cc index 77a78d9f4fe..8a5f0b570fb 100644 --- a/tensorflow/core/kernels/dynamic_stitch_op.cc +++ b/tensorflow/core/kernels/dynamic_stitch_op.cc @@ -22,7 +22,7 @@ limitations under the License. #include "tensorflow/core/lib/core/threadpool.h" #ifdef GOOGLE_CUDA -#include "tensorflow/core/kernels/cuda_device_array.h" +#include "tensorflow/core/kernels/gpu_device_array.h" #endif // GOOGLE_CUDA namespace tensorflow { diff --git a/tensorflow/core/kernels/dynamic_stitch_op_gpu.cu.cc b/tensorflow/core/kernels/dynamic_stitch_op_gpu.cu.cc index 9ed2c540091..0c60a405a16 100644 --- a/tensorflow/core/kernels/dynamic_stitch_op_gpu.cu.cc +++ b/tensorflow/core/kernels/dynamic_stitch_op_gpu.cu.cc @@ -19,7 +19,7 @@ limitations under the License. #include "tensorflow/core/framework/register_types.h" #include "tensorflow/core/framework/tensor_types.h" -#include "tensorflow/core/kernels/cuda_device_array_gpu.h" +#include "tensorflow/core/kernels/gpu_device_array_gpu.h" #include "tensorflow/core/util/cuda_kernel_helper.h" namespace tensorflow { diff --git a/tensorflow/core/kernels/cuda_device_array.h b/tensorflow/core/kernels/gpu_device_array.h similarity index 98% rename from tensorflow/core/kernels/cuda_device_array.h rename to tensorflow/core/kernels/gpu_device_array.h index 74dc298c7a5..62e39b6e75c 100644 --- a/tensorflow/core/kernels/cuda_device_array.h +++ b/tensorflow/core/kernels/gpu_device_array.h @@ -19,7 +19,7 @@ limitations under the License. #include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h" #include "tensorflow/core/framework/op_kernel.h" -#include "tensorflow/core/kernels/cuda_device_array_gpu.h" +#include "tensorflow/core/kernels/gpu_device_array_gpu.h" namespace tensorflow { diff --git a/tensorflow/core/kernels/cuda_device_array_gpu.h b/tensorflow/core/kernels/gpu_device_array_gpu.h similarity index 100% rename from tensorflow/core/kernels/cuda_device_array_gpu.h rename to tensorflow/core/kernels/gpu_device_array_gpu.h diff --git a/tensorflow/core/kernels/split_lib_gpu.cu.cc b/tensorflow/core/kernels/split_lib_gpu.cu.cc index 441526311b8..3df7306c31b 100644 --- a/tensorflow/core/kernels/split_lib_gpu.cu.cc +++ b/tensorflow/core/kernels/split_lib_gpu.cu.cc @@ -23,7 +23,7 @@ limitations under the License. #include "tensorflow/core/framework/register_types.h" #include "tensorflow/core/framework/tensor_types.h" -#include "tensorflow/core/kernels/cuda_device_array_gpu.h" +#include "tensorflow/core/kernels/gpu_device_array_gpu.h" #include "tensorflow/core/kernels/split_lib_gpu.h" #include "tensorflow/core/util/cuda_kernel_helper.h" diff --git a/tensorflow/core/kernels/split_lib_gpu.h b/tensorflow/core/kernels/split_lib_gpu.h index c667068e34b..ff76d072319 100644 --- a/tensorflow/core/kernels/split_lib_gpu.h +++ b/tensorflow/core/kernels/split_lib_gpu.h @@ -24,7 +24,7 @@ limitations under the License. #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "tensorflow/core/framework/register_types.h" -#include "tensorflow/core/kernels/cuda_device_array_gpu.h" +#include "tensorflow/core/kernels/gpu_device_array_gpu.h" #include "tensorflow/core/kernels/split_lib.h" namespace tensorflow { diff --git a/tensorflow/core/kernels/split_op.cc b/tensorflow/core/kernels/split_op.cc index d69ce3f5853..522419c294c 100644 --- a/tensorflow/core/kernels/split_op.cc +++ b/tensorflow/core/kernels/split_op.cc @@ -29,7 +29,7 @@ limitations under the License. #include "tensorflow/core/util/work_sharder.h" #if GOOGLE_CUDA #include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h" -#include "tensorflow/core/kernels/cuda_device_array.h" +#include "tensorflow/core/kernels/gpu_device_array.h" #include "tensorflow/core/kernels/split_lib_gpu.h" #include "tensorflow/core/platform/stream_executor.h" #endif // GOOGLE_CUDA diff --git a/tensorflow/core/kernels/split_v_op.cc b/tensorflow/core/kernels/split_v_op.cc index 96b240b1ac7..01cd8d81e9b 100644 --- a/tensorflow/core/kernels/split_v_op.cc +++ b/tensorflow/core/kernels/split_v_op.cc @@ -35,7 +35,7 @@ limitations under the License. #include "tensorflow/core/util/work_sharder.h" #if GOOGLE_CUDA #include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h" -#include "tensorflow/core/kernels/cuda_device_array.h" +#include "tensorflow/core/kernels/gpu_device_array.h" #include "tensorflow/core/kernels/split_lib_gpu.h" #include "tensorflow/core/platform/stream_executor.h" #endif // GOOGLE_CUDA