Skip to content

Commit 3ca1b08

Browse files
dsharletgxnnpack-bot
authored andcommitted
Relax tolerance for sum squared kernel test
I saw a rare flake of this test whiel running locally PiperOrigin-RevId: 917338626
1 parent 98c8ded commit 3ca1b08

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

ynnpack/kernels/reduce/test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ float Tolerance(ReduceOp op, size_t k, float max_abs_value) {
5757
case ReduceOp::kSum:
5858
return type_info<T>::epsilon() * k * max_abs_value;
5959
case ReduceOp::kSumSquared:
60-
return type_info<T>::epsilon() * k * max_abs_value * max_abs_value;
60+
return type_info<T>::epsilon() * k * max_abs_value * max_abs_value * 2.0f;
6161
case ReduceOp::kMin:
6262
case ReduceOp::kMax:
6363
case ReduceOp::kMinMax:

0 commit comments

Comments
 (0)