[CodeGen] Add target MIR text schema

1 files changed, 6 insertions, 0 deletions

diff --git a/backend/lib/CodeGen/MachineFunctionCloner.cpp b/backend/lib/CodeGen/MachineFunctionCloner.cpp
index b769f7d..7dcf5e2 100644
--- a/backend/lib/CodeGen/MachineFunctionCloner.cpp
+++ b/backend/lib/CodeGen/MachineFunctionCloner.cpp
@@ -214,6 +214,12 @@ cloneMemOperand(const MachineMemOperand &SourceOperand, CloneState &State) {
       if (!SetResult)
         return std::unexpected(SetResult.error());
     }
+    if (SourceInfo.TargetFlags != 0) {
+      auto SetResult = State.Destination.registerInfo().setTargetRegisterFlags(
+          *DestinationReg, SourceInfo.TargetFlags);
+      if (!SetResult)
+        return std::unexpected(SetResult.error());
+    }
     for (const PhysicalRegister Hint : SourceInfo.AllocationHints) {
       auto AddResult = State.Destination.registerInfo().addAllocationHint(*DestinationReg, Hint);
       if (!AddResult)