Move the max RP calculation to end of stage

Change-Id: If9378e0c3682ef62768756bea8a0315fdd203594

Author: jrbyrnes

llvm/lib/Target/AMDGPU/GCNSchedStrategy.cpp

@@ -1130,6 +1130,15 @@ bool PreRARematStage::initGCNSchedStage() {
 }
 
 void GCNSchedStage::finalizeGCNSchedStage() {
+  unsigned MaxArchVGPR = 0;
+  for (auto P : DAG.Pressure) {
+    if (P.getArchVGPRNum() > MaxArchVGPR)
+      MaxArchVGPR = P.getArchVGPRNum();
+  }
+
+  MF.getInfo<SIMachineFunctionInfo>()->setMaxArchVGPRPressure(MaxArchVGPR);
+
+
   DAG.finishBlock();
   LLVM_DEBUG(dbgs() << "Ending scheduling stage: " << StageID << "\n");
 }
@@ -1263,14 +1272,6 @@ void GCNSchedStage::finalizeGCNRegion() {
   // reason that the original schedule is better.
   checkScheduling();
 
-  unsigned MaxArchVGPR = 0;
-  for (auto P : DAG.Pressure) {
-    if (P.getArchVGPRNum() > MaxArchVGPR)
-      MaxArchVGPR = P.getArchVGPRNum();
-  }
-
-  MF.getInfo<SIMachineFunctionInfo>()->setMaxArchVGPRPressure(MaxArchVGPR);
-
   if (DAG.RegionsWithIGLPInstrs[RegionIdx] &&
       StageID != GCNSchedStageID::UnclusteredHighRPReschedule)
     SavedMutations.swap(DAG.Mutations);