cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix tests to work with arch haswell/sandybridge.

Browse Source 
PiperOrigin-RevId: 538538356
This commit is contained in:
Ilya Tokar 2023-06-07 11:04:52 -07:00 committed by Copybara-Service
parent 4b0f3cacae
commit 489e927410
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mediapipe/tasks/cc/text/text_embedder/text_embedder_test.cc
View File

@ -45,7 +45,7 @@ constexpr char kUniversalSentenceEncoderModel[] =
// Tolerance for embedding vector coordinate values. // Tolerance for embedding vector coordinate values.
constexpr float kEpsilon = 1e-4; constexpr float kEpsilon = 1e-4;
// Tolerancy for cosine similarity evaluation. // Tolerancy for cosine similarity evaluation.
constexpr double kSimilarityTolerancy = 1e-6; constexpr double kSimilarityTolerancy = 2e-2;
using ::mediapipe::file::JoinPath; using ::mediapipe::file::JoinPath;
using ::testing::HasSubstr; using ::testing::HasSubstr;
@ -79,6 +79,8 @@ TEST_F(EmbedderTest, SucceedsWithMobileBert) {
ASSERT_EQ(result0.embeddings[0].float_embedding.size(), 512); ASSERT_EQ(result0.embeddings[0].float_embedding.size(), 512);
#ifdef _WIN32 #ifdef _WIN32
ASSERT_NEAR(result0.embeddings[0].float_embedding[0], 21.2148f, kEpsilon); ASSERT_NEAR(result0.embeddings[0].float_embedding[0], 21.2148f, kEpsilon);
#elif defined(__FMA__)
ASSERT_NEAR(result0.embeddings[0].float_embedding[0], 21.3605f, kEpsilon);
#else #else
ASSERT_NEAR(result0.embeddings[0].float_embedding[0], 19.9016f, kEpsilon); ASSERT_NEAR(result0.embeddings[0].float_embedding[0], 19.9016f, kEpsilon);
#endif // _WIN32 #endif // _WIN32
@ -87,7 +89,11 @@ TEST_F(EmbedderTest, SucceedsWithMobileBert) {
auto result1, text_embedder->Embed("what a great and fantastic trip")); auto result1, text_embedder->Embed("what a great and fantastic trip"));
ASSERT_EQ(result1.embeddings.size(), 1); ASSERT_EQ(result1.embeddings.size(), 1);
ASSERT_EQ(result1.embeddings[0].float_embedding.size(), 512); ASSERT_EQ(result1.embeddings[0].float_embedding.size(), 512);
#ifdef __FMA__
ASSERT_NEAR(result1.embeddings[0].float_embedding[0], 21.254150f, kEpsilon);
#else
ASSERT_NEAR(result1.embeddings[0].float_embedding[0], 22.626251f, kEpsilon); ASSERT_NEAR(result1.embeddings[0].float_embedding[0], 22.626251f, kEpsilon);
#endif
// Check cosine similarity. // Check cosine similarity.
MP_ASSERT_OK_AND_ASSIGN( MP_ASSERT_OK_AND_ASSIGN(