@@ -323,72 +323,6 @@ class IR2VecTestFixture : public ::testing::Test {
323323 }
324324};
325325
326- // TEST_F(IR2VecTestFixture, GetInstVecMap_Symbolic) {
327- // auto Emb = Embedder::create(IR2VecKind::Symbolic, *F, *V);
328- // ASSERT_TRUE(static_cast<bool>(Emb));
329-
330- // const auto &InstMap = Emb->getInstVecMap();
331-
332- // EXPECT_EQ(InstMap.size(), 2u);
333- // EXPECT_TRUE(InstMap.count(AddInst));
334- // EXPECT_TRUE(InstMap.count(RetInst));
335-
336- // const auto &AddEmb = InstMap.at(AddInst);
337- // const auto &RetEmb = InstMap.at(RetInst);
338- // EXPECT_EQ(AddEmb.size(), 2u);
339- // EXPECT_EQ(RetEmb.size(), 2u);
340-
341- // EXPECT_TRUE(AddEmb.approximatelyEquals(Embedding(2, 25.5)));
342- // EXPECT_TRUE(RetEmb.approximatelyEquals(Embedding(2, 15.5)));
343- // }
344-
345- // TEST_F(IR2VecTestFixture, GetInstVecMap_FlowAware) {
346- // auto Emb = Embedder::create(IR2VecKind::FlowAware, *F, *V);
347- // ASSERT_TRUE(static_cast<bool>(Emb));
348-
349- // const auto &InstMap = Emb->getInstVecMap();
350-
351- // EXPECT_EQ(InstMap.size(), 2u);
352- // EXPECT_TRUE(InstMap.count(AddInst));
353- // EXPECT_TRUE(InstMap.count(RetInst));
354-
355- // EXPECT_EQ(InstMap.at(AddInst).size(), 2u);
356- // EXPECT_EQ(InstMap.at(RetInst).size(), 2u);
357-
358- // EXPECT_TRUE(InstMap.at(AddInst).approximatelyEquals(Embedding(2, 25.5)));
359- // EXPECT_TRUE(InstMap.at(RetInst).approximatelyEquals(Embedding(2, 32.6)));
360- // }
361-
362- // TEST_F(IR2VecTestFixture, GetBBVecMap_Symbolic) {
363- // auto Emb = Embedder::create(IR2VecKind::Symbolic, *F, *V);
364- // ASSERT_TRUE(static_cast<bool>(Emb));
365-
366- // const auto &BBMap = Emb->getBBVecMap();
367-
368- // EXPECT_EQ(BBMap.size(), 1u);
369- // EXPECT_TRUE(BBMap.count(BB));
370- // EXPECT_EQ(BBMap.at(BB).size(), 2u);
371-
372- // // BB vector should be sum of add and ret: {25.5, 25.5} + {15.5, 15.5} =
373- // // {41.0, 41.0}
374- // EXPECT_TRUE(BBMap.at(BB).approximatelyEquals(Embedding(2, 41.0)));
375- // }
376-
377- // TEST_F(IR2VecTestFixture, GetBBVecMap_FlowAware) {
378- // auto Emb = Embedder::create(IR2VecKind::FlowAware, *F, *V);
379- // ASSERT_TRUE(static_cast<bool>(Emb));
380-
381- // const auto &BBMap = Emb->getBBVecMap();
382-
383- // EXPECT_EQ(BBMap.size(), 1u);
384- // EXPECT_TRUE(BBMap.count(BB));
385- // EXPECT_EQ(BBMap.at(BB).size(), 2u);
386-
387- // // BB vector should be sum of add and ret: {25.5, 25.5} + {32.6, 32.6} =
388- // // {58.1, 58.1}
389- // EXPECT_TRUE(BBMap.at(BB).approximatelyEquals(Embedding(2, 58.1)));
390- // }
391-
392326TEST_F (IR2VecTestFixture, GetBBVector_Symbolic) {
393327 auto Emb = Embedder::create (IR2VecKind::Symbolic, *F, *V);
394328 ASSERT_TRUE (static_cast <bool >(Emb));
@@ -474,16 +408,6 @@ TEST_F(IR2VecTestFixture, MultipleComputeEmbeddingsConsistency_FlowAware) {
474408 EXPECT_TRUE (FuncVec1.approximatelyEquals (FuncVec2));
475409 EXPECT_TRUE (FuncVec1.approximatelyEquals (FuncVec3));
476410 EXPECT_TRUE (FuncVec2.approximatelyEquals (FuncVec3));
477-
478- // Also check that instruction vectors remain consistent
479- // const auto &InstMap1 = Emb->getInstVecMap();
480- // const auto &InstMap2 = Emb->getInstVecMap();
481-
482- // EXPECT_EQ(InstMap1.size(), InstMap2.size());
483- // for (const auto &[Inst, Vec1] : InstMap1) {
484- // ASSERT_TRUE(InstMap2.count(Inst));
485- // EXPECT_TRUE(Vec1.approximatelyEquals(InstMap2.at(Inst)));
486- // }
487411}
488412
489413static constexpr unsigned MaxOpcodes = Vocabulary::MaxOpcodes;
0 commit comments