1 files changed, 10 insertions, 5 deletions
diff --git a/llvm/lib/Target/BPF/BPFMISimplifyPatchable.cpp b/llvm/lib/Target/BPF/BPFMISimplifyPatchable.cpp
index 9c689aed641..d21b977af90 100644
--- a/llvm/lib/Target/BPF/BPFMISimplifyPatchable.cpp
+++ b/llvm/lib/Target/BPF/BPFMISimplifyPatchable.cpp
@@ -124,11 +124,16 @@ bool BPFMISimplifyPatchable::removeLD() {
       if (!IsCandidate)
         continue;
 
-      auto Begin = MRI->use_begin(DstReg), End = MRI->use_end();
-      decltype(End) NextI;
-      for (auto I = Begin; I != End; I = NextI) {
-        NextI = std::next(I);
-        I->setReg(SrcReg);
+      if (MRI->getRegClass(DstReg) == &BPF::GPR32RegClass) {
+        BuildMI(MBB, MI, MI.getDebugLoc(), TII->get(BPF::COPY), DstReg)
+            .addReg(SrcReg, 0, BPF::sub_32);
+      } else {
+        auto Begin = MRI->use_begin(DstReg), End = MRI->use_end();
+        decltype(End) NextI;
+        for (auto I = Begin; I != End; I = NextI) {
+          NextI = std::next(I);
+          I->setReg(SrcReg);
+        }
       }
 
       ToErase = &MI;