Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 1 | // $Id$ |
| 2 | //*************************************************************************** |
| 3 | // File: |
| 4 | // PhyRegAlloc.cpp |
| 5 | // |
| 6 | // Purpose: |
| 7 | // Register allocation for LLVM. |
| 8 | // |
| 9 | // History: |
| 10 | // 9/10/01 - Ruchira Sasanka - created. |
| 11 | //**************************************************************************/ |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 12 | |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 13 | #include "llvm/CodeGen/PhyRegAlloc.h" |
| 14 | #include "llvm/CodeGen/MachineInstr.h" |
| 15 | #include "llvm/Target/TargetMachine.h" |
| 16 | #include "llvm/Target/MachineFrameInfo.h" |
| 17 | |
| 18 | |
| 19 | // ***TODO: There are several places we add instructions. Validate the order |
| 20 | // of adding these instructions. |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 21 | |
| 22 | |
| 23 | |
Chris Lattner | 045e7c8 | 2001-09-19 16:26:23 +0000 | [diff] [blame] | 24 | cl::Enum<RegAllocDebugLevel_t> DEBUG_RA("dregalloc", cl::NoFlags, |
| 25 | "enable register allocation debugging information", |
| 26 | clEnumValN(RA_DEBUG_None , "n", "disable debug output"), |
| 27 | clEnumValN(RA_DEBUG_Normal , "y", "enable debug output"), |
| 28 | clEnumValN(RA_DEBUG_Verbose, "v", "enable extra debug output"), 0); |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 29 | |
| 30 | |
| 31 | //---------------------------------------------------------------------------- |
| 32 | // Constructor: Init local composite objects and create register classes. |
| 33 | //---------------------------------------------------------------------------- |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 34 | PhyRegAlloc::PhyRegAlloc(Method *M, |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 35 | const TargetMachine& tm, |
| 36 | MethodLiveVarInfo *const Lvi) |
| 37 | : RegClassList(), |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 38 | TM(tm), |
| 39 | Meth(M), |
| 40 | mcInfo(MachineCodeForMethod::get(M)), |
| 41 | LVI(Lvi), LRI(M, tm, RegClassList), |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 42 | MRI( tm.getRegInfo() ), |
| 43 | NumOfRegClasses(MRI.getNumOfRegClasses()), |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 44 | AddedInstrMap() |
| 45 | /*, PhiInstList()*/ |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 46 | { |
| 47 | // **TODO: use an actual reserved color list |
| 48 | ReservedColorListType *RCL = new ReservedColorListType(); |
| 49 | |
| 50 | // create each RegisterClass and put in RegClassList |
| 51 | for( unsigned int rc=0; rc < NumOfRegClasses; rc++) |
| 52 | RegClassList.push_back( new RegClass(M, MRI.getMachineRegClass(rc), RCL) ); |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 53 | } |
| 54 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 55 | //---------------------------------------------------------------------------- |
| 56 | // This method initally creates interference graphs (one in each reg class) |
| 57 | // and IGNodeList (one in each IG). The actual nodes will be pushed later. |
| 58 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 59 | |
| 60 | void PhyRegAlloc::createIGNodeListsAndIGs() |
| 61 | { |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 62 | if(DEBUG_RA ) cout << "Creating LR lists ..." << endl; |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 63 | |
| 64 | // hash map iterator |
| 65 | LiveRangeMapType::const_iterator HMI = (LRI.getLiveRangeMap())->begin(); |
| 66 | |
| 67 | // hash map end |
| 68 | LiveRangeMapType::const_iterator HMIEnd = (LRI.getLiveRangeMap())->end(); |
| 69 | |
| 70 | for( ; HMI != HMIEnd ; ++HMI ) { |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 71 | |
| 72 | if( (*HMI).first ) { |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 73 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 74 | LiveRange *L = (*HMI).second; // get the LiveRange |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 75 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 76 | if( !L) { |
| 77 | if( DEBUG_RA) { |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 78 | cout << "\n*?!?Warning: Null liver range found for: "; |
| 79 | printValue( (*HMI).first) ; cout << endl; |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 80 | } |
| 81 | continue; |
| 82 | } |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 83 | // if the Value * is not null, and LR |
| 84 | // is not yet written to the IGNodeList |
| 85 | if( !(L->getUserIGNode()) ) { |
| 86 | |
| 87 | RegClass *const RC = // RegClass of first value in the LR |
| 88 | //RegClassList [MRI.getRegClassIDOfValue(*(L->begin()))]; |
| 89 | RegClassList[ L->getRegClass()->getID() ]; |
| 90 | |
| 91 | RC-> addLRToIG( L ); // add this LR to an IG |
| 92 | } |
| 93 | } |
| 94 | } |
| 95 | |
| 96 | // init RegClassList |
| 97 | for( unsigned int rc=0; rc < NumOfRegClasses ; rc++) |
| 98 | RegClassList[ rc ]->createInterferenceGraph(); |
| 99 | |
| 100 | if( DEBUG_RA) |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 101 | cout << "LRLists Created!" << endl; |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 102 | } |
| 103 | |
| 104 | |
| 105 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 106 | //---------------------------------------------------------------------------- |
| 107 | // This method will add all interferences at for a given instruction. |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 108 | // Interence occurs only if the LR of Def (Inst or Arg) is of the same reg |
| 109 | // class as that of live var. The live var passed to this function is the |
| 110 | // LVset AFTER the instruction |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 111 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 112 | |
| 113 | void PhyRegAlloc::addInterference(const Value *const Def, |
| 114 | const LiveVarSet *const LVSet, |
| 115 | const bool isCallInst) { |
| 116 | |
| 117 | LiveVarSet::const_iterator LIt = LVSet->begin(); |
| 118 | |
| 119 | // get the live range of instruction |
| 120 | const LiveRange *const LROfDef = LRI.getLiveRangeForValue( Def ); |
| 121 | |
| 122 | IGNode *const IGNodeOfDef = LROfDef->getUserIGNode(); |
| 123 | assert( IGNodeOfDef ); |
| 124 | |
| 125 | RegClass *const RCOfDef = LROfDef->getRegClass(); |
| 126 | |
| 127 | // for each live var in live variable set |
| 128 | for( ; LIt != LVSet->end(); ++LIt) { |
| 129 | |
| 130 | if( DEBUG_RA > 1) { |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 131 | cout << "< Def="; printValue(Def); |
| 132 | cout << ", Lvar="; printValue( *LIt); cout << "> "; |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 133 | } |
| 134 | |
| 135 | // get the live range corresponding to live var |
| 136 | LiveRange *const LROfVar = LRI.getLiveRangeForValue(*LIt ); |
| 137 | |
| 138 | // LROfVar can be null if it is a const since a const |
| 139 | // doesn't have a dominating def - see Assumptions above |
| 140 | if( LROfVar) { |
| 141 | |
| 142 | if(LROfDef == LROfVar) // do not set interf for same LR |
| 143 | continue; |
| 144 | |
| 145 | // if 2 reg classes are the same set interference |
| 146 | if( RCOfDef == LROfVar->getRegClass() ){ |
| 147 | RCOfDef->setInterference( LROfDef, LROfVar); |
| 148 | |
| 149 | } |
| 150 | |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 151 | else if(DEBUG_RA > 1) { |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 152 | // we will not have LRs for values not explicitly allocated in the |
| 153 | // instruction stream (e.g., constants) |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 154 | cout << " warning: no live range for " ; |
| 155 | printValue( *LIt); cout << endl; } |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 156 | |
Ruchira Sasanka | 958faf3 | 2001-10-19 17:21:03 +0000 | [diff] [blame] | 157 | } |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 158 | |
Ruchira Sasanka | 958faf3 | 2001-10-19 17:21:03 +0000 | [diff] [blame] | 159 | } |
| 160 | |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 161 | } |
| 162 | |
Ruchira Sasanka | 958faf3 | 2001-10-19 17:21:03 +0000 | [diff] [blame] | 163 | |
| 164 | //---------------------------------------------------------------------------- |
| 165 | // For a call instruction, this method sets the CallInterference flag in |
| 166 | // the LR of each variable live int the Live Variable Set live after the |
| 167 | // call instruction (except the return value of the call instruction - since |
| 168 | // the return value does not interfere with that call itself). |
| 169 | //---------------------------------------------------------------------------- |
| 170 | |
| 171 | void PhyRegAlloc::setCallInterferences(const MachineInstr *MInst, |
| 172 | const LiveVarSet *const LVSetAft ) |
| 173 | { |
| 174 | // Now find the LR of the return value of the call |
Ruchira Sasanka | b3b6f53 | 2001-10-21 16:43:41 +0000 | [diff] [blame] | 175 | |
Ruchira Sasanka | 958faf3 | 2001-10-19 17:21:03 +0000 | [diff] [blame] | 176 | |
| 177 | // We do this because, we look at the LV set *after* the instruction |
| 178 | // to determine, which LRs must be saved across calls. The return value |
| 179 | // of the call is live in this set - but it does not interfere with call |
| 180 | // (i.e., we can allocate a volatile register to the return value) |
| 181 | |
| 182 | LiveRange *RetValLR = NULL; |
| 183 | |
Ruchira Sasanka | b3b6f53 | 2001-10-21 16:43:41 +0000 | [diff] [blame] | 184 | const Value *RetVal = MRI.getCallInstRetVal( MInst ); |
Ruchira Sasanka | 958faf3 | 2001-10-19 17:21:03 +0000 | [diff] [blame] | 185 | |
Ruchira Sasanka | b3b6f53 | 2001-10-21 16:43:41 +0000 | [diff] [blame] | 186 | if( RetVal ) { |
| 187 | RetValLR = LRI.getLiveRangeForValue( RetVal ); |
| 188 | assert( RetValLR && "No LR for RetValue of call"); |
Ruchira Sasanka | 958faf3 | 2001-10-19 17:21:03 +0000 | [diff] [blame] | 189 | } |
| 190 | |
Ruchira Sasanka | 958faf3 | 2001-10-19 17:21:03 +0000 | [diff] [blame] | 191 | if( DEBUG_RA) |
| 192 | cout << "\n For call inst: " << *MInst; |
| 193 | |
| 194 | LiveVarSet::const_iterator LIt = LVSetAft->begin(); |
| 195 | |
| 196 | // for each live var in live variable set after machine inst |
| 197 | for( ; LIt != LVSetAft->end(); ++LIt) { |
| 198 | |
| 199 | // get the live range corresponding to live var |
| 200 | LiveRange *const LR = LRI.getLiveRangeForValue(*LIt ); |
| 201 | |
| 202 | if( LR && DEBUG_RA) { |
| 203 | cout << "\n\tLR Aft Call: "; |
| 204 | LR->printSet(); |
| 205 | } |
| 206 | |
| 207 | |
| 208 | // LR can be null if it is a const since a const |
| 209 | // doesn't have a dominating def - see Assumptions above |
| 210 | if( LR && (LR != RetValLR) ) { |
| 211 | LR->setCallInterference(); |
| 212 | if( DEBUG_RA) { |
| 213 | cout << "\n ++Added call interf for LR: " ; |
| 214 | LR->printSet(); |
| 215 | } |
| 216 | } |
| 217 | |
| 218 | } |
| 219 | |
| 220 | } |
| 221 | |
| 222 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 223 | //---------------------------------------------------------------------------- |
| 224 | // This method will walk thru code and create interferences in the IG of |
| 225 | // each RegClass. |
| 226 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 227 | |
| 228 | void PhyRegAlloc::buildInterferenceGraphs() |
| 229 | { |
| 230 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 231 | if(DEBUG_RA) cout << "Creating interference graphs ..." << endl; |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 232 | |
| 233 | Method::const_iterator BBI = Meth->begin(); // random iterator for BBs |
| 234 | |
| 235 | for( ; BBI != Meth->end(); ++BBI) { // traverse BBs in random order |
| 236 | |
| 237 | // get the iterator for machine instructions |
| 238 | const MachineCodeForBasicBlock& MIVec = (*BBI)->getMachineInstrVec(); |
| 239 | MachineCodeForBasicBlock::const_iterator |
| 240 | MInstIterator = MIVec.begin(); |
| 241 | |
| 242 | // iterate over all the machine instructions in BB |
| 243 | for( ; MInstIterator != MIVec.end(); ++MInstIterator) { |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 244 | |
Ruchira Sasanka | ef1b0cb | 2001-11-03 17:13:27 +0000 | [diff] [blame] | 245 | const MachineInstr * MInst = *MInstIterator; |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 246 | |
| 247 | // get the LV set after the instruction |
| 248 | const LiveVarSet *const LVSetAI = |
| 249 | LVI->getLiveVarSetAfterMInst(MInst, *BBI); |
| 250 | |
| 251 | const bool isCallInst = TM.getInstrInfo().isCall(MInst->getOpCode()); |
| 252 | |
Ruchira Sasanka | 958faf3 | 2001-10-19 17:21:03 +0000 | [diff] [blame] | 253 | if( isCallInst ) { |
| 254 | //cout << "\nFor call inst: " << *MInst; |
| 255 | |
| 256 | // set the isCallInterference flag of each live range wich extends |
| 257 | // accross this call instruction. This information is used by graph |
| 258 | // coloring algo to avoid allocating volatile colors to live ranges |
| 259 | // that span across calls (since they have to be saved/restored) |
| 260 | setCallInterferences( MInst, LVSetAI); |
| 261 | } |
| 262 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 263 | |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 264 | // iterate over MI operands to find defs |
| 265 | for( MachineInstr::val_op_const_iterator OpI(MInst);!OpI.done(); ++OpI) { |
| 266 | |
| 267 | if( OpI.isDef() ) { |
| 268 | // create a new LR iff this operand is a def |
| 269 | addInterference(*OpI, LVSetAI, isCallInst ); |
| 270 | |
| 271 | } //if this is a def |
| 272 | |
| 273 | } // for all operands |
| 274 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 275 | |
| 276 | // Also add interference for any implicit definitions in a machine |
| 277 | // instr (currently, only calls have this). |
| 278 | |
| 279 | unsigned NumOfImpRefs = MInst->getNumImplicitRefs(); |
| 280 | if( NumOfImpRefs > 0 ) { |
| 281 | for(unsigned z=0; z < NumOfImpRefs; z++) |
| 282 | if( MInst->implicitRefIsDefined(z) ) |
| 283 | addInterference( MInst->getImplicitRef(z), LVSetAI, isCallInst ); |
| 284 | } |
| 285 | |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 286 | /* |
Ruchira Sasanka | ef1b0cb | 2001-11-03 17:13:27 +0000 | [diff] [blame] | 287 | // record phi instrns in PhiInstList |
| 288 | if( TM.getInstrInfo().isDummyPhiInstr(MInst->getOpCode()) ) |
| 289 | PhiInstList.push_back( MInst ); |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 290 | */ |
Ruchira Sasanka | ef1b0cb | 2001-11-03 17:13:27 +0000 | [diff] [blame] | 291 | |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 292 | } // for all machine instructions in BB |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 293 | |
| 294 | } // for all BBs in method |
| 295 | |
| 296 | |
| 297 | // add interferences for method arguments. Since there are no explict |
| 298 | // defs in method for args, we have to add them manually |
| 299 | |
| 300 | addInterferencesForArgs(); // add interference for method args |
| 301 | |
| 302 | if( DEBUG_RA) |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 303 | cout << "Interference graphs calculted!" << endl; |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 304 | |
| 305 | } |
| 306 | |
| 307 | |
| 308 | |
| 309 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 310 | //---------------------------------------------------------------------------- |
| 311 | // This method will add interferences for incoming arguments to a method. |
| 312 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 313 | void PhyRegAlloc::addInterferencesForArgs() |
| 314 | { |
| 315 | // get the InSet of root BB |
| 316 | const LiveVarSet *const InSet = LVI->getInSetOfBB( Meth->front() ); |
| 317 | |
| 318 | // get the argument list |
| 319 | const Method::ArgumentListType& ArgList = Meth->getArgumentList(); |
| 320 | |
| 321 | // get an iterator to arg list |
| 322 | Method::ArgumentListType::const_iterator ArgIt = ArgList.begin(); |
| 323 | |
| 324 | |
| 325 | for( ; ArgIt != ArgList.end() ; ++ArgIt) { // for each argument |
| 326 | addInterference( *ArgIt, InSet, false ); // add interferences between |
| 327 | // args and LVars at start |
| 328 | if( DEBUG_RA > 1) { |
Ruchira Sasanka | 97b8b44 | 2001-10-18 22:36:26 +0000 | [diff] [blame] | 329 | cout << " - %% adding interference for argument "; |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 330 | printValue( (const Value *) *ArgIt); cout << endl; |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 331 | } |
| 332 | } |
| 333 | } |
| 334 | |
| 335 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 336 | //---------------------------------------------------------------------------- |
| 337 | // This method is called after register allocation is complete to set the |
| 338 | // allocated reisters in the machine code. This code will add register numbers |
| 339 | // to MachineOperands that contain a Value. |
| 340 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 8e60479 | 2001-09-14 21:18:34 +0000 | [diff] [blame] | 341 | |
| 342 | void PhyRegAlloc::updateMachineCode() |
| 343 | { |
| 344 | |
| 345 | Method::const_iterator BBI = Meth->begin(); // random iterator for BBs |
| 346 | |
| 347 | for( ; BBI != Meth->end(); ++BBI) { // traverse BBs in random order |
| 348 | |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 349 | // get the iterator for machine instructions |
| 350 | MachineCodeForBasicBlock& MIVec = (*BBI)->getMachineInstrVec(); |
| 351 | MachineCodeForBasicBlock::iterator MInstIterator = MIVec.begin(); |
| 352 | |
| 353 | // iterate over all the machine instructions in BB |
| 354 | for( ; MInstIterator != MIVec.end(); ++MInstIterator) { |
| 355 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 356 | MachineInstr *MInst = *MInstIterator; |
| 357 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 358 | // if this machine instr is call, insert caller saving code |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 359 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 360 | if( (TM.getInstrInfo()).isCall( MInst->getOpCode()) ) |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 361 | MRI.insertCallerSavingCode(MInst, *BBI, *this ); |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 362 | |
| 363 | // If there are instructions to be added, *before* this machine |
| 364 | // instruction, add them now. |
| 365 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 366 | if( AddedInstrMap[ MInst ] ) { |
| 367 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 368 | deque<MachineInstr *> &IBef = (AddedInstrMap[MInst])->InstrnsBefore; |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 369 | |
| 370 | if( ! IBef.empty() ) { |
| 371 | |
Ruchira Sasanka | a90e770 | 2001-10-15 16:26:38 +0000 | [diff] [blame] | 372 | deque<MachineInstr *>::iterator AdIt; |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 373 | |
| 374 | for( AdIt = IBef.begin(); AdIt != IBef.end() ; ++AdIt ) { |
| 375 | |
Ruchira Sasanka | 97b8b44 | 2001-10-18 22:36:26 +0000 | [diff] [blame] | 376 | if( DEBUG_RA) |
| 377 | cerr << " *$* PREPENDed instr " << *AdIt << endl; |
| 378 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 379 | MInstIterator = MIVec.insert( MInstIterator, *AdIt ); |
| 380 | ++MInstIterator; |
| 381 | } |
| 382 | |
| 383 | } |
| 384 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 385 | } |
| 386 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 387 | // reset the stack offset for temporary variables since we may |
| 388 | // need that to spill |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 389 | mcInfo.popAllTempValues(TM); |
| 390 | |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 391 | //for(MachineInstr::val_op_const_iterator OpI(MInst);!OpI.done();++OpI) { |
| 392 | |
| 393 | for(unsigned OpNum=0; OpNum < MInst->getNumOperands(); ++OpNum) { |
| 394 | |
| 395 | MachineOperand& Op = MInst->getOperand(OpNum); |
| 396 | |
| 397 | if( Op.getOperandType() == MachineOperand::MO_VirtualRegister || |
| 398 | Op.getOperandType() == MachineOperand::MO_CCRegister) { |
| 399 | |
| 400 | const Value *const Val = Op.getVRegValue(); |
| 401 | |
| 402 | // delete this condition checking later (must assert if Val is null) |
Chris Lattner | 045e7c8 | 2001-09-19 16:26:23 +0000 | [diff] [blame] | 403 | if( !Val) { |
| 404 | if (DEBUG_RA) |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 405 | cout << "Warning: NULL Value found for operand" << endl; |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 406 | continue; |
| 407 | } |
| 408 | assert( Val && "Value is NULL"); |
| 409 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 410 | LiveRange *const LR = LRI.getLiveRangeForValue(Val); |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 411 | |
| 412 | if ( !LR ) { |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 413 | |
| 414 | // nothing to worry if it's a const or a label |
| 415 | |
Chris Lattner | 4c3aaa4 | 2001-09-19 16:09:04 +0000 | [diff] [blame] | 416 | if (DEBUG_RA) { |
Ruchira Sasanka | 1b732fd | 2001-10-16 16:34:44 +0000 | [diff] [blame] | 417 | cout << "*NO LR for operand : " << Op ; |
| 418 | cout << " [reg:" << Op.getAllocatedRegNum() << "]"; |
| 419 | cout << " in inst:\t" << *MInst << endl; |
Chris Lattner | 4c3aaa4 | 2001-09-19 16:09:04 +0000 | [diff] [blame] | 420 | } |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 421 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 422 | // if register is not allocated, mark register as invalid |
Ruchira Sasanka | a90e770 | 2001-10-15 16:26:38 +0000 | [diff] [blame] | 423 | if( Op.getAllocatedRegNum() == -1) |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 424 | Op.setRegForValue( MRI.getInvalidRegNum()); |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 425 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 426 | |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 427 | continue; |
| 428 | } |
| 429 | |
| 430 | unsigned RCID = (LR->getRegClass())->getID(); |
| 431 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 432 | if( LR->hasColor() ) { |
| 433 | Op.setRegForValue( MRI.getUnifiedRegNum(RCID, LR->getColor()) ); |
| 434 | } |
| 435 | else { |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 436 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 437 | // LR did NOT receive a color (register). Now, insert spill code |
| 438 | // for spilled opeands in this machine instruction |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 439 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 440 | assert(0 && "LR must be spilled"); |
| 441 | // insertCode4SpilledLR(LR, MInst, *BBI, OpNum ); |
| 442 | |
| 443 | } |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 444 | } |
| 445 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 446 | } // for each operand |
| 447 | |
| 448 | |
| 449 | // If there are instructions to be added *after* this machine |
| 450 | // instruction, add them now |
| 451 | |
Ruchira Sasanka | 251d8db | 2001-10-23 21:38:00 +0000 | [diff] [blame] | 452 | if( AddedInstrMap[ MInst ] && |
| 453 | ! (AddedInstrMap[ MInst ]->InstrnsAfter).empty() ) { |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 454 | |
Ruchira Sasanka | 251d8db | 2001-10-23 21:38:00 +0000 | [diff] [blame] | 455 | // if there are delay slots for this instruction, the instructions |
| 456 | // added after it must really go after the delayed instruction(s) |
| 457 | // So, we move the InstrAfter of the current instruction to the |
| 458 | // corresponding delayed instruction |
| 459 | |
| 460 | unsigned delay; |
| 461 | if((delay=TM.getInstrInfo().getNumDelaySlots(MInst->getOpCode())) >0){ |
| 462 | move2DelayedInstr(MInst, *(MInstIterator+delay) ); |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 463 | |
Ruchira Sasanka | 251d8db | 2001-10-23 21:38:00 +0000 | [diff] [blame] | 464 | if(DEBUG_RA) cout<< "\nMoved an added instr after the delay slot"; |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 465 | } |
Ruchira Sasanka | 251d8db | 2001-10-23 21:38:00 +0000 | [diff] [blame] | 466 | |
| 467 | else { |
| 468 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 469 | |
Ruchira Sasanka | 251d8db | 2001-10-23 21:38:00 +0000 | [diff] [blame] | 470 | // Here we can add the "instructions after" to the current |
| 471 | // instruction since there are no delay slots for this instruction |
| 472 | |
| 473 | deque<MachineInstr *> &IAft = (AddedInstrMap[MInst])->InstrnsAfter; |
| 474 | |
| 475 | if( ! IAft.empty() ) { |
| 476 | |
| 477 | deque<MachineInstr *>::iterator AdIt; |
| 478 | |
| 479 | ++MInstIterator; // advance to the next instruction |
| 480 | |
| 481 | for( AdIt = IAft.begin(); AdIt != IAft.end() ; ++AdIt ) { |
| 482 | |
| 483 | if(DEBUG_RA) |
| 484 | cerr << " *#* APPENDed instr opcode: " << *AdIt << endl; |
| 485 | |
| 486 | MInstIterator = MIVec.insert( MInstIterator, *AdIt ); |
| 487 | ++MInstIterator; |
| 488 | } |
| 489 | |
| 490 | // MInsterator already points to the next instr. Since the |
| 491 | // for loop also increments it, decrement it to point to the |
| 492 | // instruction added last |
| 493 | --MInstIterator; |
| 494 | |
| 495 | } |
| 496 | |
| 497 | } // if not delay |
| 498 | |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 499 | } |
Ruchira Sasanka | 251d8db | 2001-10-23 21:38:00 +0000 | [diff] [blame] | 500 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 501 | } // for each machine instruction |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 502 | } |
| 503 | } |
| 504 | |
| 505 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 506 | //---------------------------------------------------------------------------- |
| 507 | // We can use the following method to get a temporary register to be used |
| 508 | // BEFORE any given machine instruction. If there is a register available, |
| 509 | // this method will simply return that register and set MIBef = MIAft = NULL. |
| 510 | // Otherwise, it will return a register and MIAft and MIBef will contain |
| 511 | // two instructions used to free up this returned register. |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 512 | // Returned register number is the UNIFIED register number |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 513 | //---------------------------------------------------------------------------- |
| 514 | |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 515 | int PhyRegAlloc::getUsableRegAtMI(RegClass *RC, |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 516 | const int RegType, |
| 517 | const MachineInstr *MInst, |
| 518 | const LiveVarSet *LVSetBef, |
| 519 | MachineInstr *MIBef, |
| 520 | MachineInstr *MIAft) { |
| 521 | |
| 522 | int Reg = getUnusedRegAtMI(RC, MInst, LVSetBef); |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 523 | Reg = MRI.getUnifiedRegNum(RC->getID(), Reg); |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 524 | |
| 525 | if( Reg != -1) { |
| 526 | // we found an unused register, so we can simply used |
| 527 | MIBef = MIAft = NULL; |
| 528 | } |
| 529 | else { |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 530 | // we couldn't find an unused register. Generate code to free up a reg by |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 531 | // saving it on stack and restoring after the instruction |
| 532 | |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 533 | /**** NOTE: THIS SHOULD USE THE RIGHT SIZE FOR THE REG BEING PUSHED ****/ |
| 534 | int TmpOff = mcInfo.pushTempValue(TM, /*size*/ 8); |
| 535 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 536 | Reg = getRegNotUsedByThisInst(RC, MInst); |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 537 | MIBef = MRI.cpReg2MemMI(Reg, MRI.getFramePointer(), TmpOff, RegType ); |
| 538 | MIAft = MRI.cpMem2RegMI(MRI.getFramePointer(), TmpOff, Reg, RegType ); |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 539 | } |
| 540 | |
| 541 | return Reg; |
| 542 | } |
| 543 | |
| 544 | //---------------------------------------------------------------------------- |
| 545 | // This method is called to get a new unused register that can be used to |
| 546 | // accomodate a spilled value. |
| 547 | // This method may be called several times for a single machine instruction |
| 548 | // if it contains many spilled operands. Each time it is called, it finds |
| 549 | // a register which is not live at that instruction and also which is not |
| 550 | // used by other spilled operands of the same instruction. |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 551 | // Return register number is relative to the register class. NOT |
| 552 | // unified number |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 553 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 554 | int PhyRegAlloc::getUnusedRegAtMI(RegClass *RC, |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 555 | const MachineInstr *MInst, |
| 556 | const LiveVarSet *LVSetBef) { |
| 557 | |
| 558 | unsigned NumAvailRegs = RC->getNumOfAvailRegs(); |
| 559 | |
| 560 | bool *IsColorUsedArr = RC->getIsColorUsedArr(); |
| 561 | |
Ruchira Sasanka | 80b1a1a | 2001-11-03 20:41:22 +0000 | [diff] [blame] | 562 | for(unsigned i=0; i < NumAvailRegs; i++) |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 563 | IsColorUsedArr[i] = false; |
| 564 | |
| 565 | LiveVarSet::const_iterator LIt = LVSetBef->begin(); |
| 566 | |
| 567 | // for each live var in live variable set after machine inst |
| 568 | for( ; LIt != LVSetBef->end(); ++LIt) { |
| 569 | |
| 570 | // get the live range corresponding to live var |
| 571 | LiveRange *const LRofLV = LRI.getLiveRangeForValue(*LIt ); |
| 572 | |
| 573 | // LR can be null if it is a const since a const |
| 574 | // doesn't have a dominating def - see Assumptions above |
| 575 | if( LRofLV ) |
| 576 | if( LRofLV->hasColor() ) |
| 577 | IsColorUsedArr[ LRofLV->getColor() ] = true; |
| 578 | } |
| 579 | |
| 580 | // It is possible that one operand of this MInst was already spilled |
| 581 | // and it received some register temporarily. If that's the case, |
| 582 | // it is recorded in machine operand. We must skip such registers. |
| 583 | |
| 584 | setRegsUsedByThisInst(RC, MInst); |
| 585 | |
| 586 | unsigned c; // find first unused color |
| 587 | for( c=0; c < NumAvailRegs; c++) |
| 588 | if( ! IsColorUsedArr[ c ] ) break; |
| 589 | |
| 590 | if(c < NumAvailRegs) |
| 591 | return c; |
| 592 | else |
| 593 | return -1; |
| 594 | |
| 595 | |
| 596 | } |
| 597 | |
| 598 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 599 | |
| 600 | //---------------------------------------------------------------------------- |
| 601 | // This method modifies the IsColorUsedArr of the register class passed to it. |
| 602 | // It sets the bits corresponding to the registers used by this machine |
| 603 | // instructions. Explicit operands are set. |
| 604 | //---------------------------------------------------------------------------- |
| 605 | void PhyRegAlloc::setRegsUsedByThisInst(RegClass *RC, |
| 606 | const MachineInstr *MInst ) { |
| 607 | |
| 608 | bool *IsColorUsedArr = RC->getIsColorUsedArr(); |
| 609 | |
| 610 | for(unsigned OpNum=0; OpNum < MInst->getNumOperands(); ++OpNum) { |
| 611 | |
| 612 | const MachineOperand& Op = MInst->getOperand(OpNum); |
| 613 | |
| 614 | if( Op.getOperandType() == MachineOperand::MO_VirtualRegister || |
| 615 | Op.getOperandType() == MachineOperand::MO_CCRegister) { |
| 616 | |
| 617 | const Value *const Val = Op.getVRegValue(); |
| 618 | |
| 619 | if( !Val ) |
| 620 | if( MRI.getRegClassIDOfValue( Val )== RC->getID() ) { |
| 621 | int Reg; |
| 622 | if( (Reg=Op.getAllocatedRegNum()) != -1) |
| 623 | IsColorUsedArr[ Reg ] = true; |
| 624 | |
| 625 | } |
| 626 | } |
| 627 | } |
| 628 | |
| 629 | // If there are implicit references, mark them as well |
| 630 | |
| 631 | for(unsigned z=0; z < MInst->getNumImplicitRefs(); z++) { |
| 632 | |
| 633 | LiveRange *const LRofImpRef = |
| 634 | LRI.getLiveRangeForValue( MInst->getImplicitRef(z) ); |
| 635 | |
| 636 | if( LRofImpRef ) |
| 637 | if( LRofImpRef->hasColor() ) |
| 638 | IsColorUsedArr[ LRofImpRef->getColor() ] = true; |
| 639 | } |
| 640 | |
| 641 | |
| 642 | |
| 643 | } |
| 644 | |
| 645 | |
| 646 | |
| 647 | //---------------------------------------------------------------------------- |
| 648 | // Get any other register in a register class, other than what is used |
| 649 | // by operands of a machine instruction. |
| 650 | //---------------------------------------------------------------------------- |
| 651 | int PhyRegAlloc::getRegNotUsedByThisInst(RegClass *RC, |
| 652 | const MachineInstr *MInst) { |
| 653 | |
| 654 | bool *IsColorUsedArr = RC->getIsColorUsedArr(); |
| 655 | unsigned NumAvailRegs = RC->getNumOfAvailRegs(); |
| 656 | |
| 657 | |
| 658 | for(unsigned i=0; i < NumAvailRegs ; i++) |
| 659 | IsColorUsedArr[i] = false; |
| 660 | |
| 661 | setRegsUsedByThisInst(RC, MInst); |
| 662 | |
| 663 | unsigned c; // find first unused color |
| 664 | for( c=0; c < RC->getNumOfAvailRegs(); c++) |
| 665 | if( ! IsColorUsedArr[ c ] ) break; |
| 666 | |
| 667 | if(c < NumAvailRegs) |
| 668 | return c; |
| 669 | else |
| 670 | assert( 0 && "FATAL: No free register could be found in reg class!!"); |
| 671 | |
| 672 | } |
| 673 | |
| 674 | |
| 675 | |
| 676 | |
| 677 | |
| 678 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 251d8db | 2001-10-23 21:38:00 +0000 | [diff] [blame] | 679 | // If there are delay slots for an instruction, the instructions |
| 680 | // added after it must really go after the delayed instruction(s). |
| 681 | // So, we move the InstrAfter of that instruction to the |
| 682 | // corresponding delayed instruction using the following method. |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 683 | |
Ruchira Sasanka | 251d8db | 2001-10-23 21:38:00 +0000 | [diff] [blame] | 684 | //---------------------------------------------------------------------------- |
| 685 | void PhyRegAlloc:: move2DelayedInstr(const MachineInstr *OrigMI, |
| 686 | const MachineInstr *DelayedMI) { |
| 687 | |
| 688 | |
| 689 | // "added after" instructions of the original instr |
| 690 | deque<MachineInstr *> &OrigAft = (AddedInstrMap[OrigMI])->InstrnsAfter; |
| 691 | |
| 692 | // "added instructions" of the delayed instr |
| 693 | AddedInstrns *DelayAdI = AddedInstrMap[DelayedMI]; |
| 694 | |
| 695 | if(! DelayAdI ) { // create a new "added after" if necessary |
| 696 | DelayAdI = new AddedInstrns(); |
| 697 | AddedInstrMap[DelayedMI] = DelayAdI; |
| 698 | } |
| 699 | |
| 700 | // "added after" instructions of the delayed instr |
| 701 | deque<MachineInstr *> &DelayedAft = DelayAdI->InstrnsAfter; |
| 702 | |
| 703 | // go thru all the "added after instructions" of the original instruction |
| 704 | // and append them to the "addded after instructions" of the delayed |
| 705 | // instructions |
| 706 | |
| 707 | deque<MachineInstr *>::iterator OrigAdIt; |
| 708 | |
| 709 | for( OrigAdIt = OrigAft.begin(); OrigAdIt != OrigAft.end() ; ++OrigAdIt ) { |
| 710 | DelayedAft.push_back( *OrigAdIt ); |
| 711 | } |
| 712 | |
| 713 | // empty the "added after instructions" of the original instruction |
| 714 | OrigAft.clear(); |
| 715 | |
| 716 | } |
Ruchira Sasanka | 0931a01 | 2001-09-15 19:06:58 +0000 | [diff] [blame] | 717 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 718 | //---------------------------------------------------------------------------- |
| 719 | // This method prints the code with registers after register allocation is |
| 720 | // complete. |
| 721 | //---------------------------------------------------------------------------- |
| 722 | void PhyRegAlloc::printMachineCode() |
| 723 | { |
| 724 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 725 | cout << endl << ";************** Method "; |
| 726 | cout << Meth->getName() << " *****************" << endl; |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 727 | |
| 728 | Method::const_iterator BBI = Meth->begin(); // random iterator for BBs |
| 729 | |
| 730 | for( ; BBI != Meth->end(); ++BBI) { // traverse BBs in random order |
| 731 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 732 | cout << endl ; printLabel( *BBI); cout << ": "; |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 733 | |
| 734 | // get the iterator for machine instructions |
| 735 | MachineCodeForBasicBlock& MIVec = (*BBI)->getMachineInstrVec(); |
| 736 | MachineCodeForBasicBlock::iterator MInstIterator = MIVec.begin(); |
| 737 | |
| 738 | // iterate over all the machine instructions in BB |
| 739 | for( ; MInstIterator != MIVec.end(); ++MInstIterator) { |
| 740 | |
| 741 | MachineInstr *const MInst = *MInstIterator; |
| 742 | |
| 743 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 744 | cout << endl << "\t"; |
| 745 | cout << TargetInstrDescriptors[MInst->getOpCode()].opCodeString; |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 746 | |
| 747 | |
| 748 | //for(MachineInstr::val_op_const_iterator OpI(MInst);!OpI.done();++OpI) { |
| 749 | |
| 750 | for(unsigned OpNum=0; OpNum < MInst->getNumOperands(); ++OpNum) { |
| 751 | |
| 752 | MachineOperand& Op = MInst->getOperand(OpNum); |
| 753 | |
| 754 | if( Op.getOperandType() == MachineOperand::MO_VirtualRegister || |
Ruchira Sasanka | 97b8b44 | 2001-10-18 22:36:26 +0000 | [diff] [blame] | 755 | Op.getOperandType() == MachineOperand::MO_CCRegister /*|| |
| 756 | Op.getOperandType() == MachineOperand::MO_PCRelativeDisp*/ ) { |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 757 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 758 | const Value *const Val = Op.getVRegValue () ; |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 759 | // ****this code is temporary till NULL Values are fixed |
| 760 | if( ! Val ) { |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 761 | cout << "\t<*NULL*>"; |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 762 | continue; |
| 763 | } |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 764 | |
| 765 | // if a label or a constant |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 766 | if( (Val->getValueType() == Value::BasicBlockVal) ) { |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 767 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 768 | cout << "\t"; printLabel( Op.getVRegValue () ); |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 769 | } |
| 770 | else { |
| 771 | // else it must be a register value |
| 772 | const int RegNum = Op.getAllocatedRegNum(); |
| 773 | |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 774 | cout << "\t" << "%" << MRI.getUnifiedRegName( RegNum ); |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 775 | } |
| 776 | |
| 777 | } |
| 778 | else if(Op.getOperandType() == MachineOperand::MO_MachineRegister) { |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 779 | cout << "\t" << "%" << MRI.getUnifiedRegName(Op.getMachineRegNum()); |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 780 | } |
| 781 | |
| 782 | else |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 783 | cout << "\t" << Op; // use dump field |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 784 | } |
| 785 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 786 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 787 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 788 | unsigned NumOfImpRefs = MInst->getNumImplicitRefs(); |
| 789 | if( NumOfImpRefs > 0 ) { |
| 790 | |
| 791 | cout << "\tImplicit:"; |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 792 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 793 | for(unsigned z=0; z < NumOfImpRefs; z++) { |
| 794 | printValue( MInst->getImplicitRef(z) ); |
| 795 | cout << "\t"; |
| 796 | } |
| 797 | |
| 798 | } |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 799 | |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 800 | } // for all machine instructions |
| 801 | |
| 802 | |
| 803 | cout << endl; |
| 804 | |
| 805 | } // for all BBs |
| 806 | |
| 807 | cout << endl; |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 808 | } |
| 809 | |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 810 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 811 | //---------------------------------------------------------------------------- |
| 812 | // |
| 813 | //---------------------------------------------------------------------------- |
| 814 | |
| 815 | void PhyRegAlloc::colorCallRetArgs() |
| 816 | { |
| 817 | |
| 818 | CallRetInstrListType &CallRetInstList = LRI.getCallRetInstrList(); |
| 819 | CallRetInstrListType::const_iterator It = CallRetInstList.begin(); |
| 820 | |
| 821 | for( ; It != CallRetInstList.end(); ++It ) { |
| 822 | |
Ruchira Sasanka | a90e770 | 2001-10-15 16:26:38 +0000 | [diff] [blame] | 823 | const MachineInstr *const CRMI = *It; |
| 824 | unsigned OpCode = CRMI->getOpCode(); |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 825 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 826 | // get the added instructions for this Call/Ret instruciton |
| 827 | AddedInstrns *AI = AddedInstrMap[ CRMI ]; |
| 828 | if ( !AI ) { |
| 829 | AI = new AddedInstrns(); |
| 830 | AddedInstrMap[ CRMI ] = AI; |
| 831 | } |
| 832 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 833 | // Tmp stack poistions are needed by some calls that have spilled args |
| 834 | // So reset it before we call each such method |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 835 | mcInfo.popAllTempValues(TM); |
| 836 | |
Ruchira Sasanka | a90e770 | 2001-10-15 16:26:38 +0000 | [diff] [blame] | 837 | if( (TM.getInstrInfo()).isCall( OpCode ) ) |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 838 | MRI.colorCallArgs( CRMI, LRI, AI, *this ); |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 839 | |
Ruchira Sasanka | a90e770 | 2001-10-15 16:26:38 +0000 | [diff] [blame] | 840 | else if ( (TM.getInstrInfo()).isReturn(OpCode) ) |
| 841 | MRI.colorRetValue( CRMI, LRI, AI ); |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 842 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 843 | else assert( 0 && "Non Call/Ret instrn in CallRetInstrList\n" ); |
| 844 | |
| 845 | } |
| 846 | |
| 847 | } |
| 848 | |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 849 | |
| 850 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 851 | //---------------------------------------------------------------------------- |
| 852 | |
| 853 | //---------------------------------------------------------------------------- |
| 854 | void PhyRegAlloc::colorIncomingArgs() |
| 855 | { |
| 856 | const BasicBlock *const FirstBB = Meth->front(); |
| 857 | const MachineInstr *FirstMI = *((FirstBB->getMachineInstrVec()).begin()); |
| 858 | assert( FirstMI && "No machine instruction in entry BB"); |
| 859 | |
| 860 | AddedInstrns *AI = AddedInstrMap[ FirstMI ]; |
| 861 | if ( !AI ) { |
| 862 | AI = new AddedInstrns(); |
| 863 | AddedInstrMap[ FirstMI ] = AI; |
| 864 | } |
| 865 | |
| 866 | MRI.colorMethodArgs(Meth, LRI, AI ); |
| 867 | } |
| 868 | |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 869 | |
| 870 | //---------------------------------------------------------------------------- |
| 871 | // Used to generate a label for a basic block |
| 872 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 873 | void PhyRegAlloc::printLabel(const Value *const Val) |
| 874 | { |
| 875 | if( Val->hasName() ) |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 876 | cout << Val->getName(); |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 877 | else |
Ruchira Sasanka | c4d4b76 | 2001-10-16 01:23:19 +0000 | [diff] [blame] | 878 | cout << "Label" << Val; |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 879 | } |
| 880 | |
| 881 | |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 882 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 883 | // This method calls setSugColorUsable method of each live range. This |
| 884 | // will determine whether the suggested color of LR is really usable. |
| 885 | // A suggested color is not usable when the suggested color is volatile |
| 886 | // AND when there are call interferences |
| 887 | //---------------------------------------------------------------------------- |
| 888 | |
| 889 | void PhyRegAlloc::markUnusableSugColors() |
| 890 | { |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 891 | if(DEBUG_RA ) cout << "\nmarking unusable suggested colors ..." << endl; |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 892 | |
| 893 | // hash map iterator |
| 894 | LiveRangeMapType::const_iterator HMI = (LRI.getLiveRangeMap())->begin(); |
| 895 | LiveRangeMapType::const_iterator HMIEnd = (LRI.getLiveRangeMap())->end(); |
| 896 | |
| 897 | for( ; HMI != HMIEnd ; ++HMI ) { |
| 898 | |
| 899 | if( (*HMI).first ) { |
| 900 | |
| 901 | LiveRange *L = (*HMI).second; // get the LiveRange |
| 902 | |
| 903 | if(L) { |
| 904 | if( L->hasSuggestedColor() ) { |
| 905 | |
| 906 | int RCID = (L->getRegClass())->getID(); |
| 907 | if( MRI.isRegVolatile( RCID, L->getSuggestedColor()) && |
| 908 | L->isCallInterference() ) |
| 909 | L->setSuggestedColorUsable( false ); |
| 910 | else |
| 911 | L->setSuggestedColorUsable( true ); |
| 912 | } |
| 913 | } // if L->hasSuggestedColor() |
| 914 | } |
| 915 | } // for all LR's in hash map |
| 916 | } |
| 917 | |
| 918 | |
| 919 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 920 | //---------------------------------------------------------------------------- |
| 921 | // The following method will set the stack offsets of the live ranges that |
| 922 | // are decided to be spillled. This must be called just after coloring the |
| 923 | // LRs using the graph coloring algo. For each live range that is spilled, |
| 924 | // this method allocate a new spill position on the stack. |
| 925 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 926 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 927 | void PhyRegAlloc::allocateStackSpace4SpilledLRs() |
| 928 | { |
| 929 | if(DEBUG_RA ) cout << "\nsetting LR stack offsets ..." << endl; |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 930 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 931 | // hash map iterator |
| 932 | LiveRangeMapType::const_iterator HMI = (LRI.getLiveRangeMap())->begin(); |
| 933 | LiveRangeMapType::const_iterator HMIEnd = (LRI.getLiveRangeMap())->end(); |
| 934 | |
| 935 | for( ; HMI != HMIEnd ; ++HMI ) { |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 936 | if( (*HMI).first ) { |
| 937 | LiveRange *L = (*HMI).second; // get the LiveRange |
| 938 | if(L) |
| 939 | if( ! L->hasColor() ) |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 940 | L->setSpillOffFromFP(mcInfo.allocateSpilledValue(TM,L->getType())); |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 941 | } |
| 942 | } // for all LR's in hash map |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 943 | } |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 944 | |
| 945 | |
| 946 | |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 947 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 948 | // The entry pont to Register Allocation |
| 949 | //---------------------------------------------------------------------------- |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 950 | |
| 951 | void PhyRegAlloc::allocateRegisters() |
| 952 | { |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 953 | |
| 954 | // make sure that we put all register classes into the RegClassList |
| 955 | // before we call constructLiveRanges (now done in the constructor of |
| 956 | // PhyRegAlloc class). |
| 957 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 958 | constructLiveRanges(); // create LR info |
| 959 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 960 | if( DEBUG_RA ) |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 961 | LRI.printLiveRanges(); |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 962 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 963 | createIGNodeListsAndIGs(); // create IGNode list and IGs |
| 964 | |
| 965 | buildInterferenceGraphs(); // build IGs in all reg classes |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 966 | |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 967 | |
| 968 | if( DEBUG_RA ) { |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 969 | // print all LRs in all reg classes |
| 970 | for( unsigned int rc=0; rc < NumOfRegClasses ; rc++) |
| 971 | RegClassList[ rc ]->printIGNodeList(); |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 972 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 973 | // print IGs in all register classes |
| 974 | for( unsigned int rc=0; rc < NumOfRegClasses ; rc++) |
| 975 | RegClassList[ rc ]->printIG(); |
| 976 | } |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 977 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 978 | LRI.coalesceLRs(); // coalesce all live ranges |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 979 | |
Ruchira Sasanka | ef1b0cb | 2001-11-03 17:13:27 +0000 | [diff] [blame] | 980 | // coalscing could not get rid of all phi's, add phi elimination |
| 981 | // instructions |
| 982 | // insertPhiEleminateInstrns(); |
| 983 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 984 | if( DEBUG_RA) { |
| 985 | // print all LRs in all reg classes |
| 986 | for( unsigned int rc=0; rc < NumOfRegClasses ; rc++) |
| 987 | RegClassList[ rc ]->printIGNodeList(); |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 988 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 989 | // print IGs in all register classes |
| 990 | for( unsigned int rc=0; rc < NumOfRegClasses ; rc++) |
| 991 | RegClassList[ rc ]->printIG(); |
| 992 | } |
| 993 | |
Ruchira Sasanka | 0e62aa6 | 2001-10-19 21:39:31 +0000 | [diff] [blame] | 994 | |
| 995 | // mark un-usable suggested color before graph coloring algorithm. |
| 996 | // When this is done, the graph coloring algo will not reserve |
| 997 | // suggested color unnecessarily - they can be used by another LR |
| 998 | markUnusableSugColors(); |
| 999 | |
| 1000 | // color all register classes using the graph coloring algo |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 1001 | for( unsigned int rc=0; rc < NumOfRegClasses ; rc++) |
| 1002 | RegClassList[ rc ]->colorAllRegs(); |
| 1003 | |
Ruchira Sasanka | 174bded | 2001-10-28 18:12:02 +0000 | [diff] [blame] | 1004 | // Atter grpah coloring, if some LRs did not receive a color (i.e, spilled) |
| 1005 | // a poistion for such spilled LRs |
| 1006 | allocateStackSpace4SpilledLRs(); |
Ruchira Sasanka | a5ab964 | 2001-09-30 23:11:59 +0000 | [diff] [blame] | 1007 | |
| 1008 | // color incoming args and call args |
| 1009 | colorIncomingArgs(); |
| 1010 | colorCallRetArgs(); |
| 1011 | |
Ruchira Sasanka | 97b8b44 | 2001-10-18 22:36:26 +0000 | [diff] [blame] | 1012 | |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 1013 | updateMachineCode(); |
Chris Lattner | 045e7c8 | 2001-09-19 16:26:23 +0000 | [diff] [blame] | 1014 | if (DEBUG_RA) { |
Vikram S. Adve | 12af164 | 2001-11-08 04:48:50 +0000 | [diff] [blame] | 1015 | MachineCodeForMethod::get(Meth).dump(); |
Chris Lattner | 045e7c8 | 2001-09-19 16:26:23 +0000 | [diff] [blame] | 1016 | printMachineCode(); // only for DEBUGGING |
| 1017 | } |
Ruchira Sasanka | 6b0a8b5 | 2001-09-15 21:11:11 +0000 | [diff] [blame] | 1018 | } |
| 1019 | |
Ruchira Sasanka | e727f85 | 2001-09-18 22:43:57 +0000 | [diff] [blame] | 1020 | |
| 1021 | |