Move DefRewriter::setStorageAddress above

Author: meg-gupta

lib/SILOptimizer/Mandatory/AddressLowering.cpp

@@ -2900,6 +2900,14 @@ class DefRewriter : SILInstructionVisitor<DefRewriter> {
     storage.storageAddress = addrMat.materializeAddress(arg);
   }
 
+  void setStorageAddress(SILValue oldValue, SILValue addr) {
+    auto &storage = pass.valueStorageMap.getStorage(oldValue);
+    // getReusedStorageOperand() ensures that oldValue does not already have
+    // separate storage. So there's no need to delete its alloc_stack.
+    assert(!storage.storageAddress || storage.storageAddress == addr);
+    storage.storageAddress = addr;
+  }
+
   void beforeVisit(SILInstruction *inst) {
     LLVM_DEBUG(llvm::dbgs() << "REWRITE DEF "; inst->dump());
     if (storage.storageAddress)
@@ -3003,14 +3011,6 @@ class DefRewriter : SILInstructionVisitor<DefRewriter> {
       addrMat.initializeComposingUse(&operand);
   }
 
-  void setStorageAddress(SILValue oldValue, SILValue addr) {
-    auto &storage = pass.valueStorageMap.getStorage(oldValue);
-    // getReusedStorageOperand() ensures that oldValue does not already have
-    // separate storage. So there's no need to delete its alloc_stack.
-    assert(!storage.storageAddress || storage.storageAddress == addr);
-    storage.storageAddress = addr;
-  }
-
   void visitUnconditionalCheckedCastInst(
       UnconditionalCheckedCastInst *uncondCheckedCast) {
     SILValue srcVal = uncondCheckedCast->getOperand();