Update tensor reduction test to avoid undefined division of bfloat16 by int.

This commit is contained in:
Rasmus Munk Larsen 2020-07-22 00:35:51 +00:00
parent 0aeaf5f451
commit ac2eca6b11

View File

@ -204,7 +204,7 @@ static void test_simple_reductions() {
++count;
}
}
VERIFY_IS_APPROX(result(i, j), sum / count);
VERIFY_IS_APPROX(result(i, j), sum / Scalar(count));
}
}