[Cleanup] Use HloPredicateIs(Not)Op in all_reduce_blueconnect.cc

PiperOrigin-RevId: 699048665
openxla · Nov 22, 2024 · f708b19 · f708b19
1 parent 603e97f
commit f708b19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xla/service/gpu/transforms/all_reduce_blueconnect.cc b/xla/service/gpu/transforms/all_reduce_blueconnect.cc
@@ -353,7 +353,7 @@ absl::StatusOr<bool> AllReduceBlueConnect::Run(
   for (HloComputation* computation :
        module->MakeNonfusionComputations(execution_threads)) {
     for (HloInstruction* instruction : computation->instructions()) {
-      if (instruction->opcode() == HloOpcode::kAllReduce) {
+      if (HloPredicateIsOp<HloOpcode::kAllReduce>(instruction)) {
         all_reduces.push_back(Cast<HloAllReduceInstruction>(instruction));
       }
     }