message: Make a method virtual so that it can be overriden.
diff: @@ -76,15 +76,15 @@ class BFloat16Propagation : public HloModulePass {
  // of the producers and users.
  virtual bool ShouldKeepPrecisionUnchanged(const HloInstruction* inst);
  + // Determines whether we should consider changing the precision of the given
  + // instruction in the forward pass.
  + virtual bool InstructionIsCandidateForBF16Output(HloInstruction* hlo);
  +
  private:
  // ***************************
  // Function called and state produced by the forward analysis pass (from
  // parameters to root) that determines the candidate HLOs to use BF16 outputs.
  - // Determines whether we should consider changing the precision of the given
  - // instruction in the forward pass.
  - bool InstructionIsCandidateForBF16Output(HloInstruction* hlo);
  -
  // The set of instructions to consider using bfloat16, computed in the forward
  // pass.
  absl::flat_hash_set<const HloInstruction*> consider_using_bfloat16_;
  