From baedcf50239e23dd1b9eb769987f3d58b7e945ba Mon Sep 17 00:00:00 2001 From: Frederik Gossen Date: Wed, 26 Feb 2025 20:04:48 -0800 Subject: [PATCH] [XLA:GPU] Remove unused should_process callback in pipeliner PiperOrigin-RevId: 731553717 --- xla/service/collective_pipeliner.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/xla/service/collective_pipeliner.cc b/xla/service/collective_pipeliner.cc index 735f3de16c103..07325fe77170a 100644 --- a/xla/service/collective_pipeliner.cc +++ b/xla/service/collective_pipeliner.cc @@ -2672,7 +2672,7 @@ absl::Status TransformLoopForwardSink(const WhileLoopAnalysis& loop_analysis, static absl::Status TransformLoopBackward( const WhileLoopAnalysis& loop_analysis, bool insert_non_alias_custom_call, int64_t level_to_operate_on, bool process_different_sized_ops, - HloPredicate should_process, HloPredicate acceptable_formatting, + HloPredicate acceptable_formatting, CollectivePipeliner::HloPostprocessor postprocess_peeled, CollectivePipeliner::HloPostprocessor postprocess_rotated, CollectivePipeliner::HloPostprocessor postprocess_peeled_trailing_op, @@ -3119,8 +3119,8 @@ absl::StatusOr CollectivePipeliner::RunPipeliner( CHECK_EQ(config_.pipelining_direction, PipeliningDirection::kBackward); TF_RETURN_IF_ERROR(TransformLoopBackward( *loop_analysis, !config_.last_run, config_.level_to_operate_on, - config_.process_different_sized_ops, config_.should_process, - config_.acceptable_formatting, config_.postprocess_backward_peeled_op, + config_.process_different_sized_ops, config_.acceptable_formatting, + config_.postprocess_backward_peeled_op, config_.postprocess_backward_rotated_op, config_.postprocess_backward_peeled_trailing_op, next_channel_id, config_.postprocess_pipelined_ops));