Skip to content

Commit 9e65c6c

Browse files
committed
[NFC] Fix memory leak in IR2Vec tests
After refactoring in ed1d954, VocabStorage are being leaked.
1 parent 679d2b2 commit 9e65c6c

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

llvm/unittests/Analysis/FunctionPropertiesAnalysisTest.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,11 @@ class FunctionPropertiesAnalysisTest : public testing::Test {
5959
ir2vec::ArgWeight = 1.0;
6060
}
6161

62+
~FunctionPropertiesAnalysisTest() override {
63+
delete IR2VecVocab;
64+
IR2VecVocab = nullptr;
65+
}
66+
6267
private:
6368
float OriginalOpcWeight = ir2vec::OpcWeight;
6469
float OriginalTypeWeight = ir2vec::TypeWeight;

llvm/unittests/Analysis/IR2VecTest.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -319,6 +319,10 @@ class IR2VecTestFixture : public ::testing::Test {
319319
AddInst = BinaryOperator::CreateAdd(Arg, Const, "add", BB);
320320
RetInst = ReturnInst::Create(Ctx, AddInst, BB);
321321
}
322+
void TearDown() override {
323+
delete V ;
324+
V = nullptr;
325+
}
322326
};
323327

324328
TEST_F(IR2VecTestFixture, GetInstVecMap_Symbolic) {

0 commit comments

Comments
 (0)