7
7
// ===----------------------------------------------------------------------===//
8
8
9
9
#include " AMDGPUHSAMetadataStreamer.h"
10
- #include " AMDGPUTargetMachine.h"
11
- #include " GCNSubtarget.h"
12
10
#include " SIProgramInfo.h"
11
+ #include " llvm/CodeGen/MachineFunction.h"
13
12
#include " llvm/CodeGen/MachineModuleInfo.h"
14
13
#include " llvm/MC/MCContext.h"
15
14
#include " llvm/MC/MCExpr.h"
@@ -25,9 +24,8 @@ using namespace llvm;
25
24
26
25
class SIProgramInfoMCExprsTest : public testing ::Test {
27
26
protected:
28
- std::unique_ptr<GCNTargetMachine > TM;
27
+ std::unique_ptr<LLVMTargetMachine > TM;
29
28
std::unique_ptr<LLVMContext> Ctx;
30
- std::unique_ptr<GCNSubtarget> ST;
31
29
std::unique_ptr<MachineModuleInfo> MMI;
32
30
std::unique_ptr<MachineFunction> MF;
33
31
std::unique_ptr<Module> M;
@@ -49,7 +47,7 @@ class SIProgramInfoMCExprsTest : public testing::Test {
49
47
const Target *TheTarget = TargetRegistry::lookupTarget (Triple, Error);
50
48
TargetOptions Options;
51
49
52
- TM.reset (static_cast <GCNTargetMachine *>(TheTarget->createTargetMachine (
50
+ TM.reset (static_cast <LLVMTargetMachine *>(TheTarget->createTargetMachine (
53
51
Triple, CPU, FS, Options, std::nullopt, std::nullopt)));
54
52
55
53
Ctx = std::make_unique<LLVMContext>();
@@ -59,9 +57,7 @@ class SIProgramInfoMCExprsTest : public testing::Test {
59
57
auto *F = Function::Create (FType, GlobalValue::ExternalLinkage, " Test" , *M);
60
58
MMI = std::make_unique<MachineModuleInfo>(TM.get ());
61
59
62
- ST = std::make_unique<GCNSubtarget>(TM->getTargetTriple (),
63
- TM->getTargetCPU (),
64
- TM->getTargetFeatureString (), *TM);
60
+ auto *ST = TM->getSubtargetImpl (*F);
65
61
66
62
MF = std::make_unique<MachineFunction>(*F, *TM, *ST, 1 , *MMI);
67
63
MF->initTargetMachineFunctionInfo (*ST);
0 commit comments