From 06926bb7bd8fbabc577581b2cac7dacdd3d44eae Mon Sep 17 00:00:00 2001 From: Greg Olechwierowicz Date: Mon, 14 Oct 2024 07:30:48 -0700 Subject: [PATCH] [XLA:GPU] Remove unused SuggestCombinerThreshold method. PiperOrigin-RevId: 685700157 --- xla/service/gpu/BUILD | 1 - xla/service/gpu/all_gather_combiner.cc | 11 ----------- 2 files changed, 12 deletions(-) diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD index 830f15bd0b9df..5c8a18564688d 100644 --- a/xla/service/gpu/BUILD +++ b/xla/service/gpu/BUILD @@ -3126,7 +3126,6 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/service:all_gather_combiner", - "//xla/service:collective_utils", "//xla/service:hlo_domain_map", "//xla/stream_executor:device_description", "@com_google_absl//absl/container:flat_hash_set", diff --git a/xla/service/gpu/all_gather_combiner.cc b/xla/service/gpu/all_gather_combiner.cc index 956c4a3d48e49..1318369ca6b75 100644 --- a/xla/service/gpu/all_gather_combiner.cc +++ b/xla/service/gpu/all_gather_combiner.cc @@ -30,23 +30,12 @@ limitations under the License. #include "xla/service/gpu/gpu_collective_combiner_utils.h" #include "xla/service/gpu/gpu_hlo_schedule.h" #include "xla/service/hlo_domain_map.h" -#include "xla/stream_executor/device_description.h" #include "tsl/platform/statusor.h" namespace xla::gpu { namespace { -// In the absence of combiner threshold flag we try to determine the appropriate -// value. -int64_t SuggestCombinerThreshold(HloModule* module, - const se::DeviceDescription& device_info, - int64_t pointer_size) { - return ComputeSuggestedCombinerThreshold(*module, device_info, - ScheduleGpuModuleWithMemoryScheduler, - HloOpcode::kAllGather, pointer_size); -} - std::optional PipelinedCombinerKey( const HloInstruction* instruction, const HloDomainMap& domain_map, bool combine_by_dim, bool combine_different_dtypes) {