@@ -137,7 +137,7 @@ TEST(DistanceMatrix, SquaredEuclidean_General) {
137137template <size_t M, size_t N>
138138void TestEuclideanMatrix (void ) {
139139 std::mt19937 gen (static_cast <uint32_t >(0x5EED1234u + M * 131u + N * 17u ));
140- constexpr int kFp16MatrixUlpTolerance = 20000 ;
140+ constexpr int kFp16MatrixUlpTolerance = 40000 ;
141141
142142 const size_t batch_size = M;
143143 const size_t query_size = N;
@@ -184,7 +184,7 @@ void TestEuclideanMatrix(void) {
184184template <size_t M, size_t N>
185185void TestSquaredEuclideanMatrix (void ) {
186186 std::mt19937 gen (static_cast <uint32_t >(0x5EED5678u + M * 131u + N * 17u ));
187- constexpr int kFp16MatrixUlpTolerance = 20000 ;
187+ constexpr int kFp16MatrixUlpTolerance = 40000 ;
188188
189189 const size_t batch_size = M;
190190 const size_t query_size = N;
@@ -559,7 +559,7 @@ void EuclideanBenchmark(void) {
559559
560560 std::cout << " # (" << IntelIntrinsics () << " ) FP16 " << dimension << " d, "
561561 << batch_size << " * " << query_size << " * " << block_size
562- << std::endl ;
562+ << ' \n ' ;
563563
564564 // 1 Batched Euclidean
565565 elapsed_time.reset ();
@@ -575,7 +575,7 @@ void EuclideanBenchmark(void) {
575575 }
576576 }
577577 std::cout << " * 1 Batched Euclidean (us) \t " << elapsed_time.micro_seconds ()
578- << std::endl ;
578+ << ' \n ' ;
579579
580580 // N Batched Euclidean
581581 elapsed_time.reset ();
@@ -586,7 +586,7 @@ void EuclideanBenchmark(void) {
586586 matrix_batch, &query2[0 ], dimension, results);
587587 }
588588 std::cout << " * N Batched Euclidean (us) \t " << elapsed_time.micro_seconds ()
589- << std::endl ;
589+ << ' \n ' ;
590590
591591 // Unbatched Euclidean
592592 elapsed_time.reset ();
@@ -605,7 +605,7 @@ void EuclideanBenchmark(void) {
605605 }
606606 }
607607 std::cout << " * Unbatched Euclidean (us) \t " << elapsed_time.micro_seconds ()
608- << std::endl ;
608+ << ' \n ' ;
609609}
610610
611611template <size_t M, size_t N, size_t B, size_t D>
@@ -643,7 +643,7 @@ void SquaredEuclideanBenchmark(void) {
643643
644644 std::cout << " # (" << IntelIntrinsics () << " ) FP16 " << dimension << " d, "
645645 << batch_size << " * " << query_size << " * " << block_size
646- << std::endl ;
646+ << ' \n ' ;
647647
648648 // 1 Batched Euclidean
649649 elapsed_time.reset ();
@@ -659,7 +659,7 @@ void SquaredEuclideanBenchmark(void) {
659659 }
660660 }
661661 std::cout << " * 1 Batched SquaredEuclidean (us) \t "
662- << elapsed_time.micro_seconds () << std::endl ;
662+ << elapsed_time.micro_seconds () << ' \n ' ;
663663
664664 // N Batched Euclidean
665665 elapsed_time.reset ();
@@ -670,7 +670,7 @@ void SquaredEuclideanBenchmark(void) {
670670 matrix_batch, &query2[0 ], dimension, results);
671671 }
672672 std::cout << " * N Batched SquaredEuclidean (us) \t "
673- << elapsed_time.micro_seconds () << std::endl ;
673+ << elapsed_time.micro_seconds () << ' \n ' ;
674674
675675 // Unbatched Euclidean
676676 elapsed_time.reset ();
@@ -689,7 +689,7 @@ void SquaredEuclideanBenchmark(void) {
689689 }
690690 }
691691 std::cout << " * Unbatched SquaredEuclidean (us) \t "
692- << elapsed_time.micro_seconds () << std::endl ;
692+ << elapsed_time.micro_seconds () << ' \n ' ;
693693}
694694
695695TEST (DistanceMatrix, DISABLED_Euclidean_Benchmark) {
0 commit comments