PageRenderTime 26ms CodeModel.GetById 11ms RepoModel.GetById 0ms app.codeStats 0ms

/rpython/jit/backend/zarch/registers.py

https://bitbucket.org/timfel/pypy
Python | 54 lines | 39 code | 10 blank | 5 comment | 5 complexity | a6594291d7362e48dc6b30d629a9ad7e MD5 | raw file
Possible License(s): Apache-2.0, AGPL-3.0, BSD-3-Clause 
    

  1. from rpython.jit.backend.zarch.locations import (FloatRegisterLocation,
    2. 

  2.         RegisterLocation, VectorRegisterLocation)
    3. 

  3. 

  4. registers = [RegisterLocation(i) for i in range(16)]
    5. 

  5. fpregisters = [FloatRegisterLocation(i) for i in range(16)]
    6. 

  6. vregisters = [VectorRegisterLocation(16+i) for i in range(16)]
    7. 

  7. 

  8. [r0,r1,r2,r3,r4,r5,r6,r7,r8,
    9. 

  9.  r9,r10,r11,r12,r13,r14,r15] = registers
    10. 

  10. 

  11. MANAGED_REGS = [r2,r3,r4,r5,r6,r7,r8,r9,r10,r11] # keep this list sorted (asc)!
    12. 

  12. MANAGED_REG_PAIRS = [(r2,r3), (r4,r5), (r6,r7), (r8,r9), (r10,r11)]
    13. 

  13. VOLATILES = [r2,r3,r4,r5,r6]
    14. 

  14. SP = r15
    15. 

  15. RETURN = r14
    16. 

  16. POOL = r13
    17. 

  17. SPP = r12
    18. 

  18. SCRATCH = r1
    19. 

  19. SCRATCH2 = r0
    20. 

  20. GPR_RETURN = r2
    21. 

  21. RES = r2
    22. 

  22. RSZ = r11 # do not use a volatile register
    23. 

  23. 

  24. [f0,f1,f2,f3,f4,f5,f6,f7,f8,
    25. 

  25.  f9,f10,f11,f12,f13,f14,f15] = fpregisters
    26. 

  26. 

  27. # there are 4 float returns, but we only care for the first!
    28. 

  28. FPR_RETURN = f0
    29. 

  29. FP_SCRATCH = f15
    30. 

  30. 

  31. MANAGED_FP_REGS = fpregisters[:-1]
    32. 

  32. VOLATILES_FLOAT = [f0,f1,f2,f3,f4,f5,f6,f7]
    33. 

  33. 

  34. MANAGED_VECTOR_REGS = vregisters
    35. 

  35. 

  36. # The JITFRAME_FIXED_SIZE is measured in words, and should be the
    37. 

  37. # number of registers that need to be saved into the jitframe when
    38. 

  38. # failing a guard, for example.
    39. 

  39. ALL_REG_INDEXES = {}
    40. 

  40. for _r in MANAGED_REGS:
    41. 

  41.     ALL_REG_INDEXES[_r] = len(ALL_REG_INDEXES)
    42. 

  42. for _r in MANAGED_FP_REGS:
    43. 

  43.     ALL_REG_INDEXES[_r] = len(ALL_REG_INDEXES)
    44. 

  44. # NOT used, but keeps JITFRAME_FIXED_SIZE even
    45. 

  45. ALL_REG_INDEXES[f15] = len(ALL_REG_INDEXES)
    46. 

  46. JITFRAME_FIXED_SIZE = len(ALL_REG_INDEXES)
    47. 

  47. 

  48. def odd_reg(r):
    49. 

  49.     assert r.value % 2 == 0
    50. 

  50.     return registers[r.value+1]
    51. 

  51. 

  52. def even_reg(r):
    53. 

  53.     assert r.value % 2 == 1
    54. 

  54.     return registers[r.value-1]
    55. 

  55.