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 = [
":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",

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.
==============================================================================*/
#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"

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(
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

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/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"