[XLA:GPU] Stop creating nested fusion nodes in multi-output fusion.

PiperOrigin-RevId: 202624150

1 files changed, 4 insertions, 4 deletions

diff --git a/tensorflow/compiler/xla/service/multi_output_fusion.h b/tensorflow/compiler/xla/service/multi_output_fusion.h
index d23822e33e..0019cd7254 100644
--- a/tensorflow/compiler/xla/service/multi_output_fusion.h
+++ b/tensorflow/compiler/xla/service/multi_output_fusion.h
@@ -78,6 +78,10 @@ class MultiOutputFusion : public HloPassInterface {
   // Test if it's legal to fuse instr1 and instr2 into one fusion instruction.
   virtual bool LegalToFuse(HloInstruction* instr1, HloInstruction* instr2);
 
+  // Fuse HloInstrctuion instr1 and instr2 and return the fused instruction.
+  // The other instruction is removed from its parent computation.
+  virtual HloInstruction* Fuse(HloInstruction* instr1, HloInstruction* instr2);
+
   // Recompute reachability for the current computation.
   void RecomputeReachability();
 
@@ -101,10 +105,6 @@ class MultiOutputFusion : public HloPassInterface {
   virtual bool DoProducerConsumerMultiOutputFusion();
 
  private:
-  // Fuse HloInstrctuion instr1 and instr2 and return the fused instruction.
-  // The other instruction is removed from its parent computation.
-  HloInstruction* Fuse(HloInstruction* instr1, HloInstruction* instr2);
-
   // Update the internal data structures after instr1 and instr2 are fused into
   // one fusion instruction.
   void Update(HloInstruction* instr1, HloInstruction* instr2);