diff --git a/tensorflow/compiler/xla/service/gpu/BUILD b/tensorflow/compiler/xla/service/gpu/BUILD index 65dd03e6f6a..5ac77e84020 100755 --- a/tensorflow/compiler/xla/service/gpu/BUILD +++ b/tensorflow/compiler/xla/service/gpu/BUILD @@ -236,9 +236,9 @@ cc_library( deps = [ ":backend_configs", ":buffer_allocations", - ":cudnn_conv_runner", ":elemental_ir_emitter", ":gpu_constants", + ":gpu_conv_runner", ":gpu_executable", ":hlo_to_ir_bindings", ":ir_emission_utils", @@ -505,7 +505,7 @@ cc_library( ":backend_configs", ":buffer_allocations", ":cusolver_context", - ":cudnn_conv_runner", + ":gpu_conv_runner", ":gpu_debug_info_manager", ":gpu_types", ":hlo_execution_profiler", @@ -615,7 +615,7 @@ cc_library( deps = [ ":backend_configs", ":buffer_comparator", - ":cudnn_conv_runner", + ":gpu_conv_runner", ":gpu_executable", ":ir_emission_utils", ":stream_executor_util", @@ -637,14 +637,14 @@ cc_library( ) cc_library( - name = "cudnn_conv_algorithm_picker", - srcs = ["cudnn_conv_algorithm_picker.cc"], - hdrs = ["cudnn_conv_algorithm_picker.h"], + name = "gpu_conv_algorithm_picker", + srcs = ["gpu_conv_algorithm_picker.cc"], + hdrs = ["gpu_conv_algorithm_picker.h"], deps = [ ":backend_configs", ":buffer_comparator", - ":cudnn_conv_runner", ":gpu_autotuning_proto", + ":gpu_conv_runner", ":gpu_executable", ":hlo_algorithm_blacklist", ":ir_emission_utils", @@ -672,7 +672,7 @@ cc_library( ) cc_library( - name = "cudnn_conv_runner", + name = "gpu_conv_runner", srcs = ["gpu_conv_runner.cc"], hdrs = ["gpu_conv_runner.h"], deps = [ @@ -1057,6 +1057,7 @@ cc_library( ":cudnn_pad_for_convolutions", ":fusion_merger", ":gpu_constants", + ":gpu_conv_algorithm_picker", ":gpu_conv_padding_legalization", ":gpu_conv_rewriter", ":gpu_copy_insertion", @@ -1156,13 +1157,13 @@ cc_library( ], deps = [ ":cublas_gemm_pad_for_tensor_cores", - ":cudnn_conv_algorithm_picker", ":cudnn_fused_conv_rewriter", ":cudnn_pad_for_convolutions", ":cusolver_rewriter", ":gemm_algorithm_picker", ":gemm_rewriter", ":gpu_compiler", + ":gpu_conv_algorithm_picker", ":gpu_conv_padding_legalization", ":gpu_conv_rewriter", ":gpu_layout_assignment", diff --git a/tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.cc b/tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.cc similarity index 99% rename from tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.cc rename to tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.cc index f68f695b4d7..57a90283d75 100644 --- a/tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.cc +++ b/tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.cc @@ -13,14 +13,14 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ -#include "tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.h" +#include "tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.h" -#include "google/protobuf/any.pb.h" #include "absl/algorithm/container.h" #include "absl/strings/str_cat.h" #include "absl/strings/str_format.h" #include "absl/time/time.h" #include "absl/types/optional.h" +#include "google/protobuf/any.pb.h" #include "tensorflow/compiler/xla/literal_util.h" #include "tensorflow/compiler/xla/service/gpu/backend_configs.pb.h" #include "tensorflow/compiler/xla/service/gpu/buffer_comparator.h" diff --git a/tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.h b/tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.h similarity index 95% rename from tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.h rename to tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.h index 75a1087002b..18d62a0c025 100644 --- a/tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.h +++ b/tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.h @@ -55,8 +55,8 @@ class CudnnConvAlgorithmPicker : public HloModulePass { StatusOr PickBestAlgorithmNoCache( const HloCustomCallInstruction* instr); - se::StreamExecutor* stream_exec_; // never null - se::DeviceMemoryAllocator* allocator_; // may be null + se::StreamExecutor* stream_exec_; // never null + se::DeviceMemoryAllocator* allocator_; // may be null }; } // namespace gpu diff --git a/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc b/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc index 95a61089563..36e34e1ffce 100755 --- a/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc +++ b/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc @@ -22,12 +22,12 @@ limitations under the License. #include "tensorflow/compiler/xla/service/algebraic_simplifier.h" #include "tensorflow/compiler/xla/service/dump.h" #include "tensorflow/compiler/xla/service/gpu/cublas_gemm_pad_for_tensor_cores.h" -#include "tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.h" #include "tensorflow/compiler/xla/service/gpu/cudnn_fused_conv_rewriter.h" #include "tensorflow/compiler/xla/service/gpu/cudnn_pad_for_convolutions.h" #include "tensorflow/compiler/xla/service/gpu/cusolver_rewriter.h" #include "tensorflow/compiler/xla/service/gpu/gemm_algorithm_picker.h" #include "tensorflow/compiler/xla/service/gpu/gemm_rewriter.h" +#include "tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.h" #include "tensorflow/compiler/xla/service/gpu/gpu_conv_padding_legalization.h" #include "tensorflow/compiler/xla/service/gpu/gpu_conv_rewriter.h" #include "tensorflow/compiler/xla/service/gpu/gpu_layout_assignment.h"