diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/multi_output_fusion.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/multi_output_fusion.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/multi_output_fusion.h b/tensorflow/compiler/xla/service/multi_output_fusion.h index cfdf83cfe8..d9c36fa284 100644 --- a/tensorflow/compiler/xla/service/multi_output_fusion.h +++ b/tensorflow/compiler/xla/service/multi_output_fusion.h @@ -72,8 +72,8 @@ class MultiOutputFusion : public HloPassInterface { // multi-output fusion instruction. virtual int64 GetProfit(HloInstruction* instr1, HloInstruction* instr2) = 0; - // Whether fusing the instruction can reduce cost. - virtual bool IsProfitableOperand(HloInstruction* instr) = 0; + // Whether fusing the instruction can reduce memory reads. + virtual bool IsProfitableOperand(HloInstruction* instr); // Test if it's legal to fuse instr1 and instr2 into one fusion instruction. virtual bool LegalToFuse(HloInstruction* instr1, HloInstruction* instr2); |