Skip to content

Commit

Permalink
[XLA:GPU] Remove unused SuggestCombinerThreshold method.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 685700157
  • Loading branch information
golechwierowicz authored and Google-ML-Automation committed Oct 14, 2024
1 parent 7f6fdf1 commit 06926bb
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 12 deletions.
1 change: 0 additions & 1 deletion xla/service/gpu/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
11 changes: 0 additions & 11 deletions xla/service/gpu/all_gather_combiner.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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<AllGatherCombiner::GroupKey> PipelinedCombinerKey(
const HloInstruction* instruction, const HloDomainMap& domain_map,
bool combine_by_dim, bool combine_different_dtypes) {
Expand Down

0 comments on commit 06926bb

Please sign in to comment.