Skip to content

Commit 3f03231

Browse files
authored
[CodeGen] Fix ponential memory leak in CodeGenPassBuilderTest (#77864)
Found by https://lab.llvm.org/buildbot/#/builders/5/builds/40038.
1 parent 6a0c440 commit 3f03231

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

llvm/unittests/CodeGen/CodeGenPassBuilderTest.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ class DummyCodeGenPassBuilder
4646

4747
class CodeGenPassBuilderTest : public testing::Test {
4848
public:
49-
LLVMTargetMachine *TM;
49+
std::unique_ptr<LLVMTargetMachine> TM;
5050

5151
static void SetUpTestCase() {
5252
InitializeAllTargets();
@@ -69,8 +69,8 @@ class CodeGenPassBuilderTest : public testing::Test {
6969
GTEST_SKIP();
7070

7171
TargetOptions Options;
72-
TM = static_cast<LLVMTargetMachine *>(
73-
TheTarget->createTargetMachine("", "", "", Options, std::nullopt));
72+
TM = std::unique_ptr<LLVMTargetMachine>(static_cast<LLVMTargetMachine *>(
73+
TheTarget->createTargetMachine("", "", "", Options, std::nullopt)));
7474
if (!TM)
7575
GTEST_SKIP();
7676
}
@@ -85,7 +85,7 @@ TEST_F(CodeGenPassBuilderTest, basic) {
8585
PassInstrumentationCallbacks PIC;
8686
DummyCodeGenPassBuilder CGPB(*TM, getCGPassBuilderOption(), &PIC);
8787
PipelineTuningOptions PTO;
88-
PassBuilder PB(TM, PTO, std::nullopt, &PIC);
88+
PassBuilder PB(TM.get(), PTO, std::nullopt, &PIC);
8989

9090
PB.registerModuleAnalyses(MAM);
9191
PB.registerCGSCCAnalyses(CGAM);

0 commit comments

Comments
 (0)