message: [XLA] Use the highest possible precision for large Iota inputs.
diff: @@ -91,7 +91,14 @@ class ForwardPassConvolution_3x3x256_256_OutputZ_Iota : public ConvolutionTest {
  XlaBuilder builder(TestName());
  auto lhs = ConstantR4FromArray4D<T>(&builder, *alhs);
  auto rhs = ConstantR4FromArray4D<T>(&builder, *arhs);
  - Conv(lhs, rhs, {1, 1}, Padding::kValid);
  + PrecisionConfig precision;
  + // The left hand side of the convolution is numbers between 0 and 2304 which
  + // requires at least 11 mantissa bits and the DEFAULT precision config is
  + // allowed to round to bfloat16 which only has 7 mantissa bits.
  + precision.add_operand_precision(PrecisionConfig::HIGHEST);
  + precision.add_operand_precision(PrecisionConfig::DEFAULT);
  + Conv(lhs, rhs, {1, 1}, Padding::kValid, /*feature_group_count=*/1,
  + &precision);
  ComputeAndCompare(&builder, {}, error_spec_);
  }
  