github.com/mattn/go@v0.0.0-20171011075504-07f7db3ea99f/src/cmd/compile/internal/ssa/regalloc.go (about) 1 // Copyright 2015 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 // Register allocation. 6 // 7 // We use a version of a linear scan register allocator. We treat the 8 // whole function as a single long basic block and run through 9 // it using a greedy register allocator. Then all merge edges 10 // (those targeting a block with len(Preds)>1) are processed to 11 // shuffle data into the place that the target of the edge expects. 12 // 13 // The greedy allocator moves values into registers just before they 14 // are used, spills registers only when necessary, and spills the 15 // value whose next use is farthest in the future. 16 // 17 // The register allocator requires that a block is not scheduled until 18 // at least one of its predecessors have been scheduled. The most recent 19 // such predecessor provides the starting register state for a block. 20 // 21 // It also requires that there are no critical edges (critical = 22 // comes from a block with >1 successor and goes to a block with >1 23 // predecessor). This makes it easy to add fixup code on merge edges - 24 // the source of a merge edge has only one successor, so we can add 25 // fixup code to the end of that block. 26 27 // Spilling 28 // 29 // During the normal course of the allocator, we might throw a still-live 30 // value out of all registers. When that value is subsequently used, we must 31 // load it from a slot on the stack. We must also issue an instruction to 32 // initialize that stack location with a copy of v. 33 // 34 // pre-regalloc: 35 // (1) v = Op ... 36 // (2) x = Op ... 37 // (3) ... = Op v ... 38 // 39 // post-regalloc: 40 // (1) v = Op ... : AX // computes v, store result in AX 41 // s = StoreReg v // spill v to a stack slot 42 // (2) x = Op ... : AX // some other op uses AX 43 // c = LoadReg s : CX // restore v from stack slot 44 // (3) ... = Op c ... // use the restored value 45 // 46 // Allocation occurs normally until we reach (3) and we realize we have 47 // a use of v and it isn't in any register. At that point, we allocate 48 // a spill (a StoreReg) for v. We can't determine the correct place for 49 // the spill at this point, so we allocate the spill as blockless initially. 50 // The restore is then generated to load v back into a register so it can 51 // be used. Subsequent uses of v will use the restored value c instead. 52 // 53 // What remains is the question of where to schedule the spill. 54 // During allocation, we keep track of the dominator of all restores of v. 55 // The spill of v must dominate that block. The spill must also be issued at 56 // a point where v is still in a register. 57 // 58 // To find the right place, start at b, the block which dominates all restores. 59 // - If b is v.Block, then issue the spill right after v. 60 // It is known to be in a register at that point, and dominates any restores. 61 // - Otherwise, if v is in a register at the start of b, 62 // put the spill of v at the start of b. 63 // - Otherwise, set b = immediate dominator of b, and repeat. 64 // 65 // Phi values are special, as always. We define two kinds of phis, those 66 // where the merge happens in a register (a "register" phi) and those where 67 // the merge happens in a stack location (a "stack" phi). 68 // 69 // A register phi must have the phi and all of its inputs allocated to the 70 // same register. Register phis are spilled similarly to regular ops. 71 // 72 // A stack phi must have the phi and all of its inputs allocated to the same 73 // stack location. Stack phis start out life already spilled - each phi 74 // input must be a store (using StoreReg) at the end of the corresponding 75 // predecessor block. 76 // b1: y = ... : AX b2: z = ... : BX 77 // y2 = StoreReg y z2 = StoreReg z 78 // goto b3 goto b3 79 // b3: x = phi(y2, z2) 80 // The stack allocator knows that StoreReg args of stack-allocated phis 81 // must be allocated to the same stack slot as the phi that uses them. 82 // x is now a spilled value and a restore must appear before its first use. 83 84 // TODO 85 86 // Use an affinity graph to mark two values which should use the 87 // same register. This affinity graph will be used to prefer certain 88 // registers for allocation. This affinity helps eliminate moves that 89 // are required for phi implementations and helps generate allocations 90 // for 2-register architectures. 91 92 // Note: regalloc generates a not-quite-SSA output. If we have: 93 // 94 // b1: x = ... : AX 95 // x2 = StoreReg x 96 // ... AX gets reused for something else ... 97 // if ... goto b3 else b4 98 // 99 // b3: x3 = LoadReg x2 : BX b4: x4 = LoadReg x2 : CX 100 // ... use x3 ... ... use x4 ... 101 // 102 // b2: ... use x3 ... 103 // 104 // If b3 is the primary predecessor of b2, then we use x3 in b2 and 105 // add a x4:CX->BX copy at the end of b4. 106 // But the definition of x3 doesn't dominate b2. We should really 107 // insert a dummy phi at the start of b2 (x5=phi(x3,x4):BX) to keep 108 // SSA form. For now, we ignore this problem as remaining in strict 109 // SSA form isn't needed after regalloc. We'll just leave the use 110 // of x3 not dominated by the definition of x3, and the CX->BX copy 111 // will have no use (so don't run deadcode after regalloc!). 112 // TODO: maybe we should introduce these extra phis? 113 114 package ssa 115 116 import ( 117 "cmd/compile/internal/types" 118 "cmd/internal/objabi" 119 "cmd/internal/src" 120 "fmt" 121 "unsafe" 122 ) 123 124 const ( 125 moveSpills = iota 126 logSpills 127 regDebug 128 stackDebug 129 ) 130 131 // distance is a measure of how far into the future values are used. 132 // distance is measured in units of instructions. 133 const ( 134 likelyDistance = 1 135 normalDistance = 10 136 unlikelyDistance = 100 137 ) 138 139 // regalloc performs register allocation on f. It sets f.RegAlloc 140 // to the resulting allocation. 141 func regalloc(f *Func) { 142 var s regAllocState 143 s.init(f) 144 s.regalloc(f) 145 } 146 147 type register uint8 148 149 const noRegister register = 255 150 151 type regMask uint64 152 153 func (m regMask) String() string { 154 s := "" 155 for r := register(0); m != 0; r++ { 156 if m>>r&1 == 0 { 157 continue 158 } 159 m &^= regMask(1) << r 160 if s != "" { 161 s += " " 162 } 163 s += fmt.Sprintf("r%d", r) 164 } 165 return s 166 } 167 168 // countRegs returns the number of set bits in the register mask. 169 func countRegs(r regMask) int { 170 n := 0 171 for r != 0 { 172 n += int(r & 1) 173 r >>= 1 174 } 175 return n 176 } 177 178 // pickReg picks an arbitrary register from the register mask. 179 func pickReg(r regMask) register { 180 // pick the lowest one 181 if r == 0 { 182 panic("can't pick a register from an empty set") 183 } 184 for i := register(0); ; i++ { 185 if r&1 != 0 { 186 return i 187 } 188 r >>= 1 189 } 190 } 191 192 type use struct { 193 dist int32 // distance from start of the block to a use of a value 194 pos src.XPos // source position of the use 195 next *use // linked list of uses of a value in nondecreasing dist order 196 } 197 198 // A valState records the register allocation state for a (pre-regalloc) value. 199 type valState struct { 200 regs regMask // the set of registers holding a Value (usually just one) 201 uses *use // list of uses in this block 202 spill *Value // spilled copy of the Value (if any) 203 restoreMin int32 // minimum of all restores' blocks' sdom.entry 204 restoreMax int32 // maximum of all restores' blocks' sdom.exit 205 needReg bool // cached value of !v.Type.IsMemory() && !v.Type.IsVoid() && !.v.Type.IsFlags() 206 rematerializeable bool // cached value of v.rematerializeable() 207 } 208 209 type regState struct { 210 v *Value // Original (preregalloc) Value stored in this register. 211 c *Value // A Value equal to v which is currently in a register. Might be v or a copy of it. 212 // If a register is unused, v==c==nil 213 } 214 215 type regAllocState struct { 216 f *Func 217 218 sdom SparseTree 219 registers []Register 220 numRegs register 221 SPReg register 222 SBReg register 223 GReg register 224 allocatable regMask 225 226 // for each block, its primary predecessor. 227 // A predecessor of b is primary if it is the closest 228 // predecessor that appears before b in the layout order. 229 // We record the index in the Preds list where the primary predecessor sits. 230 primary []int32 231 232 // live values at the end of each block. live[b.ID] is a list of value IDs 233 // which are live at the end of b, together with a count of how many instructions 234 // forward to the next use. 235 live [][]liveInfo 236 // desired register assignments at the end of each block. 237 // Note that this is a static map computed before allocation occurs. Dynamic 238 // register desires (from partially completed allocations) will trump 239 // this information. 240 desired []desiredState 241 242 // current state of each (preregalloc) Value 243 values []valState 244 245 // names associated with each Value 246 valueNames [][]LocalSlot 247 248 // ID of SP, SB values 249 sp, sb ID 250 251 // For each Value, map from its value ID back to the 252 // preregalloc Value it was derived from. 253 orig []*Value 254 255 // current state of each register 256 regs []regState 257 258 // registers that contain values which can't be kicked out 259 nospill regMask 260 261 // mask of registers currently in use 262 used regMask 263 264 // mask of registers used in the current instruction 265 tmpused regMask 266 267 // current block we're working on 268 curBlock *Block 269 270 // cache of use records 271 freeUseRecords *use 272 273 // endRegs[blockid] is the register state at the end of each block. 274 // encoded as a set of endReg records. 275 endRegs [][]endReg 276 277 // startRegs[blockid] is the register state at the start of merge blocks. 278 // saved state does not include the state of phi ops in the block. 279 startRegs [][]startReg 280 281 // spillLive[blockid] is the set of live spills at the end of each block 282 spillLive [][]ID 283 284 // a set of copies we generated to move things around, and 285 // whether it is used in shuffle. Unused copies will be deleted. 286 copies map[*Value]bool 287 288 loopnest *loopnest 289 } 290 291 type endReg struct { 292 r register 293 v *Value // pre-regalloc value held in this register (TODO: can we use ID here?) 294 c *Value // cached version of the value 295 } 296 297 type startReg struct { 298 r register 299 v *Value // pre-regalloc value needed in this register 300 c *Value // cached version of the value 301 pos src.XPos // source position of use of this register 302 } 303 304 // freeReg frees up register r. Any current user of r is kicked out. 305 func (s *regAllocState) freeReg(r register) { 306 s.freeOrResetReg(r, false) 307 } 308 309 // freeOrResetReg frees up register r. Any current user of r is kicked out. 310 // resetting indicates that the operation is only for bookkeeping, 311 // e.g. when clearing out state upon entry to a new block. 312 func (s *regAllocState) freeOrResetReg(r register, resetting bool) { 313 v := s.regs[r].v 314 if v == nil { 315 s.f.Fatalf("tried to free an already free register %d\n", r) 316 } 317 318 // Mark r as unused. 319 if s.f.pass.debug > regDebug { 320 fmt.Printf("freeReg %s (dump %s/%s)\n", &s.registers[r], v, s.regs[r].c) 321 } 322 if !resetting && s.f.Config.ctxt.Flag_locationlists && len(s.valueNames[v.ID]) != 0 { 323 kill := s.curBlock.NewValue0(src.NoXPos, OpRegKill, types.TypeVoid) 324 for int(kill.ID) >= len(s.orig) { 325 s.orig = append(s.orig, nil) 326 } 327 for _, name := range s.valueNames[v.ID] { 328 s.f.NamedValues[name] = append(s.f.NamedValues[name], kill) 329 } 330 s.f.setHome(kill, &s.registers[r]) 331 } 332 s.regs[r] = regState{} 333 s.values[v.ID].regs &^= regMask(1) << r 334 s.used &^= regMask(1) << r 335 } 336 337 // freeRegs frees up all registers listed in m. 338 func (s *regAllocState) freeRegs(m regMask) { 339 for m&s.used != 0 { 340 s.freeReg(pickReg(m & s.used)) 341 } 342 } 343 344 // setOrig records that c's original value is the same as 345 // v's original value. 346 func (s *regAllocState) setOrig(c *Value, v *Value) { 347 for int(c.ID) >= len(s.orig) { 348 s.orig = append(s.orig, nil) 349 } 350 if s.orig[c.ID] != nil { 351 s.f.Fatalf("orig value set twice %s %s", c, v) 352 } 353 s.orig[c.ID] = s.orig[v.ID] 354 } 355 356 // assignReg assigns register r to hold c, a copy of v. 357 // r must be unused. 358 func (s *regAllocState) assignReg(r register, v *Value, c *Value) { 359 if s.f.pass.debug > regDebug { 360 fmt.Printf("assignReg %s %s/%s\n", &s.registers[r], v, c) 361 } 362 if s.regs[r].v != nil { 363 s.f.Fatalf("tried to assign register %d to %s/%s but it is already used by %s", r, v, c, s.regs[r].v) 364 } 365 366 // Update state. 367 s.regs[r] = regState{v, c} 368 s.values[v.ID].regs |= regMask(1) << r 369 s.used |= regMask(1) << r 370 s.f.setHome(c, &s.registers[r]) 371 } 372 373 // allocReg chooses a register from the set of registers in mask. 374 // If there is no unused register, a Value will be kicked out of 375 // a register to make room. 376 func (s *regAllocState) allocReg(mask regMask, v *Value) register { 377 mask &= s.allocatable 378 mask &^= s.nospill 379 if mask == 0 { 380 s.f.Fatalf("no register available for %s", v) 381 } 382 383 // Pick an unused register if one is available. 384 if mask&^s.used != 0 { 385 return pickReg(mask &^ s.used) 386 } 387 388 // Pick a value to spill. Spill the value with the 389 // farthest-in-the-future use. 390 // TODO: Prefer registers with already spilled Values? 391 // TODO: Modify preference using affinity graph. 392 // TODO: if a single value is in multiple registers, spill one of them 393 // before spilling a value in just a single register. 394 395 // Find a register to spill. We spill the register containing the value 396 // whose next use is as far in the future as possible. 397 // https://en.wikipedia.org/wiki/Page_replacement_algorithm#The_theoretically_optimal_page_replacement_algorithm 398 var r register 399 maxuse := int32(-1) 400 for t := register(0); t < s.numRegs; t++ { 401 if mask>>t&1 == 0 { 402 continue 403 } 404 v := s.regs[t].v 405 if n := s.values[v.ID].uses.dist; n > maxuse { 406 // v's next use is farther in the future than any value 407 // we've seen so far. A new best spill candidate. 408 r = t 409 maxuse = n 410 } 411 } 412 if maxuse == -1 { 413 s.f.Fatalf("couldn't find register to spill") 414 } 415 416 // Try to move it around before kicking out, if there is a free register. 417 // We generate a Copy and record it. It will be deleted if never used. 418 v2 := s.regs[r].v 419 m := s.compatRegs(v2.Type) &^ s.used &^ s.tmpused &^ (regMask(1) << r) 420 if m != 0 && !s.values[v2.ID].rematerializeable && countRegs(s.values[v2.ID].regs) == 1 { 421 r2 := pickReg(m) 422 c := s.curBlock.NewValue1(v2.Pos, OpCopy, v2.Type, s.regs[r].c) 423 s.copies[c] = false 424 if s.f.pass.debug > regDebug { 425 fmt.Printf("copy %s to %s : %s\n", v2, c, &s.registers[r2]) 426 } 427 s.setOrig(c, v2) 428 s.assignReg(r2, v2, c) 429 } 430 s.freeReg(r) 431 return r 432 } 433 434 // makeSpill returns a Value which represents the spilled value of v. 435 // b is the block in which the spill is used. 436 func (s *regAllocState) makeSpill(v *Value, b *Block) *Value { 437 vi := &s.values[v.ID] 438 if vi.spill != nil { 439 // Final block not known - keep track of subtree where restores reside. 440 vi.restoreMin = min32(vi.restoreMin, s.sdom[b.ID].entry) 441 vi.restoreMax = max32(vi.restoreMax, s.sdom[b.ID].exit) 442 return vi.spill 443 } 444 // Make a spill for v. We don't know where we want 445 // to put it yet, so we leave it blockless for now. 446 spill := s.f.newValueNoBlock(OpStoreReg, v.Type, v.Pos) 447 // We also don't know what the spill's arg will be. 448 // Leave it argless for now. 449 s.setOrig(spill, v) 450 vi.spill = spill 451 vi.restoreMin = s.sdom[b.ID].entry 452 vi.restoreMax = s.sdom[b.ID].exit 453 return spill 454 } 455 456 // allocValToReg allocates v to a register selected from regMask and 457 // returns the register copy of v. Any previous user is kicked out and spilled 458 // (if necessary). Load code is added at the current pc. If nospill is set the 459 // allocated register is marked nospill so the assignment cannot be 460 // undone until the caller allows it by clearing nospill. Returns a 461 // *Value which is either v or a copy of v allocated to the chosen register. 462 func (s *regAllocState) allocValToReg(v *Value, mask regMask, nospill bool, pos src.XPos) *Value { 463 vi := &s.values[v.ID] 464 465 // Check if v is already in a requested register. 466 if mask&vi.regs != 0 { 467 r := pickReg(mask & vi.regs) 468 if s.regs[r].v != v || s.regs[r].c == nil { 469 panic("bad register state") 470 } 471 if nospill { 472 s.nospill |= regMask(1) << r 473 } 474 return s.regs[r].c 475 } 476 477 // Allocate a register. 478 r := s.allocReg(mask, v) 479 480 // Allocate v to the new register. 481 var c *Value 482 if vi.regs != 0 { 483 // Copy from a register that v is already in. 484 r2 := pickReg(vi.regs) 485 if s.regs[r2].v != v { 486 panic("bad register state") 487 } 488 c = s.curBlock.NewValue1(pos, OpCopy, v.Type, s.regs[r2].c) 489 } else if v.rematerializeable() { 490 // Rematerialize instead of loading from the spill location. 491 c = v.copyIntoWithXPos(s.curBlock, pos) 492 } else { 493 // Load v from its spill location. 494 spill := s.makeSpill(v, s.curBlock) 495 if s.f.pass.debug > logSpills { 496 s.f.Warnl(vi.spill.Pos, "load spill for %v from %v", v, spill) 497 } 498 c = s.curBlock.NewValue1(pos, OpLoadReg, v.Type, spill) 499 } 500 s.setOrig(c, v) 501 s.assignReg(r, v, c) 502 if nospill { 503 s.nospill |= regMask(1) << r 504 } 505 return c 506 } 507 508 // isLeaf reports whether f performs any calls. 509 func isLeaf(f *Func) bool { 510 for _, b := range f.Blocks { 511 for _, v := range b.Values { 512 if opcodeTable[v.Op].call { 513 return false 514 } 515 } 516 } 517 return true 518 } 519 520 func (s *regAllocState) init(f *Func) { 521 s.f = f 522 s.f.RegAlloc = s.f.Cache.locs[:0] 523 s.registers = f.Config.registers 524 if nr := len(s.registers); nr == 0 || nr > int(noRegister) || nr > int(unsafe.Sizeof(regMask(0))*8) { 525 s.f.Fatalf("bad number of registers: %d", nr) 526 } else { 527 s.numRegs = register(nr) 528 } 529 // Locate SP, SB, and g registers. 530 s.SPReg = noRegister 531 s.SBReg = noRegister 532 s.GReg = noRegister 533 for r := register(0); r < s.numRegs; r++ { 534 switch s.registers[r].String() { 535 case "SP": 536 s.SPReg = r 537 case "SB": 538 s.SBReg = r 539 case "g": 540 s.GReg = r 541 } 542 } 543 // Make sure we found all required registers. 544 switch noRegister { 545 case s.SPReg: 546 s.f.Fatalf("no SP register found") 547 case s.SBReg: 548 s.f.Fatalf("no SB register found") 549 case s.GReg: 550 if f.Config.hasGReg { 551 s.f.Fatalf("no g register found") 552 } 553 } 554 555 // Figure out which registers we're allowed to use. 556 s.allocatable = s.f.Config.gpRegMask | s.f.Config.fpRegMask | s.f.Config.specialRegMask 557 s.allocatable &^= 1 << s.SPReg 558 s.allocatable &^= 1 << s.SBReg 559 if s.f.Config.hasGReg { 560 s.allocatable &^= 1 << s.GReg 561 } 562 if s.f.Config.ctxt.Framepointer_enabled && s.f.Config.FPReg >= 0 { 563 s.allocatable &^= 1 << uint(s.f.Config.FPReg) 564 } 565 if s.f.Config.LinkReg != -1 { 566 if isLeaf(f) { 567 // Leaf functions don't save/restore the link register. 568 s.allocatable &^= 1 << uint(s.f.Config.LinkReg) 569 } 570 if s.f.Config.arch == "arm" && objabi.GOARM == 5 { 571 // On ARMv5 we insert softfloat calls at each FP instruction. 572 // This clobbers LR almost everywhere. Disable allocating LR 573 // on ARMv5. 574 s.allocatable &^= 1 << uint(s.f.Config.LinkReg) 575 } 576 } 577 if s.f.Config.ctxt.Flag_dynlink { 578 switch s.f.Config.arch { 579 case "amd64": 580 s.allocatable &^= 1 << 15 // R15 581 case "arm": 582 s.allocatable &^= 1 << 9 // R9 583 case "ppc64le": // R2 already reserved. 584 // nothing to do 585 case "arm64": 586 // nothing to do? 587 case "386": 588 // nothing to do. 589 // Note that for Flag_shared (position independent code) 590 // we do need to be careful, but that carefulness is hidden 591 // in the rewrite rules so we always have a free register 592 // available for global load/stores. See gen/386.rules (search for Flag_shared). 593 case "s390x": 594 // nothing to do, R10 & R11 already reserved 595 default: 596 s.f.fe.Fatalf(src.NoXPos, "arch %s not implemented", s.f.Config.arch) 597 } 598 } 599 if s.f.Config.nacl { 600 switch s.f.Config.arch { 601 case "arm": 602 s.allocatable &^= 1 << 9 // R9 is "thread pointer" on nacl/arm 603 case "amd64p32": 604 s.allocatable &^= 1 << 5 // BP - reserved for nacl 605 s.allocatable &^= 1 << 15 // R15 - reserved for nacl 606 } 607 } 608 if s.f.Config.use387 { 609 s.allocatable &^= 1 << 15 // X7 disallowed (one 387 register is used as scratch space during SSE->387 generation in ../x86/387.go) 610 } 611 612 s.regs = make([]regState, s.numRegs) 613 s.values = make([]valState, f.NumValues()) 614 s.orig = make([]*Value, f.NumValues()) 615 s.copies = make(map[*Value]bool) 616 if s.f.Config.ctxt.Flag_locationlists { 617 s.valueNames = make([][]LocalSlot, f.NumValues()) 618 for slot, values := range f.NamedValues { 619 if isSynthetic(&slot) { 620 continue 621 } 622 for _, value := range values { 623 s.valueNames[value.ID] = append(s.valueNames[value.ID], slot) 624 } 625 } 626 } 627 for _, b := range f.Blocks { 628 for _, v := range b.Values { 629 if !v.Type.IsMemory() && !v.Type.IsVoid() && !v.Type.IsFlags() && !v.Type.IsTuple() { 630 s.values[v.ID].needReg = true 631 s.values[v.ID].rematerializeable = v.rematerializeable() 632 s.orig[v.ID] = v 633 } 634 // Note: needReg is false for values returning Tuple types. 635 // Instead, we mark the corresponding Selects as needReg. 636 } 637 } 638 s.computeLive() 639 640 // Compute block order. This array allows us to distinguish forward edges 641 // from backward edges and compute how far they go. 642 blockOrder := make([]int32, f.NumBlocks()) 643 for i, b := range f.Blocks { 644 blockOrder[b.ID] = int32(i) 645 } 646 647 // Compute primary predecessors. 648 s.primary = make([]int32, f.NumBlocks()) 649 for _, b := range f.Blocks { 650 best := -1 651 for i, e := range b.Preds { 652 p := e.b 653 if blockOrder[p.ID] >= blockOrder[b.ID] { 654 continue // backward edge 655 } 656 if best == -1 || blockOrder[p.ID] > blockOrder[b.Preds[best].b.ID] { 657 best = i 658 } 659 } 660 s.primary[b.ID] = int32(best) 661 } 662 663 s.endRegs = make([][]endReg, f.NumBlocks()) 664 s.startRegs = make([][]startReg, f.NumBlocks()) 665 s.spillLive = make([][]ID, f.NumBlocks()) 666 s.sdom = f.sdom() 667 } 668 669 // Adds a use record for id at distance dist from the start of the block. 670 // All calls to addUse must happen with nonincreasing dist. 671 func (s *regAllocState) addUse(id ID, dist int32, pos src.XPos) { 672 r := s.freeUseRecords 673 if r != nil { 674 s.freeUseRecords = r.next 675 } else { 676 r = &use{} 677 } 678 r.dist = dist 679 r.pos = pos 680 r.next = s.values[id].uses 681 s.values[id].uses = r 682 if r.next != nil && dist > r.next.dist { 683 s.f.Fatalf("uses added in wrong order") 684 } 685 } 686 687 // advanceUses advances the uses of v's args from the state before v to the state after v. 688 // Any values which have no more uses are deallocated from registers. 689 func (s *regAllocState) advanceUses(v *Value) { 690 for _, a := range v.Args { 691 if !s.values[a.ID].needReg { 692 continue 693 } 694 ai := &s.values[a.ID] 695 r := ai.uses 696 ai.uses = r.next 697 if r.next == nil { 698 // Value is dead, free all registers that hold it. 699 s.freeRegs(ai.regs) 700 } 701 r.next = s.freeUseRecords 702 s.freeUseRecords = r 703 } 704 } 705 706 // liveAfterCurrentInstruction reports whether v is live after 707 // the current instruction is completed. v must be used by the 708 // current instruction. 709 func (s *regAllocState) liveAfterCurrentInstruction(v *Value) bool { 710 u := s.values[v.ID].uses 711 d := u.dist 712 for u != nil && u.dist == d { 713 u = u.next 714 } 715 return u != nil && u.dist > d 716 } 717 718 // Sets the state of the registers to that encoded in regs. 719 func (s *regAllocState) setState(regs []endReg) { 720 for s.used != 0 { 721 s.freeOrResetReg(pickReg(s.used), true) 722 } 723 for _, x := range regs { 724 s.assignReg(x.r, x.v, x.c) 725 } 726 } 727 728 // compatRegs returns the set of registers which can store a type t. 729 func (s *regAllocState) compatRegs(t *types.Type) regMask { 730 var m regMask 731 if t.IsTuple() || t.IsFlags() { 732 return 0 733 } 734 if t.IsFloat() || t == types.TypeInt128 { 735 m = s.f.Config.fpRegMask 736 } else { 737 m = s.f.Config.gpRegMask 738 } 739 return m & s.allocatable 740 } 741 742 func (s *regAllocState) regalloc(f *Func) { 743 regValLiveSet := f.newSparseSet(f.NumValues()) // set of values that may be live in register 744 defer f.retSparseSet(regValLiveSet) 745 var oldSched []*Value 746 var phis []*Value 747 var phiRegs []register 748 var args []*Value 749 750 // Data structure used for computing desired registers. 751 var desired desiredState 752 753 // Desired registers for inputs & outputs for each instruction in the block. 754 type dentry struct { 755 out [4]register // desired output registers 756 in [3][4]register // desired input registers (for inputs 0,1, and 2) 757 } 758 var dinfo []dentry 759 760 if f.Entry != f.Blocks[0] { 761 f.Fatalf("entry block must be first") 762 } 763 764 for _, b := range f.Blocks { 765 if s.f.pass.debug > regDebug { 766 fmt.Printf("Begin processing block %v\n", b) 767 } 768 s.curBlock = b 769 770 // Initialize regValLiveSet and uses fields for this block. 771 // Walk backwards through the block doing liveness analysis. 772 regValLiveSet.clear() 773 for _, e := range s.live[b.ID] { 774 s.addUse(e.ID, int32(len(b.Values))+e.dist, e.pos) // pseudo-uses from beyond end of block 775 regValLiveSet.add(e.ID) 776 } 777 if v := b.Control; v != nil && s.values[v.ID].needReg { 778 s.addUse(v.ID, int32(len(b.Values)), b.Pos) // pseudo-use by control value 779 regValLiveSet.add(v.ID) 780 } 781 for i := len(b.Values) - 1; i >= 0; i-- { 782 v := b.Values[i] 783 regValLiveSet.remove(v.ID) 784 if v.Op == OpPhi { 785 // Remove v from the live set, but don't add 786 // any inputs. This is the state the len(b.Preds)>1 787 // case below desires; it wants to process phis specially. 788 continue 789 } 790 if opcodeTable[v.Op].call { 791 // Function call clobbers all the registers but SP and SB. 792 regValLiveSet.clear() 793 if s.sp != 0 && s.values[s.sp].uses != nil { 794 regValLiveSet.add(s.sp) 795 } 796 if s.sb != 0 && s.values[s.sb].uses != nil { 797 regValLiveSet.add(s.sb) 798 } 799 } 800 for _, a := range v.Args { 801 if !s.values[a.ID].needReg { 802 continue 803 } 804 s.addUse(a.ID, int32(i), v.Pos) 805 regValLiveSet.add(a.ID) 806 } 807 } 808 if s.f.pass.debug > regDebug { 809 fmt.Printf("uses for %s:%s\n", s.f.Name, b) 810 for i := range s.values { 811 vi := &s.values[i] 812 u := vi.uses 813 if u == nil { 814 continue 815 } 816 fmt.Printf(" v%d:", i) 817 for u != nil { 818 fmt.Printf(" %d", u.dist) 819 u = u.next 820 } 821 fmt.Println() 822 } 823 } 824 825 // Make a copy of the block schedule so we can generate a new one in place. 826 // We make a separate copy for phis and regular values. 827 nphi := 0 828 for _, v := range b.Values { 829 if v.Op != OpPhi { 830 break 831 } 832 nphi++ 833 } 834 phis = append(phis[:0], b.Values[:nphi]...) 835 oldSched = append(oldSched[:0], b.Values[nphi:]...) 836 b.Values = b.Values[:0] 837 838 // Initialize start state of block. 839 if b == f.Entry { 840 // Regalloc state is empty to start. 841 if nphi > 0 { 842 f.Fatalf("phis in entry block") 843 } 844 } else if len(b.Preds) == 1 { 845 // Start regalloc state with the end state of the previous block. 846 s.setState(s.endRegs[b.Preds[0].b.ID]) 847 if nphi > 0 { 848 f.Fatalf("phis in single-predecessor block") 849 } 850 // Drop any values which are no longer live. 851 // This may happen because at the end of p, a value may be 852 // live but only used by some other successor of p. 853 for r := register(0); r < s.numRegs; r++ { 854 v := s.regs[r].v 855 if v != nil && !regValLiveSet.contains(v.ID) { 856 s.freeReg(r) 857 } 858 } 859 } else { 860 // This is the complicated case. We have more than one predecessor, 861 // which means we may have Phi ops. 862 863 // Start with the final register state of the primary predecessor 864 idx := s.primary[b.ID] 865 if idx < 0 { 866 f.Fatalf("block with no primary predecessor %s", b) 867 } 868 p := b.Preds[idx].b 869 s.setState(s.endRegs[p.ID]) 870 871 if s.f.pass.debug > regDebug { 872 fmt.Printf("starting merge block %s with end state of %s:\n", b, p) 873 for _, x := range s.endRegs[p.ID] { 874 fmt.Printf(" %s: orig:%s cache:%s\n", &s.registers[x.r], x.v, x.c) 875 } 876 } 877 878 // Decide on registers for phi ops. Use the registers determined 879 // by the primary predecessor if we can. 880 // TODO: pick best of (already processed) predecessors? 881 // Majority vote? Deepest nesting level? 882 phiRegs = phiRegs[:0] 883 var phiUsed regMask 884 for _, v := range phis { 885 if !s.values[v.ID].needReg { 886 phiRegs = append(phiRegs, noRegister) 887 continue 888 } 889 a := v.Args[idx] 890 // Some instructions target not-allocatable registers. 891 // They're not suitable for further (phi-function) allocation. 892 m := s.values[a.ID].regs &^ phiUsed & s.allocatable 893 if m != 0 { 894 r := pickReg(m) 895 phiUsed |= regMask(1) << r 896 phiRegs = append(phiRegs, r) 897 } else { 898 phiRegs = append(phiRegs, noRegister) 899 } 900 } 901 902 // Second pass - deallocate any phi inputs which are now dead. 903 for i, v := range phis { 904 if !s.values[v.ID].needReg { 905 continue 906 } 907 a := v.Args[idx] 908 if !regValLiveSet.contains(a.ID) { 909 // Input is dead beyond the phi, deallocate 910 // anywhere else it might live. 911 s.freeRegs(s.values[a.ID].regs) 912 } else { 913 // Input is still live. 914 // Try to move it around before kicking out, if there is a free register. 915 // We generate a Copy in the predecessor block and record it. It will be 916 // deleted if never used. 917 r := phiRegs[i] 918 if r == noRegister { 919 continue 920 } 921 // Pick a free register. At this point some registers used in the predecessor 922 // block may have been deallocated. Those are the ones used for Phis. Exclude 923 // them (and they are not going to be helpful anyway). 924 m := s.compatRegs(a.Type) &^ s.used &^ phiUsed 925 if m != 0 && !s.values[a.ID].rematerializeable && countRegs(s.values[a.ID].regs) == 1 { 926 r2 := pickReg(m) 927 c := p.NewValue1(a.Pos, OpCopy, a.Type, s.regs[r].c) 928 s.copies[c] = false 929 if s.f.pass.debug > regDebug { 930 fmt.Printf("copy %s to %s : %s\n", a, c, &s.registers[r2]) 931 } 932 s.setOrig(c, a) 933 s.assignReg(r2, a, c) 934 s.endRegs[p.ID] = append(s.endRegs[p.ID], endReg{r2, a, c}) 935 } 936 s.freeReg(r) 937 } 938 } 939 940 // Copy phi ops into new schedule. 941 b.Values = append(b.Values, phis...) 942 943 // Third pass - pick registers for phis whose inputs 944 // were not in a register. 945 for i, v := range phis { 946 if !s.values[v.ID].needReg { 947 continue 948 } 949 if phiRegs[i] != noRegister { 950 continue 951 } 952 if s.f.Config.use387 && v.Type.IsFloat() { 953 continue // 387 can't handle floats in registers between blocks 954 } 955 m := s.compatRegs(v.Type) &^ phiUsed &^ s.used 956 if m != 0 { 957 r := pickReg(m) 958 phiRegs[i] = r 959 phiUsed |= regMask(1) << r 960 } 961 } 962 963 // Set registers for phis. Add phi spill code. 964 for i, v := range phis { 965 if !s.values[v.ID].needReg { 966 continue 967 } 968 r := phiRegs[i] 969 if r == noRegister { 970 // stack-based phi 971 // Spills will be inserted in all the predecessors below. 972 s.values[v.ID].spill = v // v starts life spilled 973 continue 974 } 975 // register-based phi 976 s.assignReg(r, v, v) 977 } 978 979 // Deallocate any values which are no longer live. Phis are excluded. 980 for r := register(0); r < s.numRegs; r++ { 981 if phiUsed>>r&1 != 0 { 982 continue 983 } 984 v := s.regs[r].v 985 if v != nil && !regValLiveSet.contains(v.ID) { 986 s.freeReg(r) 987 } 988 } 989 990 // Save the starting state for use by merge edges. 991 var regList []startReg 992 for r := register(0); r < s.numRegs; r++ { 993 v := s.regs[r].v 994 if v == nil { 995 continue 996 } 997 if phiUsed>>r&1 != 0 { 998 // Skip registers that phis used, we'll handle those 999 // specially during merge edge processing. 1000 continue 1001 } 1002 regList = append(regList, startReg{r, v, s.regs[r].c, s.values[v.ID].uses.pos}) 1003 } 1004 s.startRegs[b.ID] = regList 1005 1006 if s.f.pass.debug > regDebug { 1007 fmt.Printf("after phis\n") 1008 for _, x := range s.startRegs[b.ID] { 1009 fmt.Printf(" %s: v%d\n", &s.registers[x.r], x.v.ID) 1010 } 1011 } 1012 } 1013 1014 // Allocate space to record the desired registers for each value. 1015 dinfo = dinfo[:0] 1016 for i := 0; i < len(oldSched); i++ { 1017 dinfo = append(dinfo, dentry{}) 1018 } 1019 1020 // Load static desired register info at the end of the block. 1021 desired.copy(&s.desired[b.ID]) 1022 1023 // Check actual assigned registers at the start of the next block(s). 1024 // Dynamically assigned registers will trump the static 1025 // desired registers computed during liveness analysis. 1026 // Note that we do this phase after startRegs is set above, so that 1027 // we get the right behavior for a block which branches to itself. 1028 for _, e := range b.Succs { 1029 succ := e.b 1030 // TODO: prioritize likely successor? 1031 for _, x := range s.startRegs[succ.ID] { 1032 desired.add(x.v.ID, x.r) 1033 } 1034 // Process phi ops in succ. 1035 pidx := e.i 1036 for _, v := range succ.Values { 1037 if v.Op != OpPhi { 1038 continue 1039 } 1040 if !s.values[v.ID].needReg { 1041 continue 1042 } 1043 rp, ok := s.f.getHome(v.ID).(*Register) 1044 if !ok { 1045 continue 1046 } 1047 desired.add(v.Args[pidx].ID, register(rp.num)) 1048 } 1049 } 1050 // Walk values backwards computing desired register info. 1051 // See computeLive for more comments. 1052 for i := len(oldSched) - 1; i >= 0; i-- { 1053 v := oldSched[i] 1054 prefs := desired.remove(v.ID) 1055 desired.clobber(opcodeTable[v.Op].reg.clobbers) 1056 for _, j := range opcodeTable[v.Op].reg.inputs { 1057 if countRegs(j.regs) != 1 { 1058 continue 1059 } 1060 desired.clobber(j.regs) 1061 desired.add(v.Args[j.idx].ID, pickReg(j.regs)) 1062 } 1063 if opcodeTable[v.Op].resultInArg0 { 1064 if opcodeTable[v.Op].commutative { 1065 desired.addList(v.Args[1].ID, prefs) 1066 } 1067 desired.addList(v.Args[0].ID, prefs) 1068 } 1069 // Save desired registers for this value. 1070 dinfo[i].out = prefs 1071 for j, a := range v.Args { 1072 if j >= len(dinfo[i].in) { 1073 break 1074 } 1075 dinfo[i].in[j] = desired.get(a.ID) 1076 } 1077 } 1078 1079 // Process all the non-phi values. 1080 for idx, v := range oldSched { 1081 if s.f.pass.debug > regDebug { 1082 fmt.Printf(" processing %s\n", v.LongString()) 1083 } 1084 regspec := opcodeTable[v.Op].reg 1085 if v.Op == OpPhi { 1086 f.Fatalf("phi %s not at start of block", v) 1087 } 1088 if v.Op == OpSP { 1089 s.assignReg(s.SPReg, v, v) 1090 b.Values = append(b.Values, v) 1091 s.advanceUses(v) 1092 s.sp = v.ID 1093 continue 1094 } 1095 if v.Op == OpSB { 1096 s.assignReg(s.SBReg, v, v) 1097 b.Values = append(b.Values, v) 1098 s.advanceUses(v) 1099 s.sb = v.ID 1100 continue 1101 } 1102 if v.Op == OpSelect0 || v.Op == OpSelect1 { 1103 if s.values[v.ID].needReg { 1104 var i = 0 1105 if v.Op == OpSelect1 { 1106 i = 1 1107 } 1108 s.assignReg(register(s.f.getHome(v.Args[0].ID).(LocPair)[i].(*Register).num), v, v) 1109 } 1110 b.Values = append(b.Values, v) 1111 s.advanceUses(v) 1112 goto issueSpill 1113 } 1114 if v.Op == OpGetG && s.f.Config.hasGReg { 1115 // use hardware g register 1116 if s.regs[s.GReg].v != nil { 1117 s.freeReg(s.GReg) // kick out the old value 1118 } 1119 s.assignReg(s.GReg, v, v) 1120 b.Values = append(b.Values, v) 1121 s.advanceUses(v) 1122 goto issueSpill 1123 } 1124 if v.Op == OpArg { 1125 // Args are "pre-spilled" values. We don't allocate 1126 // any register here. We just set up the spill pointer to 1127 // point at itself and any later user will restore it to use it. 1128 s.values[v.ID].spill = v 1129 b.Values = append(b.Values, v) 1130 s.advanceUses(v) 1131 continue 1132 } 1133 if v.Op == OpKeepAlive { 1134 // Make sure the argument to v is still live here. 1135 s.advanceUses(v) 1136 vi := &s.values[v.Args[0].ID] 1137 if vi.spill != nil { 1138 // Use the spill location. 1139 v.SetArg(0, vi.spill) 1140 } else { 1141 // No need to keep unspilled values live. 1142 // These are typically rematerializeable constants like nil, 1143 // or values of a variable that were modified since the last call. 1144 v.Op = OpCopy 1145 v.SetArgs1(v.Args[1]) 1146 } 1147 b.Values = append(b.Values, v) 1148 continue 1149 } 1150 if len(regspec.inputs) == 0 && len(regspec.outputs) == 0 { 1151 // No register allocation required (or none specified yet) 1152 s.freeRegs(regspec.clobbers) 1153 b.Values = append(b.Values, v) 1154 s.advanceUses(v) 1155 continue 1156 } 1157 1158 if s.values[v.ID].rematerializeable { 1159 // Value is rematerializeable, don't issue it here. 1160 // It will get issued just before each use (see 1161 // allocValueToReg). 1162 for _, a := range v.Args { 1163 a.Uses-- 1164 } 1165 s.advanceUses(v) 1166 continue 1167 } 1168 1169 if s.f.pass.debug > regDebug { 1170 fmt.Printf("value %s\n", v.LongString()) 1171 fmt.Printf(" out:") 1172 for _, r := range dinfo[idx].out { 1173 if r != noRegister { 1174 fmt.Printf(" %s", &s.registers[r]) 1175 } 1176 } 1177 fmt.Println() 1178 for i := 0; i < len(v.Args) && i < 3; i++ { 1179 fmt.Printf(" in%d:", i) 1180 for _, r := range dinfo[idx].in[i] { 1181 if r != noRegister { 1182 fmt.Printf(" %s", &s.registers[r]) 1183 } 1184 } 1185 fmt.Println() 1186 } 1187 } 1188 1189 // Move arguments to registers. Process in an ordering defined 1190 // by the register specification (most constrained first). 1191 args = append(args[:0], v.Args...) 1192 for _, i := range regspec.inputs { 1193 mask := i.regs 1194 if mask&s.values[args[i.idx].ID].regs == 0 { 1195 // Need a new register for the input. 1196 mask &= s.allocatable 1197 mask &^= s.nospill 1198 // Used desired register if available. 1199 if i.idx < 3 { 1200 for _, r := range dinfo[idx].in[i.idx] { 1201 if r != noRegister && (mask&^s.used)>>r&1 != 0 { 1202 // Desired register is allowed and unused. 1203 mask = regMask(1) << r 1204 break 1205 } 1206 } 1207 } 1208 // Avoid registers we're saving for other values. 1209 if mask&^desired.avoid != 0 { 1210 mask &^= desired.avoid 1211 } 1212 } 1213 args[i.idx] = s.allocValToReg(args[i.idx], mask, true, v.Pos) 1214 } 1215 1216 // If the output clobbers the input register, make sure we have 1217 // at least two copies of the input register so we don't 1218 // have to reload the value from the spill location. 1219 if opcodeTable[v.Op].resultInArg0 { 1220 var m regMask 1221 if !s.liveAfterCurrentInstruction(v.Args[0]) { 1222 // arg0 is dead. We can clobber its register. 1223 goto ok 1224 } 1225 if s.values[v.Args[0].ID].rematerializeable { 1226 // We can rematerialize the input, don't worry about clobbering it. 1227 goto ok 1228 } 1229 if countRegs(s.values[v.Args[0].ID].regs) >= 2 { 1230 // we have at least 2 copies of arg0. We can afford to clobber one. 1231 goto ok 1232 } 1233 if opcodeTable[v.Op].commutative { 1234 if !s.liveAfterCurrentInstruction(v.Args[1]) { 1235 args[0], args[1] = args[1], args[0] 1236 goto ok 1237 } 1238 if s.values[v.Args[1].ID].rematerializeable { 1239 args[0], args[1] = args[1], args[0] 1240 goto ok 1241 } 1242 if countRegs(s.values[v.Args[1].ID].regs) >= 2 { 1243 args[0], args[1] = args[1], args[0] 1244 goto ok 1245 } 1246 } 1247 1248 // We can't overwrite arg0 (or arg1, if commutative). So we 1249 // need to make a copy of an input so we have a register we can modify. 1250 1251 // Possible new registers to copy into. 1252 m = s.compatRegs(v.Args[0].Type) &^ s.used 1253 if m == 0 { 1254 // No free registers. In this case we'll just clobber 1255 // an input and future uses of that input must use a restore. 1256 // TODO(khr): We should really do this like allocReg does it, 1257 // spilling the value with the most distant next use. 1258 goto ok 1259 } 1260 1261 // Try to move an input to the desired output. 1262 for _, r := range dinfo[idx].out { 1263 if r != noRegister && m>>r&1 != 0 { 1264 m = regMask(1) << r 1265 args[0] = s.allocValToReg(v.Args[0], m, true, v.Pos) 1266 // Note: we update args[0] so the instruction will 1267 // use the register copy we just made. 1268 goto ok 1269 } 1270 } 1271 // Try to copy input to its desired location & use its old 1272 // location as the result register. 1273 for _, r := range dinfo[idx].in[0] { 1274 if r != noRegister && m>>r&1 != 0 { 1275 m = regMask(1) << r 1276 c := s.allocValToReg(v.Args[0], m, true, v.Pos) 1277 s.copies[c] = false 1278 // Note: no update to args[0] so the instruction will 1279 // use the original copy. 1280 goto ok 1281 } 1282 } 1283 if opcodeTable[v.Op].commutative { 1284 for _, r := range dinfo[idx].in[1] { 1285 if r != noRegister && m>>r&1 != 0 { 1286 m = regMask(1) << r 1287 c := s.allocValToReg(v.Args[1], m, true, v.Pos) 1288 s.copies[c] = false 1289 args[0], args[1] = args[1], args[0] 1290 goto ok 1291 } 1292 } 1293 } 1294 // Avoid future fixed uses if we can. 1295 if m&^desired.avoid != 0 { 1296 m &^= desired.avoid 1297 } 1298 // Save input 0 to a new register so we can clobber it. 1299 c := s.allocValToReg(v.Args[0], m, true, v.Pos) 1300 s.copies[c] = false 1301 } 1302 1303 ok: 1304 // Now that all args are in regs, we're ready to issue the value itself. 1305 // Before we pick a register for the output value, allow input registers 1306 // to be deallocated. We do this here so that the output can use the 1307 // same register as a dying input. 1308 if !opcodeTable[v.Op].resultNotInArgs { 1309 s.tmpused = s.nospill 1310 s.nospill = 0 1311 s.advanceUses(v) // frees any registers holding args that are no longer live 1312 } 1313 1314 // Dump any registers which will be clobbered 1315 s.freeRegs(regspec.clobbers) 1316 s.tmpused |= regspec.clobbers 1317 1318 // Pick registers for outputs. 1319 { 1320 outRegs := [2]register{noRegister, noRegister} 1321 var used regMask 1322 for _, out := range regspec.outputs { 1323 mask := out.regs & s.allocatable &^ used 1324 if mask == 0 { 1325 continue 1326 } 1327 if opcodeTable[v.Op].resultInArg0 && out.idx == 0 { 1328 if !opcodeTable[v.Op].commutative { 1329 // Output must use the same register as input 0. 1330 r := register(s.f.getHome(args[0].ID).(*Register).num) 1331 mask = regMask(1) << r 1332 } else { 1333 // Output must use the same register as input 0 or 1. 1334 r0 := register(s.f.getHome(args[0].ID).(*Register).num) 1335 r1 := register(s.f.getHome(args[1].ID).(*Register).num) 1336 // Check r0 and r1 for desired output register. 1337 found := false 1338 for _, r := range dinfo[idx].out { 1339 if (r == r0 || r == r1) && (mask&^s.used)>>r&1 != 0 { 1340 mask = regMask(1) << r 1341 found = true 1342 if r == r1 { 1343 args[0], args[1] = args[1], args[0] 1344 } 1345 break 1346 } 1347 } 1348 if !found { 1349 // Neither are desired, pick r0. 1350 mask = regMask(1) << r0 1351 } 1352 } 1353 } 1354 for _, r := range dinfo[idx].out { 1355 if r != noRegister && (mask&^s.used)>>r&1 != 0 { 1356 // Desired register is allowed and unused. 1357 mask = regMask(1) << r 1358 break 1359 } 1360 } 1361 // Avoid registers we're saving for other values. 1362 if mask&^desired.avoid != 0 { 1363 mask &^= desired.avoid 1364 } 1365 r := s.allocReg(mask, v) 1366 outRegs[out.idx] = r 1367 used |= regMask(1) << r 1368 s.tmpused |= regMask(1) << r 1369 } 1370 // Record register choices 1371 if v.Type.IsTuple() { 1372 var outLocs LocPair 1373 if r := outRegs[0]; r != noRegister { 1374 outLocs[0] = &s.registers[r] 1375 } 1376 if r := outRegs[1]; r != noRegister { 1377 outLocs[1] = &s.registers[r] 1378 } 1379 s.f.setHome(v, outLocs) 1380 // Note that subsequent SelectX instructions will do the assignReg calls. 1381 } else { 1382 if r := outRegs[0]; r != noRegister { 1383 s.assignReg(r, v, v) 1384 } 1385 } 1386 } 1387 1388 // deallocate dead args, if we have not done so 1389 if opcodeTable[v.Op].resultNotInArgs { 1390 s.nospill = 0 1391 s.advanceUses(v) // frees any registers holding args that are no longer live 1392 } 1393 s.tmpused = 0 1394 1395 // Issue the Value itself. 1396 for i, a := range args { 1397 v.SetArg(i, a) // use register version of arguments 1398 } 1399 b.Values = append(b.Values, v) 1400 1401 issueSpill: 1402 } 1403 1404 // Load control value into reg. 1405 if v := b.Control; v != nil && s.values[v.ID].needReg { 1406 if s.f.pass.debug > regDebug { 1407 fmt.Printf(" processing control %s\n", v.LongString()) 1408 } 1409 // We assume that a control input can be passed in any 1410 // type-compatible register. If this turns out not to be true, 1411 // we'll need to introduce a regspec for a block's control value. 1412 b.Control = s.allocValToReg(v, s.compatRegs(v.Type), false, b.Pos) 1413 if b.Control != v { 1414 v.Uses-- 1415 b.Control.Uses++ 1416 } 1417 // Remove this use from the uses list. 1418 vi := &s.values[v.ID] 1419 u := vi.uses 1420 vi.uses = u.next 1421 if u.next == nil { 1422 s.freeRegs(vi.regs) // value is dead 1423 } 1424 u.next = s.freeUseRecords 1425 s.freeUseRecords = u 1426 } 1427 1428 // Spill any values that can't live across basic block boundaries. 1429 if s.f.Config.use387 { 1430 s.freeRegs(s.f.Config.fpRegMask) 1431 } 1432 1433 // If we are approaching a merge point and we are the primary 1434 // predecessor of it, find live values that we use soon after 1435 // the merge point and promote them to registers now. 1436 if len(b.Succs) == 1 { 1437 // For this to be worthwhile, the loop must have no calls in it. 1438 top := b.Succs[0].b 1439 loop := s.loopnest.b2l[top.ID] 1440 if loop == nil || loop.header != top || loop.containsCall { 1441 goto badloop 1442 } 1443 1444 // TODO: sort by distance, pick the closest ones? 1445 for _, live := range s.live[b.ID] { 1446 if live.dist >= unlikelyDistance { 1447 // Don't preload anything live after the loop. 1448 continue 1449 } 1450 vid := live.ID 1451 vi := &s.values[vid] 1452 if vi.regs != 0 { 1453 continue 1454 } 1455 if vi.rematerializeable { 1456 continue 1457 } 1458 v := s.orig[vid] 1459 if s.f.Config.use387 && v.Type.IsFloat() { 1460 continue // 387 can't handle floats in registers between blocks 1461 } 1462 m := s.compatRegs(v.Type) &^ s.used 1463 if m&^desired.avoid != 0 { 1464 m &^= desired.avoid 1465 } 1466 if m != 0 { 1467 s.allocValToReg(v, m, false, b.Pos) 1468 } 1469 } 1470 } 1471 badloop: 1472 ; 1473 1474 // Save end-of-block register state. 1475 // First count how many, this cuts allocations in half. 1476 k := 0 1477 for r := register(0); r < s.numRegs; r++ { 1478 v := s.regs[r].v 1479 if v == nil { 1480 continue 1481 } 1482 k++ 1483 } 1484 regList := make([]endReg, 0, k) 1485 for r := register(0); r < s.numRegs; r++ { 1486 v := s.regs[r].v 1487 if v == nil { 1488 continue 1489 } 1490 regList = append(regList, endReg{r, v, s.regs[r].c}) 1491 } 1492 s.endRegs[b.ID] = regList 1493 1494 if checkEnabled { 1495 regValLiveSet.clear() 1496 for _, x := range s.live[b.ID] { 1497 regValLiveSet.add(x.ID) 1498 } 1499 for r := register(0); r < s.numRegs; r++ { 1500 v := s.regs[r].v 1501 if v == nil { 1502 continue 1503 } 1504 if !regValLiveSet.contains(v.ID) { 1505 s.f.Fatalf("val %s is in reg but not live at end of %s", v, b) 1506 } 1507 } 1508 } 1509 1510 // If a value is live at the end of the block and 1511 // isn't in a register, generate a use for the spill location. 1512 // We need to remember this information so that 1513 // the liveness analysis in stackalloc is correct. 1514 for _, e := range s.live[b.ID] { 1515 vi := &s.values[e.ID] 1516 if vi.regs != 0 { 1517 // in a register, we'll use that source for the merge. 1518 continue 1519 } 1520 if vi.rematerializeable { 1521 // we'll rematerialize during the merge. 1522 continue 1523 } 1524 //fmt.Printf("live-at-end spill for %s at %s\n", s.orig[e.ID], b) 1525 spill := s.makeSpill(s.orig[e.ID], b) 1526 s.spillLive[b.ID] = append(s.spillLive[b.ID], spill.ID) 1527 } 1528 1529 // Clear any final uses. 1530 // All that is left should be the pseudo-uses added for values which 1531 // are live at the end of b. 1532 for _, e := range s.live[b.ID] { 1533 u := s.values[e.ID].uses 1534 if u == nil { 1535 f.Fatalf("live at end, no uses v%d", e.ID) 1536 } 1537 if u.next != nil { 1538 f.Fatalf("live at end, too many uses v%d", e.ID) 1539 } 1540 s.values[e.ID].uses = nil 1541 u.next = s.freeUseRecords 1542 s.freeUseRecords = u 1543 } 1544 } 1545 1546 // Decide where the spills we generated will go. 1547 s.placeSpills() 1548 1549 // Anything that didn't get a register gets a stack location here. 1550 // (StoreReg, stack-based phis, inputs, ...) 1551 stacklive := stackalloc(s.f, s.spillLive) 1552 1553 // Fix up all merge edges. 1554 s.shuffle(stacklive) 1555 1556 // Erase any copies we never used. 1557 // Also, an unused copy might be the only use of another copy, 1558 // so continue erasing until we reach a fixed point. 1559 for { 1560 progress := false 1561 for c, used := range s.copies { 1562 if !used && c.Uses == 0 { 1563 if s.f.pass.debug > regDebug { 1564 fmt.Printf("delete copied value %s\n", c.LongString()) 1565 } 1566 c.RemoveArg(0) 1567 f.freeValue(c) 1568 delete(s.copies, c) 1569 progress = true 1570 } 1571 } 1572 if !progress { 1573 break 1574 } 1575 } 1576 1577 for _, b := range f.Blocks { 1578 i := 0 1579 for _, v := range b.Values { 1580 if v.Op == OpInvalid { 1581 continue 1582 } 1583 b.Values[i] = v 1584 i++ 1585 } 1586 b.Values = b.Values[:i] 1587 } 1588 } 1589 1590 func (s *regAllocState) placeSpills() { 1591 f := s.f 1592 1593 // Precompute some useful info. 1594 phiRegs := make([]regMask, f.NumBlocks()) 1595 for _, b := range f.Blocks { 1596 var m regMask 1597 for _, v := range b.Values { 1598 if v.Op == OpRegKill { 1599 continue 1600 } 1601 if v.Op != OpPhi { 1602 break 1603 } 1604 if r, ok := f.getHome(v.ID).(*Register); ok { 1605 m |= regMask(1) << uint(r.num) 1606 } 1607 } 1608 phiRegs[b.ID] = m 1609 } 1610 1611 // Start maps block IDs to the list of spills 1612 // that go at the start of the block (but after any phis). 1613 start := map[ID][]*Value{} 1614 // After maps value IDs to the list of spills 1615 // that go immediately after that value ID. 1616 after := map[ID][]*Value{} 1617 1618 for i := range s.values { 1619 vi := s.values[i] 1620 spill := vi.spill 1621 if spill == nil { 1622 continue 1623 } 1624 if spill.Block != nil { 1625 // Some spills are already fully set up, 1626 // like OpArgs and stack-based phis. 1627 continue 1628 } 1629 v := s.orig[i] 1630 1631 // Walk down the dominator tree looking for a good place to 1632 // put the spill of v. At the start "best" is the best place 1633 // we have found so far. 1634 // TODO: find a way to make this O(1) without arbitrary cutoffs. 1635 best := v.Block 1636 bestArg := v 1637 var bestDepth int16 1638 if l := s.loopnest.b2l[best.ID]; l != nil { 1639 bestDepth = l.depth 1640 } 1641 b := best 1642 const maxSpillSearch = 100 1643 for i := 0; i < maxSpillSearch; i++ { 1644 // Find the child of b in the dominator tree which 1645 // dominates all restores. 1646 p := b 1647 b = nil 1648 for c := s.sdom.Child(p); c != nil && i < maxSpillSearch; c, i = s.sdom.Sibling(c), i+1 { 1649 if s.sdom[c.ID].entry <= vi.restoreMin && s.sdom[c.ID].exit >= vi.restoreMax { 1650 // c also dominates all restores. Walk down into c. 1651 b = c 1652 break 1653 } 1654 } 1655 if b == nil { 1656 // Ran out of blocks which dominate all restores. 1657 break 1658 } 1659 1660 var depth int16 1661 if l := s.loopnest.b2l[b.ID]; l != nil { 1662 depth = l.depth 1663 } 1664 if depth > bestDepth { 1665 // Don't push the spill into a deeper loop. 1666 continue 1667 } 1668 1669 // If v is in a register at the start of b, we can 1670 // place the spill here (after the phis). 1671 if len(b.Preds) == 1 { 1672 for _, e := range s.endRegs[b.Preds[0].b.ID] { 1673 if e.v == v { 1674 // Found a better spot for the spill. 1675 best = b 1676 bestArg = e.c 1677 bestDepth = depth 1678 break 1679 } 1680 } 1681 } else { 1682 for _, e := range s.startRegs[b.ID] { 1683 if e.v == v { 1684 // Found a better spot for the spill. 1685 best = b 1686 bestArg = e.c 1687 bestDepth = depth 1688 break 1689 } 1690 } 1691 } 1692 } 1693 1694 // Put the spill in the best block we found. 1695 spill.Block = best 1696 spill.AddArg(bestArg) 1697 if best == v.Block && v.Op != OpPhi { 1698 // Place immediately after v. 1699 after[v.ID] = append(after[v.ID], spill) 1700 } else { 1701 // Place at the start of best block. 1702 start[best.ID] = append(start[best.ID], spill) 1703 } 1704 } 1705 1706 // Insert spill instructions into the block schedules. 1707 var oldSched []*Value 1708 for _, b := range f.Blocks { 1709 nphi := 0 1710 for _, v := range b.Values { 1711 if v.Op != OpRegKill && v.Op != OpPhi { 1712 break 1713 } 1714 nphi++ 1715 } 1716 oldSched = append(oldSched[:0], b.Values[nphi:]...) 1717 b.Values = b.Values[:nphi] 1718 b.Values = append(b.Values, start[b.ID]...) 1719 for _, v := range oldSched { 1720 b.Values = append(b.Values, v) 1721 b.Values = append(b.Values, after[v.ID]...) 1722 } 1723 } 1724 } 1725 1726 // shuffle fixes up all the merge edges (those going into blocks of indegree > 1). 1727 func (s *regAllocState) shuffle(stacklive [][]ID) { 1728 var e edgeState 1729 e.s = s 1730 e.cache = map[ID][]*Value{} 1731 e.contents = map[Location]contentRecord{} 1732 if s.f.pass.debug > regDebug { 1733 fmt.Printf("shuffle %s\n", s.f.Name) 1734 fmt.Println(s.f.String()) 1735 } 1736 1737 for _, b := range s.f.Blocks { 1738 if len(b.Preds) <= 1 { 1739 continue 1740 } 1741 e.b = b 1742 for i, edge := range b.Preds { 1743 p := edge.b 1744 e.p = p 1745 e.setup(i, s.endRegs[p.ID], s.startRegs[b.ID], stacklive[p.ID]) 1746 e.process() 1747 } 1748 } 1749 } 1750 1751 type edgeState struct { 1752 s *regAllocState 1753 p, b *Block // edge goes from p->b. 1754 1755 // for each pre-regalloc value, a list of equivalent cached values 1756 cache map[ID][]*Value 1757 cachedVals []ID // (superset of) keys of the above map, for deterministic iteration 1758 1759 // map from location to the value it contains 1760 contents map[Location]contentRecord 1761 1762 // desired destination locations 1763 destinations []dstRecord 1764 extra []dstRecord 1765 1766 usedRegs regMask // registers currently holding something 1767 uniqueRegs regMask // registers holding the only copy of a value 1768 finalRegs regMask // registers holding final target 1769 } 1770 1771 type contentRecord struct { 1772 vid ID // pre-regalloc value 1773 c *Value // cached value 1774 final bool // this is a satisfied destination 1775 pos src.XPos // source position of use of the value 1776 } 1777 1778 type dstRecord struct { 1779 loc Location // register or stack slot 1780 vid ID // pre-regalloc value it should contain 1781 splice **Value // place to store reference to the generating instruction 1782 pos src.XPos // source position of use of this location 1783 } 1784 1785 // setup initializes the edge state for shuffling. 1786 func (e *edgeState) setup(idx int, srcReg []endReg, dstReg []startReg, stacklive []ID) { 1787 if e.s.f.pass.debug > regDebug { 1788 fmt.Printf("edge %s->%s\n", e.p, e.b) 1789 } 1790 1791 // Clear state. 1792 for _, vid := range e.cachedVals { 1793 delete(e.cache, vid) 1794 } 1795 e.cachedVals = e.cachedVals[:0] 1796 for k := range e.contents { 1797 delete(e.contents, k) 1798 } 1799 e.usedRegs = 0 1800 e.uniqueRegs = 0 1801 e.finalRegs = 0 1802 1803 // Live registers can be sources. 1804 for _, x := range srcReg { 1805 e.set(&e.s.registers[x.r], x.v.ID, x.c, false, src.NoXPos) // don't care the position of the source 1806 } 1807 // So can all of the spill locations. 1808 for _, spillID := range stacklive { 1809 v := e.s.orig[spillID] 1810 spill := e.s.values[v.ID].spill 1811 if !e.s.sdom.isAncestorEq(spill.Block, e.p) { 1812 // Spills were placed that only dominate the uses found 1813 // during the first regalloc pass. The edge fixup code 1814 // can't use a spill location if the spill doesn't dominate 1815 // the edge. 1816 // We are guaranteed that if the spill doesn't dominate this edge, 1817 // then the value is available in a register (because we called 1818 // makeSpill for every value not in a register at the start 1819 // of an edge). 1820 continue 1821 } 1822 e.set(e.s.f.getHome(spillID), v.ID, spill, false, src.NoXPos) // don't care the position of the source 1823 } 1824 1825 // Figure out all the destinations we need. 1826 dsts := e.destinations[:0] 1827 for _, x := range dstReg { 1828 dsts = append(dsts, dstRecord{&e.s.registers[x.r], x.v.ID, nil, x.pos}) 1829 } 1830 // Phis need their args to end up in a specific location. 1831 for _, v := range e.b.Values { 1832 if v.Op == OpRegKill { 1833 continue 1834 } 1835 if v.Op != OpPhi { 1836 break 1837 } 1838 loc := e.s.f.getHome(v.ID) 1839 if loc == nil { 1840 continue 1841 } 1842 dsts = append(dsts, dstRecord{loc, v.Args[idx].ID, &v.Args[idx], v.Pos}) 1843 } 1844 e.destinations = dsts 1845 1846 if e.s.f.pass.debug > regDebug { 1847 for _, vid := range e.cachedVals { 1848 a := e.cache[vid] 1849 for _, c := range a { 1850 fmt.Printf("src %s: v%d cache=%s\n", e.s.f.getHome(c.ID), vid, c) 1851 } 1852 } 1853 for _, d := range e.destinations { 1854 fmt.Printf("dst %s: v%d\n", d.loc, d.vid) 1855 } 1856 } 1857 } 1858 1859 // process generates code to move all the values to the right destination locations. 1860 func (e *edgeState) process() { 1861 dsts := e.destinations 1862 1863 // Process the destinations until they are all satisfied. 1864 for len(dsts) > 0 { 1865 i := 0 1866 for _, d := range dsts { 1867 if !e.processDest(d.loc, d.vid, d.splice, d.pos) { 1868 // Failed - save for next iteration. 1869 dsts[i] = d 1870 i++ 1871 } 1872 } 1873 if i < len(dsts) { 1874 // Made some progress. Go around again. 1875 dsts = dsts[:i] 1876 1877 // Append any extras destinations we generated. 1878 dsts = append(dsts, e.extra...) 1879 e.extra = e.extra[:0] 1880 continue 1881 } 1882 1883 // We made no progress. That means that any 1884 // remaining unsatisfied moves are in simple cycles. 1885 // For example, A -> B -> C -> D -> A. 1886 // A ----> B 1887 // ^ | 1888 // | | 1889 // | v 1890 // D <---- C 1891 1892 // To break the cycle, we pick an unused register, say R, 1893 // and put a copy of B there. 1894 // A ----> B 1895 // ^ | 1896 // | | 1897 // | v 1898 // D <---- C <---- R=copyofB 1899 // When we resume the outer loop, the A->B move can now proceed, 1900 // and eventually the whole cycle completes. 1901 1902 // Copy any cycle location to a temp register. This duplicates 1903 // one of the cycle entries, allowing the just duplicated value 1904 // to be overwritten and the cycle to proceed. 1905 d := dsts[0] 1906 loc := d.loc 1907 vid := e.contents[loc].vid 1908 c := e.contents[loc].c 1909 r := e.findRegFor(c.Type) 1910 if e.s.f.pass.debug > regDebug { 1911 fmt.Printf("breaking cycle with v%d in %s:%s\n", vid, loc, c) 1912 } 1913 e.erase(r) 1914 if _, isReg := loc.(*Register); isReg { 1915 c = e.p.NewValue1(d.pos, OpCopy, c.Type, c) 1916 } else { 1917 c = e.p.NewValue1(d.pos, OpLoadReg, c.Type, c) 1918 } 1919 e.set(r, vid, c, false, d.pos) 1920 } 1921 } 1922 1923 // processDest generates code to put value vid into location loc. Returns true 1924 // if progress was made. 1925 func (e *edgeState) processDest(loc Location, vid ID, splice **Value, pos src.XPos) bool { 1926 occupant := e.contents[loc] 1927 if occupant.vid == vid { 1928 // Value is already in the correct place. 1929 e.contents[loc] = contentRecord{vid, occupant.c, true, pos} 1930 if splice != nil { 1931 (*splice).Uses-- 1932 *splice = occupant.c 1933 occupant.c.Uses++ 1934 } 1935 // Note: if splice==nil then c will appear dead. This is 1936 // non-SSA formed code, so be careful after this pass not to run 1937 // deadcode elimination. 1938 if _, ok := e.s.copies[occupant.c]; ok { 1939 // The copy at occupant.c was used to avoid spill. 1940 e.s.copies[occupant.c] = true 1941 } 1942 return true 1943 } 1944 1945 // Check if we're allowed to clobber the destination location. 1946 if len(e.cache[occupant.vid]) == 1 && !e.s.values[occupant.vid].rematerializeable { 1947 // We can't overwrite the last copy 1948 // of a value that needs to survive. 1949 return false 1950 } 1951 1952 // Copy from a source of v, register preferred. 1953 v := e.s.orig[vid] 1954 var c *Value 1955 var src Location 1956 if e.s.f.pass.debug > regDebug { 1957 fmt.Printf("moving v%d to %s\n", vid, loc) 1958 fmt.Printf("sources of v%d:", vid) 1959 } 1960 for _, w := range e.cache[vid] { 1961 h := e.s.f.getHome(w.ID) 1962 if e.s.f.pass.debug > regDebug { 1963 fmt.Printf(" %s:%s", h, w) 1964 } 1965 _, isreg := h.(*Register) 1966 if src == nil || isreg { 1967 c = w 1968 src = h 1969 } 1970 } 1971 if e.s.f.pass.debug > regDebug { 1972 if src != nil { 1973 fmt.Printf(" [use %s]\n", src) 1974 } else { 1975 fmt.Printf(" [no source]\n") 1976 } 1977 } 1978 _, dstReg := loc.(*Register) 1979 1980 // Pre-clobber destination. This avoids the 1981 // following situation: 1982 // - v is currently held in R0 and stacktmp0. 1983 // - We want to copy stacktmp1 to stacktmp0. 1984 // - We choose R0 as the temporary register. 1985 // During the copy, both R0 and stacktmp0 are 1986 // clobbered, losing both copies of v. Oops! 1987 // Erasing the destination early means R0 will not 1988 // be chosen as the temp register, as it will then 1989 // be the last copy of v. 1990 e.erase(loc) 1991 var x *Value 1992 if c == nil { 1993 if !e.s.values[vid].rematerializeable { 1994 e.s.f.Fatalf("can't find source for %s->%s: %s\n", e.p, e.b, v.LongString()) 1995 } 1996 if dstReg { 1997 x = v.copyIntoNoXPos(e.p) 1998 } else { 1999 // Rematerialize into stack slot. Need a free 2000 // register to accomplish this. 2001 r := e.findRegFor(v.Type) 2002 e.erase(r) 2003 x = v.copyIntoWithXPos(e.p, pos) 2004 e.set(r, vid, x, false, pos) 2005 // Make sure we spill with the size of the slot, not the 2006 // size of x (which might be wider due to our dropping 2007 // of narrowing conversions). 2008 x = e.p.NewValue1(pos, OpStoreReg, loc.(LocalSlot).Type, x) 2009 } 2010 } else { 2011 // Emit move from src to dst. 2012 _, srcReg := src.(*Register) 2013 if srcReg { 2014 if dstReg { 2015 x = e.p.NewValue1(pos, OpCopy, c.Type, c) 2016 } else { 2017 x = e.p.NewValue1(pos, OpStoreReg, loc.(LocalSlot).Type, c) 2018 } 2019 } else { 2020 if dstReg { 2021 x = e.p.NewValue1(pos, OpLoadReg, c.Type, c) 2022 } else { 2023 // mem->mem. Use temp register. 2024 r := e.findRegFor(c.Type) 2025 e.erase(r) 2026 t := e.p.NewValue1(pos, OpLoadReg, c.Type, c) 2027 e.set(r, vid, t, false, pos) 2028 x = e.p.NewValue1(pos, OpStoreReg, loc.(LocalSlot).Type, t) 2029 } 2030 } 2031 } 2032 e.set(loc, vid, x, true, pos) 2033 if splice != nil { 2034 (*splice).Uses-- 2035 *splice = x 2036 x.Uses++ 2037 } 2038 return true 2039 } 2040 2041 // set changes the contents of location loc to hold the given value and its cached representative. 2042 func (e *edgeState) set(loc Location, vid ID, c *Value, final bool, pos src.XPos) { 2043 e.s.f.setHome(c, loc) 2044 e.contents[loc] = contentRecord{vid, c, final, pos} 2045 a := e.cache[vid] 2046 if len(a) == 0 { 2047 e.cachedVals = append(e.cachedVals, vid) 2048 } 2049 a = append(a, c) 2050 e.cache[vid] = a 2051 if r, ok := loc.(*Register); ok { 2052 e.usedRegs |= regMask(1) << uint(r.num) 2053 if final { 2054 e.finalRegs |= regMask(1) << uint(r.num) 2055 } 2056 if len(a) == 1 { 2057 e.uniqueRegs |= regMask(1) << uint(r.num) 2058 } 2059 if len(a) == 2 { 2060 if t, ok := e.s.f.getHome(a[0].ID).(*Register); ok { 2061 e.uniqueRegs &^= regMask(1) << uint(t.num) 2062 } 2063 } 2064 } 2065 if e.s.f.pass.debug > regDebug { 2066 fmt.Printf("%s\n", c.LongString()) 2067 fmt.Printf("v%d now available in %s:%s\n", vid, loc, c) 2068 } 2069 } 2070 2071 // erase removes any user of loc. 2072 func (e *edgeState) erase(loc Location) { 2073 cr := e.contents[loc] 2074 if cr.c == nil { 2075 return 2076 } 2077 vid := cr.vid 2078 2079 if cr.final { 2080 // Add a destination to move this value back into place. 2081 // Make sure it gets added to the tail of the destination queue 2082 // so we make progress on other moves first. 2083 e.extra = append(e.extra, dstRecord{loc, cr.vid, nil, cr.pos}) 2084 } 2085 2086 // Remove c from the list of cached values. 2087 a := e.cache[vid] 2088 for i, c := range a { 2089 if e.s.f.getHome(c.ID) == loc { 2090 if e.s.f.pass.debug > regDebug { 2091 fmt.Printf("v%d no longer available in %s:%s\n", vid, loc, c) 2092 } 2093 a[i], a = a[len(a)-1], a[:len(a)-1] 2094 if e.s.f.Config.ctxt.Flag_locationlists { 2095 if _, isReg := loc.(*Register); isReg && int(c.ID) < len(e.s.valueNames) && len(e.s.valueNames[c.ID]) != 0 { 2096 kill := e.p.NewValue0(src.NoXPos, OpRegKill, types.TypeVoid) 2097 e.s.f.setHome(kill, loc) 2098 for _, name := range e.s.valueNames[c.ID] { 2099 e.s.f.NamedValues[name] = append(e.s.f.NamedValues[name], kill) 2100 } 2101 } 2102 } 2103 2104 break 2105 } 2106 } 2107 e.cache[vid] = a 2108 2109 // Update register masks. 2110 if r, ok := loc.(*Register); ok { 2111 e.usedRegs &^= regMask(1) << uint(r.num) 2112 if cr.final { 2113 e.finalRegs &^= regMask(1) << uint(r.num) 2114 } 2115 } 2116 if len(a) == 1 { 2117 if r, ok := e.s.f.getHome(a[0].ID).(*Register); ok { 2118 e.uniqueRegs |= regMask(1) << uint(r.num) 2119 } 2120 } 2121 } 2122 2123 // findRegFor finds a register we can use to make a temp copy of type typ. 2124 func (e *edgeState) findRegFor(typ *types.Type) Location { 2125 // Which registers are possibilities. 2126 var m regMask 2127 types := &e.s.f.Config.Types 2128 if typ.IsFloat() { 2129 m = e.s.compatRegs(types.Float64) 2130 } else { 2131 m = e.s.compatRegs(types.Int64) 2132 } 2133 2134 // Pick a register. In priority order: 2135 // 1) an unused register 2136 // 2) a non-unique register not holding a final value 2137 // 3) a non-unique register 2138 // 4) TODO: a register holding a rematerializeable value 2139 x := m &^ e.usedRegs 2140 if x != 0 { 2141 return &e.s.registers[pickReg(x)] 2142 } 2143 x = m &^ e.uniqueRegs &^ e.finalRegs 2144 if x != 0 { 2145 return &e.s.registers[pickReg(x)] 2146 } 2147 x = m &^ e.uniqueRegs 2148 if x != 0 { 2149 return &e.s.registers[pickReg(x)] 2150 } 2151 2152 // No register is available. 2153 // Pick a register to spill. 2154 for _, vid := range e.cachedVals { 2155 a := e.cache[vid] 2156 for _, c := range a { 2157 if r, ok := e.s.f.getHome(c.ID).(*Register); ok && m>>uint(r.num)&1 != 0 { 2158 if !c.rematerializeable() { 2159 x := e.p.NewValue1(c.Pos, OpStoreReg, c.Type, c) 2160 // Allocate a temp location to spill a register to. 2161 // The type of the slot is immaterial - it will not be live across 2162 // any safepoint. Just use a type big enough to hold any register. 2163 t := LocalSlot{N: e.s.f.fe.Auto(c.Pos, types.Int64), Type: types.Int64} 2164 // TODO: reuse these slots. They'll need to be erased first. 2165 e.set(t, vid, x, false, c.Pos) 2166 if e.s.f.pass.debug > regDebug { 2167 fmt.Printf(" SPILL %s->%s %s\n", r, t, x.LongString()) 2168 } 2169 } 2170 // r will now be overwritten by the caller. At some point 2171 // later, the newly saved value will be moved back to its 2172 // final destination in processDest. 2173 return r 2174 } 2175 } 2176 } 2177 2178 fmt.Printf("m:%d unique:%d final:%d\n", m, e.uniqueRegs, e.finalRegs) 2179 for _, vid := range e.cachedVals { 2180 a := e.cache[vid] 2181 for _, c := range a { 2182 fmt.Printf("v%d: %s %s\n", vid, c, e.s.f.getHome(c.ID)) 2183 } 2184 } 2185 e.s.f.Fatalf("can't find empty register on edge %s->%s", e.p, e.b) 2186 return nil 2187 } 2188 2189 // rematerializeable reports whether the register allocator should recompute 2190 // a value instead of spilling/restoring it. 2191 func (v *Value) rematerializeable() bool { 2192 if !opcodeTable[v.Op].rematerializeable { 2193 return false 2194 } 2195 for _, a := range v.Args { 2196 // SP and SB (generated by OpSP and OpSB) are always available. 2197 if a.Op != OpSP && a.Op != OpSB { 2198 return false 2199 } 2200 } 2201 return true 2202 } 2203 2204 type liveInfo struct { 2205 ID ID // ID of value 2206 dist int32 // # of instructions before next use 2207 pos src.XPos // source position of next use 2208 } 2209 2210 // computeLive computes a map from block ID to a list of value IDs live at the end 2211 // of that block. Together with the value ID is a count of how many instructions 2212 // to the next use of that value. The resulting map is stored in s.live. 2213 // computeLive also computes the desired register information at the end of each block. 2214 // This desired register information is stored in s.desired. 2215 // TODO: this could be quadratic if lots of variables are live across lots of 2216 // basic blocks. Figure out a way to make this function (or, more precisely, the user 2217 // of this function) require only linear size & time. 2218 func (s *regAllocState) computeLive() { 2219 f := s.f 2220 s.live = make([][]liveInfo, f.NumBlocks()) 2221 s.desired = make([]desiredState, f.NumBlocks()) 2222 var phis []*Value 2223 2224 live := newSparseMap(f.NumValues()) 2225 t := newSparseMap(f.NumValues()) 2226 2227 // Keep track of which value we want in each register. 2228 var desired desiredState 2229 2230 // Instead of iterating over f.Blocks, iterate over their postordering. 2231 // Liveness information flows backward, so starting at the end 2232 // increases the probability that we will stabilize quickly. 2233 // TODO: Do a better job yet. Here's one possibility: 2234 // Calculate the dominator tree and locate all strongly connected components. 2235 // If a value is live in one block of an SCC, it is live in all. 2236 // Walk the dominator tree from end to beginning, just once, treating SCC 2237 // components as single blocks, duplicated calculated liveness information 2238 // out to all of them. 2239 po := f.postorder() 2240 s.loopnest = f.loopnest() 2241 s.loopnest.calculateDepths() 2242 for { 2243 changed := false 2244 2245 for _, b := range po { 2246 // Start with known live values at the end of the block. 2247 // Add len(b.Values) to adjust from end-of-block distance 2248 // to beginning-of-block distance. 2249 live.clear() 2250 for _, e := range s.live[b.ID] { 2251 live.set(e.ID, e.dist+int32(len(b.Values)), e.pos) 2252 } 2253 2254 // Mark control value as live 2255 if b.Control != nil && s.values[b.Control.ID].needReg { 2256 live.set(b.Control.ID, int32(len(b.Values)), b.Pos) 2257 } 2258 2259 // Propagate backwards to the start of the block 2260 // Assumes Values have been scheduled. 2261 phis = phis[:0] 2262 for i := len(b.Values) - 1; i >= 0; i-- { 2263 v := b.Values[i] 2264 live.remove(v.ID) 2265 if v.Op == OpPhi { 2266 // save phi ops for later 2267 phis = append(phis, v) 2268 continue 2269 } 2270 if opcodeTable[v.Op].call { 2271 c := live.contents() 2272 for i := range c { 2273 c[i].val += unlikelyDistance 2274 } 2275 } 2276 for _, a := range v.Args { 2277 if s.values[a.ID].needReg { 2278 live.set(a.ID, int32(i), v.Pos) 2279 } 2280 } 2281 } 2282 // Propagate desired registers backwards. 2283 desired.copy(&s.desired[b.ID]) 2284 for i := len(b.Values) - 1; i >= 0; i-- { 2285 v := b.Values[i] 2286 prefs := desired.remove(v.ID) 2287 if v.Op == OpPhi { 2288 // TODO: if v is a phi, save desired register for phi inputs. 2289 // For now, we just drop it and don't propagate 2290 // desired registers back though phi nodes. 2291 continue 2292 } 2293 // Cancel desired registers if they get clobbered. 2294 desired.clobber(opcodeTable[v.Op].reg.clobbers) 2295 // Update desired registers if there are any fixed register inputs. 2296 for _, j := range opcodeTable[v.Op].reg.inputs { 2297 if countRegs(j.regs) != 1 { 2298 continue 2299 } 2300 desired.clobber(j.regs) 2301 desired.add(v.Args[j.idx].ID, pickReg(j.regs)) 2302 } 2303 // Set desired register of input 0 if this is a 2-operand instruction. 2304 if opcodeTable[v.Op].resultInArg0 { 2305 if opcodeTable[v.Op].commutative { 2306 desired.addList(v.Args[1].ID, prefs) 2307 } 2308 desired.addList(v.Args[0].ID, prefs) 2309 } 2310 } 2311 2312 // For each predecessor of b, expand its list of live-at-end values. 2313 // invariant: live contains the values live at the start of b (excluding phi inputs) 2314 for i, e := range b.Preds { 2315 p := e.b 2316 // Compute additional distance for the edge. 2317 // Note: delta must be at least 1 to distinguish the control 2318 // value use from the first user in a successor block. 2319 delta := int32(normalDistance) 2320 if len(p.Succs) == 2 { 2321 if p.Succs[0].b == b && p.Likely == BranchLikely || 2322 p.Succs[1].b == b && p.Likely == BranchUnlikely { 2323 delta = likelyDistance 2324 } 2325 if p.Succs[0].b == b && p.Likely == BranchUnlikely || 2326 p.Succs[1].b == b && p.Likely == BranchLikely { 2327 delta = unlikelyDistance 2328 } 2329 } 2330 2331 // Update any desired registers at the end of p. 2332 s.desired[p.ID].merge(&desired) 2333 2334 // Start t off with the previously known live values at the end of p. 2335 t.clear() 2336 for _, e := range s.live[p.ID] { 2337 t.set(e.ID, e.dist, e.pos) 2338 } 2339 update := false 2340 2341 // Add new live values from scanning this block. 2342 for _, e := range live.contents() { 2343 d := e.val + delta 2344 if !t.contains(e.key) || d < t.get(e.key) { 2345 update = true 2346 t.set(e.key, d, e.aux) 2347 } 2348 } 2349 // Also add the correct arg from the saved phi values. 2350 // All phis are at distance delta (we consider them 2351 // simultaneously happening at the start of the block). 2352 for _, v := range phis { 2353 id := v.Args[i].ID 2354 if s.values[id].needReg && (!t.contains(id) || delta < t.get(id)) { 2355 update = true 2356 t.set(id, delta, v.Pos) 2357 } 2358 } 2359 2360 if !update { 2361 continue 2362 } 2363 // The live set has changed, update it. 2364 l := s.live[p.ID][:0] 2365 if cap(l) < t.size() { 2366 l = make([]liveInfo, 0, t.size()) 2367 } 2368 for _, e := range t.contents() { 2369 l = append(l, liveInfo{e.key, e.val, e.aux}) 2370 } 2371 s.live[p.ID] = l 2372 changed = true 2373 } 2374 } 2375 2376 if !changed { 2377 break 2378 } 2379 } 2380 if f.pass.debug > regDebug { 2381 fmt.Println("live values at end of each block") 2382 for _, b := range f.Blocks { 2383 fmt.Printf(" %s:", b) 2384 for _, x := range s.live[b.ID] { 2385 fmt.Printf(" v%d", x.ID) 2386 for _, e := range s.desired[b.ID].entries { 2387 if e.ID != x.ID { 2388 continue 2389 } 2390 fmt.Printf("[") 2391 first := true 2392 for _, r := range e.regs { 2393 if r == noRegister { 2394 continue 2395 } 2396 if !first { 2397 fmt.Printf(",") 2398 } 2399 fmt.Print(&s.registers[r]) 2400 first = false 2401 } 2402 fmt.Printf("]") 2403 } 2404 } 2405 fmt.Printf(" avoid=%x", int64(s.desired[b.ID].avoid)) 2406 fmt.Println() 2407 } 2408 } 2409 } 2410 2411 // A desiredState represents desired register assignments. 2412 type desiredState struct { 2413 // Desired assignments will be small, so we just use a list 2414 // of valueID+registers entries. 2415 entries []desiredStateEntry 2416 // Registers that other values want to be in. This value will 2417 // contain at least the union of the regs fields of entries, but 2418 // may contain additional entries for values that were once in 2419 // this data structure but are no longer. 2420 avoid regMask 2421 } 2422 type desiredStateEntry struct { 2423 // (pre-regalloc) value 2424 ID ID 2425 // Registers it would like to be in, in priority order. 2426 // Unused slots are filled with noRegister. 2427 regs [4]register 2428 } 2429 2430 func (d *desiredState) clear() { 2431 d.entries = d.entries[:0] 2432 d.avoid = 0 2433 } 2434 2435 // get returns a list of desired registers for value vid. 2436 func (d *desiredState) get(vid ID) [4]register { 2437 for _, e := range d.entries { 2438 if e.ID == vid { 2439 return e.regs 2440 } 2441 } 2442 return [4]register{noRegister, noRegister, noRegister, noRegister} 2443 } 2444 2445 // add records that we'd like value vid to be in register r. 2446 func (d *desiredState) add(vid ID, r register) { 2447 d.avoid |= regMask(1) << r 2448 for i := range d.entries { 2449 e := &d.entries[i] 2450 if e.ID != vid { 2451 continue 2452 } 2453 if e.regs[0] == r { 2454 // Already known and highest priority 2455 return 2456 } 2457 for j := 1; j < len(e.regs); j++ { 2458 if e.regs[j] == r { 2459 // Move from lower priority to top priority 2460 copy(e.regs[1:], e.regs[:j]) 2461 e.regs[0] = r 2462 return 2463 } 2464 } 2465 copy(e.regs[1:], e.regs[:]) 2466 e.regs[0] = r 2467 return 2468 } 2469 d.entries = append(d.entries, desiredStateEntry{vid, [4]register{r, noRegister, noRegister, noRegister}}) 2470 } 2471 2472 func (d *desiredState) addList(vid ID, regs [4]register) { 2473 // regs is in priority order, so iterate in reverse order. 2474 for i := len(regs) - 1; i >= 0; i-- { 2475 r := regs[i] 2476 if r != noRegister { 2477 d.add(vid, r) 2478 } 2479 } 2480 } 2481 2482 // clobber erases any desired registers in the set m. 2483 func (d *desiredState) clobber(m regMask) { 2484 for i := 0; i < len(d.entries); { 2485 e := &d.entries[i] 2486 j := 0 2487 for _, r := range e.regs { 2488 if r != noRegister && m>>r&1 == 0 { 2489 e.regs[j] = r 2490 j++ 2491 } 2492 } 2493 if j == 0 { 2494 // No more desired registers for this value. 2495 d.entries[i] = d.entries[len(d.entries)-1] 2496 d.entries = d.entries[:len(d.entries)-1] 2497 continue 2498 } 2499 for ; j < len(e.regs); j++ { 2500 e.regs[j] = noRegister 2501 } 2502 i++ 2503 } 2504 d.avoid &^= m 2505 } 2506 2507 // copy copies a desired state from another desiredState x. 2508 func (d *desiredState) copy(x *desiredState) { 2509 d.entries = append(d.entries[:0], x.entries...) 2510 d.avoid = x.avoid 2511 } 2512 2513 // remove removes the desired registers for vid and returns them. 2514 func (d *desiredState) remove(vid ID) [4]register { 2515 for i := range d.entries { 2516 if d.entries[i].ID == vid { 2517 regs := d.entries[i].regs 2518 d.entries[i] = d.entries[len(d.entries)-1] 2519 d.entries = d.entries[:len(d.entries)-1] 2520 return regs 2521 } 2522 } 2523 return [4]register{noRegister, noRegister, noRegister, noRegister} 2524 } 2525 2526 // merge merges another desired state x into d. 2527 func (d *desiredState) merge(x *desiredState) { 2528 d.avoid |= x.avoid 2529 // There should only be a few desired registers, so 2530 // linear insert is ok. 2531 for _, e := range x.entries { 2532 d.addList(e.ID, e.regs) 2533 } 2534 } 2535 2536 func min32(x, y int32) int32 { 2537 if x < y { 2538 return x 2539 } 2540 return y 2541 } 2542 func max32(x, y int32) int32 { 2543 if x > y { 2544 return x 2545 } 2546 return y 2547 }