1313// limitations under the License.
1414
1515#include < functional>
16- #include < cstdint>
1716#include < random>
1817#include < string>
1918#include < thread>
@@ -136,8 +135,7 @@ TEST(DistanceMatrix, SquaredEuclidean_General) {
136135
137136template <size_t M, size_t N>
138137void TestEuclideanMatrix (void ) {
139- std::mt19937 gen (static_cast <uint32_t >(0x5EED1234u + M * 131u + N * 17u ));
140- constexpr int kFp16MatrixUlpTolerance = 40000 ;
138+ std::mt19937 gen ((std::random_device ())());
141139
142140 const size_t batch_size = M;
143141 const size_t query_size = N;
@@ -176,15 +174,13 @@ void TestEuclideanMatrix(void) {
176174
177175 for (size_t i = 0 ; i < batch_size * query_size; ++i) {
178176 // EXPECT_FLOAT_EQ(result1[i], result2[i]);
179- EXPECT_TRUE (MathHelper::IsAlmostEqual (
180- result1[i], result2[i], kFp16MatrixUlpTolerance ));
177+ EXPECT_TRUE (MathHelper::IsAlmostEqual (result1[i], result2[i], 10000 ));
181178 }
182179}
183180
184181template <size_t M, size_t N>
185182void TestSquaredEuclideanMatrix (void ) {
186- std::mt19937 gen (static_cast <uint32_t >(0x5EED5678u + M * 131u + N * 17u ));
187- constexpr int kFp16MatrixUlpTolerance = 40000 ;
183+ std::mt19937 gen ((std::random_device ())());
188184
189185 const size_t batch_size = M;
190186 const size_t query_size = N;
@@ -223,8 +219,7 @@ void TestSquaredEuclideanMatrix(void) {
223219
224220 for (size_t i = 0 ; i < batch_size * query_size; ++i) {
225221 // EXPECT_FLOAT_EQ(result1[i], result2[i]);
226- EXPECT_TRUE (MathHelper::IsAlmostEqual (
227- result1[i], result2[i], kFp16MatrixUlpTolerance ));
222+ EXPECT_TRUE (MathHelper::IsAlmostEqual (result1[i], result2[i], 10000 ));
228223 }
229224}
230225
@@ -559,7 +554,7 @@ void EuclideanBenchmark(void) {
559554
560555 std::cout << " # (" << IntelIntrinsics () << " ) FP16 " << dimension << " d, "
561556 << batch_size << " * " << query_size << " * " << block_size
562- << ' \n ' ;
557+ << std::endl ;
563558
564559 // 1 Batched Euclidean
565560 elapsed_time.reset ();
@@ -575,7 +570,7 @@ void EuclideanBenchmark(void) {
575570 }
576571 }
577572 std::cout << " * 1 Batched Euclidean (us) \t " << elapsed_time.micro_seconds ()
578- << ' \n ' ;
573+ << std::endl ;
579574
580575 // N Batched Euclidean
581576 elapsed_time.reset ();
@@ -586,7 +581,7 @@ void EuclideanBenchmark(void) {
586581 matrix_batch, &query2[0 ], dimension, results);
587582 }
588583 std::cout << " * N Batched Euclidean (us) \t " << elapsed_time.micro_seconds ()
589- << ' \n ' ;
584+ << std::endl ;
590585
591586 // Unbatched Euclidean
592587 elapsed_time.reset ();
@@ -605,7 +600,7 @@ void EuclideanBenchmark(void) {
605600 }
606601 }
607602 std::cout << " * Unbatched Euclidean (us) \t " << elapsed_time.micro_seconds ()
608- << ' \n ' ;
603+ << std::endl ;
609604}
610605
611606template <size_t M, size_t N, size_t B, size_t D>
@@ -643,7 +638,7 @@ void SquaredEuclideanBenchmark(void) {
643638
644639 std::cout << " # (" << IntelIntrinsics () << " ) FP16 " << dimension << " d, "
645640 << batch_size << " * " << query_size << " * " << block_size
646- << ' \n ' ;
641+ << std::endl ;
647642
648643 // 1 Batched Euclidean
649644 elapsed_time.reset ();
@@ -659,7 +654,7 @@ void SquaredEuclideanBenchmark(void) {
659654 }
660655 }
661656 std::cout << " * 1 Batched SquaredEuclidean (us) \t "
662- << elapsed_time.micro_seconds () << ' \n ' ;
657+ << elapsed_time.micro_seconds () << std::endl ;
663658
664659 // N Batched Euclidean
665660 elapsed_time.reset ();
@@ -670,7 +665,7 @@ void SquaredEuclideanBenchmark(void) {
670665 matrix_batch, &query2[0 ], dimension, results);
671666 }
672667 std::cout << " * N Batched SquaredEuclidean (us) \t "
673- << elapsed_time.micro_seconds () << ' \n ' ;
668+ << elapsed_time.micro_seconds () << std::endl ;
674669
675670 // Unbatched Euclidean
676671 elapsed_time.reset ();
@@ -689,7 +684,7 @@ void SquaredEuclideanBenchmark(void) {
689684 }
690685 }
691686 std::cout << " * Unbatched SquaredEuclidean (us) \t "
692- << elapsed_time.micro_seconds () << ' \n ' ;
687+ << elapsed_time.micro_seconds () << std::endl ;
693688}
694689
695690TEST (DistanceMatrix, DISABLED_Euclidean_Benchmark) {
0 commit comments