diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 6c05a39242c..8e26977b765 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 = [ @@ -1110,14 +1110,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"], ) @@ -2152,8 +2152,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, @@ -3301,7 +3301,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_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 0bcd9607f68..2ed30fc33a5 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_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