experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Renaming cudnn_conv_algorithm_picker to gpu_conv_algorithm_picker

Browse Source
This commit is contained in:
jerryyin 2019-09-09 15:44:12 +00:00
parent 70e85fa79a
commit a3fbb1c352
4 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tensorflow/compiler/xla/service/gpu/BUILD
View File

@ -236,9 +236,9 @@ cc_library(
deps = [ deps = [
":backend_configs", ":backend_configs",
":buffer_allocations", ":buffer_allocations",
":cudnn_conv_runner",
":elemental_ir_emitter", ":elemental_ir_emitter",
":gpu_constants", ":gpu_constants",
":gpu_conv_runner",
":gpu_executable", ":gpu_executable",
":hlo_to_ir_bindings", ":hlo_to_ir_bindings",
":ir_emission_utils", ":ir_emission_utils",
@ -505,7 +505,7 @@ cc_library(
":backend_configs", ":backend_configs",
":buffer_allocations", ":buffer_allocations",
":cusolver_context", ":cusolver_context",
":cudnn_conv_runner", ":gpu_conv_runner",
":gpu_debug_info_manager", ":gpu_debug_info_manager",
":gpu_types", ":gpu_types",
":hlo_execution_profiler", ":hlo_execution_profiler",
@ -615,7 +615,7 @@ cc_library(
deps = [ deps = [
":backend_configs", ":backend_configs",
":buffer_comparator", ":buffer_comparator",
":cudnn_conv_runner", ":gpu_conv_runner",
":gpu_executable", ":gpu_executable",
":ir_emission_utils", ":ir_emission_utils",
":stream_executor_util", ":stream_executor_util",
@ -637,14 +637,14 @@ cc_library(
) )
cc_library( cc_library(
name = "cudnn_conv_algorithm_picker", name = "gpu_conv_algorithm_picker",
srcs = ["cudnn_conv_algorithm_picker.cc"], srcs = ["gpu_conv_algorithm_picker.cc"],
hdrs = ["cudnn_conv_algorithm_picker.h"], hdrs = ["gpu_conv_algorithm_picker.h"],
deps = [ deps = [
":backend_configs", ":backend_configs",
":buffer_comparator", ":buffer_comparator",
":cudnn_conv_runner",
":gpu_autotuning_proto", ":gpu_autotuning_proto",
":gpu_conv_runner",
":gpu_executable", ":gpu_executable",
":hlo_algorithm_blacklist", ":hlo_algorithm_blacklist",
":ir_emission_utils", ":ir_emission_utils",
@ -672,7 +672,7 @@ cc_library(
) )
cc_library( cc_library(
name = "cudnn_conv_runner", name = "gpu_conv_runner",
srcs = ["gpu_conv_runner.cc"], srcs = ["gpu_conv_runner.cc"],
hdrs = ["gpu_conv_runner.h"], hdrs = ["gpu_conv_runner.h"],
deps = [ deps = [
@ -1057,6 +1057,7 @@ cc_library(
":cudnn_pad_for_convolutions", ":cudnn_pad_for_convolutions",
":fusion_merger", ":fusion_merger",
":gpu_constants", ":gpu_constants",
":gpu_conv_algorithm_picker",
":gpu_conv_padding_legalization", ":gpu_conv_padding_legalization",
":gpu_conv_rewriter", ":gpu_conv_rewriter",
":gpu_copy_insertion", ":gpu_copy_insertion",
@ -1156,13 +1157,13 @@ cc_library(
], ],
deps = [ deps = [
":cublas_gemm_pad_for_tensor_cores", ":cublas_gemm_pad_for_tensor_cores",
":cudnn_conv_algorithm_picker",
":cudnn_fused_conv_rewriter", ":cudnn_fused_conv_rewriter",
":cudnn_pad_for_convolutions", ":cudnn_pad_for_convolutions",
":cusolver_rewriter", ":cusolver_rewriter",
":gemm_algorithm_picker", ":gemm_algorithm_picker",
":gemm_rewriter", ":gemm_rewriter",
":gpu_compiler", ":gpu_compiler",
":gpu_conv_algorithm_picker",
":gpu_conv_padding_legalization", ":gpu_conv_padding_legalization",
":gpu_conv_rewriter", ":gpu_conv_rewriter",
":gpu_layout_assignment", ":gpu_layout_assignment",

4
tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.cc → tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.cc
View File

@ -13,14 +13,14 @@ See the License for the specific language governing permissions and
limitations under the License. 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/algorithm/container.h"
#include "absl/strings/str_cat.h" #include "absl/strings/str_cat.h"
#include "absl/strings/str_format.h" #include "absl/strings/str_format.h"
#include "absl/time/time.h" #include "absl/time/time.h"
#include "absl/types/optional.h" #include "absl/types/optional.h"
#include "google/protobuf/any.pb.h"
#include "tensorflow/compiler/xla/literal_util.h" #include "tensorflow/compiler/xla/literal_util.h"
#include "tensorflow/compiler/xla/service/gpu/backend_configs.pb.h" #include "tensorflow/compiler/xla/service/gpu/backend_configs.pb.h"
#include "tensorflow/compiler/xla/service/gpu/buffer_comparator.h" #include "tensorflow/compiler/xla/service/gpu/buffer_comparator.h"

4
tensorflow/compiler/xla/service/gpu/cudnn_conv_algorithm_picker.h → tensorflow/compiler/xla/service/gpu/gpu_conv_algorithm_picker.h
View File

@ -55,8 +55,8 @@ class CudnnConvAlgorithmPicker : public HloModulePass {
StatusOr<tensorflow::AutotuneResult> PickBestAlgorithmNoCache( StatusOr<tensorflow::AutotuneResult> PickBestAlgorithmNoCache(
const HloCustomCallInstruction* instr); const HloCustomCallInstruction* instr);
se::StreamExecutor* stream_exec_; // never null se::StreamExecutor* stream_exec_; // never null
se::DeviceMemoryAllocator* allocator_; // may be null se::DeviceMemoryAllocator* allocator_; // may be null
}; };
} // namespace gpu } // namespace gpu

2
tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc
View File

@ -22,12 +22,12 @@ limitations under the License.
#include "tensorflow/compiler/xla/service/algebraic_simplifier.h" #include "tensorflow/compiler/xla/service/algebraic_simplifier.h"
#include "tensorflow/compiler/xla/service/dump.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/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_fused_conv_rewriter.h"
#include "tensorflow/compiler/xla/service/gpu/cudnn_pad_for_convolutions.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/cusolver_rewriter.h"
#include "tensorflow/compiler/xla/service/gpu/gemm_algorithm_picker.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/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_padding_legalization.h"
#include "tensorflow/compiler/xla/service/gpu/gpu_conv_rewriter.h" #include "tensorflow/compiler/xla/service/gpu/gpu_conv_rewriter.h"
#include "tensorflow/compiler/xla/service/gpu/gpu_layout_assignment.h" #include "tensorflow/compiler/xla/service/gpu/gpu_layout_assignment.h"