[CodeGen] Add TargetInstrInfo ownership

1 files changed, 3 insertions, 3 deletions

diff --git a/backend/include/rcc/CodeGen/MachineFallthrough.h b/backend/include/rcc/CodeGen/MachineFallthrough.h
index 413efe2..5793330 100644
--- a/backend/include/rcc/CodeGen/MachineFallthrough.h
+++ b/backend/include/rcc/CodeGen/MachineFallthrough.h
@@ -7,11 +7,11 @@
 
 namespace rcc {
 
-class TargetBranchInfo;
+class TargetInstrInfo;
 
 class MachineFallthroughPass final : public MachineFunctionPass {
 public:
-  explicit MachineFallthroughPass(const TargetBranchInfo &Info) noexcept : Info(&Info) {}
+  explicit MachineFallthroughPass(const TargetInstrInfo &Info) noexcept : Info(&Info) {}
 
   [[nodiscard]] const MachinePassContract &contract() const noexcept override;
   [[nodiscard]] Expected<MachinePassResult> run(MachineFunction &Function,
@@ -19,7 +19,7 @@ public:
                                                 MachineAnalysisManager &Analyses) override;
 
 private:
-  const TargetBranchInfo *Info;
+  const TargetInstrInfo *Info;
 };
 
 } // namespace rcc