PageRenderTime 76ms CodeModel.GetById 31ms RepoModel.GetById 1ms app.codeStats 0ms

/pypy/jit/metainterp/pyjitpl.py

https://bitbucket.org/nbtaylor/pypy
Python | 2768 lines | 2564 code | 94 blank | 110 comment | 86 complexity | 182a7b68d48228744dfa87dd71e2e069 MD5 | raw file

Large files files are truncated, but you can click here to view the full file

  1. import py, sys
  2. from pypy.rpython.lltypesystem import lltype, rclass
  3. from pypy.rlib.objectmodel import we_are_translated
  4. from pypy.rlib.unroll import unrolling_iterable
  5. from pypy.rlib.debug import debug_start, debug_stop, debug_print
  6. from pypy.rlib.debug import make_sure_not_resized
  7. from pypy.rlib import nonconst, rstack
  8. from pypy.jit.metainterp import history, compile, resume
  9. from pypy.jit.metainterp.history import Const, ConstInt, ConstPtr, ConstFloat
  10. from pypy.jit.metainterp.history import Box, TargetToken
  11. from pypy.jit.metainterp.resoperation import rop
  12. from pypy.jit.metainterp import executor
  13. from pypy.jit.metainterp.logger import Logger
  14. from pypy.jit.metainterp.jitprof import EmptyProfiler
  15. from pypy.rlib.jit import Counters
  16. from pypy.jit.metainterp.jitexc import JitException, get_llexception
  17. from pypy.jit.metainterp.heapcache import HeapCache
  18. from pypy.rlib.objectmodel import specialize
  19. from pypy.jit.codewriter.jitcode import JitCode, SwitchDictDescr
  20. from pypy.jit.codewriter import heaptracker
  21. from pypy.jit.metainterp.optimizeopt.util import args_dict_box
  22. # ____________________________________________________________
  23. def arguments(*args):
  24. def decorate(func):
  25. func.argtypes = args
  26. return func
  27. return decorate
  28. # ____________________________________________________________
  29. class MIFrame(object):
  30. debug = False
  31. def __init__(self, metainterp):
  32. self.metainterp = metainterp
  33. self.registers_i = [None] * 256
  34. self.registers_r = [None] * 256
  35. self.registers_f = [None] * 256
  36. def setup(self, jitcode, greenkey=None):
  37. assert isinstance(jitcode, JitCode)
  38. self.jitcode = jitcode
  39. self.bytecode = jitcode.code
  40. # this is not None for frames that are recursive portal calls
  41. self.greenkey = greenkey
  42. # copy the constants in place
  43. self.copy_constants(self.registers_i, jitcode.constants_i, ConstInt)
  44. self.copy_constants(self.registers_r, jitcode.constants_r, ConstPtr)
  45. self.copy_constants(self.registers_f, jitcode.constants_f, ConstFloat)
  46. self._result_argcode = 'v'
  47. # for resume.py operation
  48. self.parent_resumedata_snapshot = None
  49. self.parent_resumedata_frame_info_list = None
  50. # counter for unrolling inlined loops
  51. self.unroll_iterations = 1
  52. @specialize.arg(3)
  53. def copy_constants(self, registers, constants, ConstClass):
  54. """Copy jitcode.constants[0] to registers[255],
  55. jitcode.constants[1] to registers[254],
  56. jitcode.constants[2] to registers[253], etc."""
  57. if nonconst.NonConstant(0): # force the right type
  58. constants[0] = ConstClass.value # (useful for small tests)
  59. i = len(constants) - 1
  60. while i >= 0:
  61. j = 255 - i
  62. assert j >= 0
  63. registers[j] = ConstClass(constants[i])
  64. i -= 1
  65. def cleanup_registers(self):
  66. # To avoid keeping references alive, this cleans up the registers_r.
  67. # It does not clear the references set by copy_constants(), but
  68. # these are all prebuilt constants anyway.
  69. for i in range(self.jitcode.num_regs_r()):
  70. self.registers_r[i] = None
  71. # ------------------------------
  72. # Decoding of the JitCode
  73. @specialize.arg(4)
  74. def prepare_list_of_boxes(self, outvalue, startindex, position, argcode):
  75. assert argcode in 'IRF'
  76. code = self.bytecode
  77. length = ord(code[position])
  78. position += 1
  79. for i in range(length):
  80. index = ord(code[position+i])
  81. if argcode == 'I': reg = self.registers_i[index]
  82. elif argcode == 'R': reg = self.registers_r[index]
  83. elif argcode == 'F': reg = self.registers_f[index]
  84. else: raise AssertionError(argcode)
  85. outvalue[startindex+i] = reg
  86. def _put_back_list_of_boxes(self, outvalue, startindex, position):
  87. code = self.bytecode
  88. length = ord(code[position])
  89. position += 1
  90. for i in range(length):
  91. index = ord(code[position+i])
  92. box = outvalue[startindex+i]
  93. if box.type == history.INT: self.registers_i[index] = box
  94. elif box.type == history.REF: self.registers_r[index] = box
  95. elif box.type == history.FLOAT: self.registers_f[index] = box
  96. else: raise AssertionError(box.type)
  97. def get_current_position_info(self):
  98. return self.jitcode.get_live_vars_info(self.pc)
  99. def get_list_of_active_boxes(self, in_a_call):
  100. if in_a_call:
  101. # If we are not the topmost frame, self._result_argcode contains
  102. # the type of the result of the call instruction in the bytecode.
  103. # We use it to clear the box that will hold the result: this box
  104. # is not defined yet.
  105. argcode = self._result_argcode
  106. index = ord(self.bytecode[self.pc - 1])
  107. if argcode == 'i': self.registers_i[index] = history.CONST_FALSE
  108. elif argcode == 'r': self.registers_r[index] = history.CONST_NULL
  109. elif argcode == 'f': self.registers_f[index] = history.CONST_FZERO
  110. self._result_argcode = '?' # done
  111. #
  112. info = self.get_current_position_info()
  113. start_i = 0
  114. start_r = start_i + info.get_register_count_i()
  115. start_f = start_r + info.get_register_count_r()
  116. total = start_f + info.get_register_count_f()
  117. # allocate a list of the correct size
  118. env = [None] * total
  119. make_sure_not_resized(env)
  120. # fill it now
  121. for i in range(info.get_register_count_i()):
  122. index = info.get_register_index_i(i)
  123. env[start_i + i] = self.registers_i[index]
  124. for i in range(info.get_register_count_r()):
  125. index = info.get_register_index_r(i)
  126. env[start_r + i] = self.registers_r[index]
  127. for i in range(info.get_register_count_f()):
  128. index = info.get_register_index_f(i)
  129. env[start_f + i] = self.registers_f[index]
  130. return env
  131. def replace_active_box_in_frame(self, oldbox, newbox):
  132. if isinstance(oldbox, history.BoxInt):
  133. count = self.jitcode.num_regs_i()
  134. registers = self.registers_i
  135. elif isinstance(oldbox, history.BoxPtr):
  136. count = self.jitcode.num_regs_r()
  137. registers = self.registers_r
  138. elif isinstance(oldbox, history.BoxFloat):
  139. count = self.jitcode.num_regs_f()
  140. registers = self.registers_f
  141. else:
  142. assert 0, oldbox
  143. for i in range(count):
  144. if registers[i] is oldbox:
  145. registers[i] = newbox
  146. if not we_are_translated():
  147. for b in registers[count:]:
  148. assert not oldbox.same_box(b)
  149. def make_result_of_lastop(self, resultbox):
  150. got_type = resultbox.type
  151. # XXX disabled for now, conflicts with str_guard_value
  152. #if not we_are_translated():
  153. # typeof = {'i': history.INT,
  154. # 'r': history.REF,
  155. # 'f': history.FLOAT}
  156. # assert typeof[self.jitcode._resulttypes[self.pc]] == got_type
  157. target_index = ord(self.bytecode[self.pc-1])
  158. if got_type == history.INT:
  159. self.registers_i[target_index] = resultbox
  160. elif got_type == history.REF:
  161. #debug_print(' ->',
  162. # llmemory.cast_ptr_to_adr(resultbox.getref_base()))
  163. self.registers_r[target_index] = resultbox
  164. elif got_type == history.FLOAT:
  165. self.registers_f[target_index] = resultbox
  166. else:
  167. raise AssertionError("bad result box type")
  168. # ------------------------------
  169. for _opimpl in ['int_add', 'int_sub', 'int_mul', 'int_floordiv', 'int_mod',
  170. 'int_lt', 'int_le', 'int_eq',
  171. 'int_ne', 'int_gt', 'int_ge',
  172. 'int_and', 'int_or', 'int_xor',
  173. 'int_rshift', 'int_lshift', 'uint_rshift',
  174. 'uint_lt', 'uint_le', 'uint_gt', 'uint_ge',
  175. 'uint_floordiv',
  176. 'float_add', 'float_sub', 'float_mul', 'float_truediv',
  177. 'float_lt', 'float_le', 'float_eq',
  178. 'float_ne', 'float_gt', 'float_ge',
  179. 'ptr_eq', 'ptr_ne', 'instance_ptr_eq', 'instance_ptr_ne',
  180. ]:
  181. exec py.code.Source('''
  182. @arguments("box", "box")
  183. def opimpl_%s(self, b1, b2):
  184. return self.execute(rop.%s, b1, b2)
  185. ''' % (_opimpl, _opimpl.upper())).compile()
  186. for _opimpl in ['int_add_ovf', 'int_sub_ovf', 'int_mul_ovf']:
  187. exec py.code.Source('''
  188. @arguments("box", "box")
  189. def opimpl_%s(self, b1, b2):
  190. self.metainterp.clear_exception()
  191. resbox = self.execute(rop.%s, b1, b2)
  192. self.make_result_of_lastop(resbox) # same as execute_varargs()
  193. if not isinstance(resbox, Const):
  194. self.metainterp.handle_possible_overflow_error()
  195. return resbox
  196. ''' % (_opimpl, _opimpl.upper())).compile()
  197. for _opimpl in ['int_is_true', 'int_is_zero', 'int_neg', 'int_invert',
  198. 'cast_float_to_int', 'cast_int_to_float',
  199. 'cast_float_to_singlefloat', 'cast_singlefloat_to_float',
  200. 'float_neg', 'float_abs',
  201. 'cast_ptr_to_int', 'cast_int_to_ptr',
  202. 'convert_float_bytes_to_longlong',
  203. 'convert_longlong_bytes_to_float', 'int_force_ge_zero',
  204. ]:
  205. exec py.code.Source('''
  206. @arguments("box")
  207. def opimpl_%s(self, b):
  208. return self.execute(rop.%s, b)
  209. ''' % (_opimpl, _opimpl.upper())).compile()
  210. @arguments("box")
  211. def opimpl_ptr_nonzero(self, box):
  212. return self.execute(rop.PTR_NE, box, history.CONST_NULL)
  213. @arguments("box")
  214. def opimpl_ptr_iszero(self, box):
  215. return self.execute(rop.PTR_EQ, box, history.CONST_NULL)
  216. @arguments("box")
  217. def opimpl_mark_opaque_ptr(self, box):
  218. return self.execute(rop.MARK_OPAQUE_PTR, box)
  219. @arguments("box", "box")
  220. def opimpl_record_known_class(self, box, clsbox):
  221. from pypy.rpython.lltypesystem import llmemory
  222. if self.metainterp.heapcache.is_class_known(box):
  223. return
  224. adr = clsbox.getaddr()
  225. bounding_class = llmemory.cast_adr_to_ptr(adr, rclass.CLASSTYPE)
  226. if bounding_class.subclassrange_max - bounding_class.subclassrange_min == 1:
  227. # precise class knowledge, this can be used
  228. self.execute(rop.RECORD_KNOWN_CLASS, box, clsbox)
  229. self.metainterp.heapcache.class_now_known(box)
  230. @arguments("box")
  231. def _opimpl_any_return(self, box):
  232. self.metainterp.finishframe(box)
  233. opimpl_int_return = _opimpl_any_return
  234. opimpl_ref_return = _opimpl_any_return
  235. opimpl_float_return = _opimpl_any_return
  236. @arguments()
  237. def opimpl_void_return(self):
  238. self.metainterp.finishframe(None)
  239. @arguments("box")
  240. def _opimpl_any_copy(self, box):
  241. return box
  242. opimpl_int_copy = _opimpl_any_copy
  243. opimpl_ref_copy = _opimpl_any_copy
  244. opimpl_float_copy = _opimpl_any_copy
  245. @arguments("box")
  246. def _opimpl_any_push(self, box):
  247. self.pushed_box = box
  248. opimpl_int_push = _opimpl_any_push
  249. opimpl_ref_push = _opimpl_any_push
  250. opimpl_float_push = _opimpl_any_push
  251. @arguments()
  252. def _opimpl_any_pop(self):
  253. box = self.pushed_box
  254. self.pushed_box = None
  255. return box
  256. opimpl_int_pop = _opimpl_any_pop
  257. opimpl_ref_pop = _opimpl_any_pop
  258. opimpl_float_pop = _opimpl_any_pop
  259. @arguments("label")
  260. def opimpl_catch_exception(self, target):
  261. """This is a no-op when run normally. We can check that
  262. last_exc_value_box is None; it should have been set to None
  263. by the previous instruction. If the previous instruction
  264. raised instead, finishframe_exception() should have been
  265. called and we would not be there."""
  266. assert self.metainterp.last_exc_value_box is None
  267. @arguments("label")
  268. def opimpl_goto(self, target):
  269. self.pc = target
  270. @arguments("box", "label")
  271. def opimpl_goto_if_not(self, box, target):
  272. switchcase = box.getint()
  273. if switchcase:
  274. opnum = rop.GUARD_TRUE
  275. else:
  276. opnum = rop.GUARD_FALSE
  277. self.generate_guard(opnum, box)
  278. if not switchcase:
  279. self.pc = target
  280. @arguments("box", "label")
  281. def opimpl_goto_if_not_int_is_true(self, box, target):
  282. condbox = self.execute(rop.INT_IS_TRUE, box)
  283. self.opimpl_goto_if_not(condbox, target)
  284. @arguments("box", "label")
  285. def opimpl_goto_if_not_int_is_zero(self, box, target):
  286. condbox = self.execute(rop.INT_IS_ZERO, box)
  287. self.opimpl_goto_if_not(condbox, target)
  288. for _opimpl in ['int_lt', 'int_le', 'int_eq', 'int_ne', 'int_gt', 'int_ge',
  289. 'ptr_eq', 'ptr_ne']:
  290. exec py.code.Source('''
  291. @arguments("box", "box", "label")
  292. def opimpl_goto_if_not_%s(self, b1, b2, target):
  293. condbox = self.execute(rop.%s, b1, b2)
  294. self.opimpl_goto_if_not(condbox, target)
  295. ''' % (_opimpl, _opimpl.upper())).compile()
  296. def _establish_nullity(self, box, orgpc):
  297. value = box.nonnull()
  298. if value:
  299. if not self.metainterp.heapcache.is_class_known(box):
  300. self.generate_guard(rop.GUARD_NONNULL, box, resumepc=orgpc)
  301. else:
  302. if not isinstance(box, Const):
  303. self.generate_guard(rop.GUARD_ISNULL, box, resumepc=orgpc)
  304. promoted_box = box.constbox()
  305. self.metainterp.replace_box(box, promoted_box)
  306. return value
  307. @arguments("box", "label", "orgpc")
  308. def opimpl_goto_if_not_ptr_nonzero(self, box, target, orgpc):
  309. if not self._establish_nullity(box, orgpc):
  310. self.pc = target
  311. @arguments("box", "label", "orgpc")
  312. def opimpl_goto_if_not_ptr_iszero(self, box, target, orgpc):
  313. if self._establish_nullity(box, orgpc):
  314. self.pc = target
  315. @arguments("box", "box", "box")
  316. def opimpl_int_between(self, b1, b2, b3):
  317. b5 = self.execute(rop.INT_SUB, b3, b1)
  318. if isinstance(b5, ConstInt) and b5.getint() == 1:
  319. # the common case of int_between(a, b, a+1) turns into just INT_EQ
  320. return self.execute(rop.INT_EQ, b2, b1)
  321. else:
  322. b4 = self.execute(rop.INT_SUB, b2, b1)
  323. return self.execute(rop.UINT_LT, b4, b5)
  324. @arguments("box", "descr", "orgpc")
  325. def opimpl_switch(self, valuebox, switchdict, orgpc):
  326. box = self.implement_guard_value(valuebox, orgpc)
  327. search_value = box.getint()
  328. assert isinstance(switchdict, SwitchDictDescr)
  329. try:
  330. self.pc = switchdict.dict[search_value]
  331. except KeyError:
  332. pass
  333. @arguments()
  334. def opimpl_unreachable(self):
  335. raise AssertionError("unreachable")
  336. @arguments("descr")
  337. def opimpl_new(self, sizedescr):
  338. resbox = self.execute_with_descr(rop.NEW, sizedescr)
  339. self.metainterp.heapcache.new(resbox)
  340. return resbox
  341. @arguments("descr")
  342. def opimpl_new_with_vtable(self, sizedescr):
  343. cpu = self.metainterp.cpu
  344. cls = heaptracker.descr2vtable(cpu, sizedescr)
  345. resbox = self.execute(rop.NEW_WITH_VTABLE, ConstInt(cls))
  346. self.metainterp.heapcache.new(resbox)
  347. self.metainterp.heapcache.class_now_known(resbox)
  348. return resbox
  349. @arguments("box", "descr")
  350. def opimpl_new_array(self, lengthbox, itemsizedescr):
  351. resbox = self.execute_with_descr(rop.NEW_ARRAY, itemsizedescr, lengthbox)
  352. self.metainterp.heapcache.new_array(resbox, lengthbox)
  353. return resbox
  354. @specialize.arg(1)
  355. def _do_getarrayitem_gc_any(self, op, arraybox, indexbox, arraydescr):
  356. tobox = self.metainterp.heapcache.getarrayitem(
  357. arraybox, indexbox, arraydescr)
  358. if tobox:
  359. # sanity check: see whether the current array value
  360. # corresponds to what the cache thinks the value is
  361. resbox = executor.execute(self.metainterp.cpu, self.metainterp, op,
  362. arraydescr, arraybox, indexbox)
  363. assert resbox.constbox().same_constant(tobox.constbox())
  364. return tobox
  365. resbox = self.execute_with_descr(op, arraydescr, arraybox, indexbox)
  366. self.metainterp.heapcache.getarrayitem_now_known(
  367. arraybox, indexbox, resbox, arraydescr)
  368. return resbox
  369. @arguments("box", "box", "descr")
  370. def _opimpl_getarrayitem_gc_any(self, arraybox, indexbox, arraydescr):
  371. return self._do_getarrayitem_gc_any(rop.GETARRAYITEM_GC, arraybox,
  372. indexbox, arraydescr)
  373. opimpl_getarrayitem_gc_i = _opimpl_getarrayitem_gc_any
  374. opimpl_getarrayitem_gc_r = _opimpl_getarrayitem_gc_any
  375. opimpl_getarrayitem_gc_f = _opimpl_getarrayitem_gc_any
  376. @arguments("box", "box", "descr")
  377. def _opimpl_getarrayitem_raw_any(self, arraybox, indexbox, arraydescr):
  378. return self.execute_with_descr(rop.GETARRAYITEM_RAW,
  379. arraydescr, arraybox, indexbox)
  380. opimpl_getarrayitem_raw_i = _opimpl_getarrayitem_raw_any
  381. opimpl_getarrayitem_raw_f = _opimpl_getarrayitem_raw_any
  382. @arguments("box", "box", "descr")
  383. def _opimpl_getarrayitem_raw_pure_any(self, arraybox, indexbox,
  384. arraydescr):
  385. return self.execute_with_descr(rop.GETARRAYITEM_RAW_PURE,
  386. arraydescr, arraybox, indexbox)
  387. opimpl_getarrayitem_raw_i_pure = _opimpl_getarrayitem_raw_pure_any
  388. opimpl_getarrayitem_raw_f_pure = _opimpl_getarrayitem_raw_pure_any
  389. @arguments("box", "box", "descr")
  390. def _opimpl_getarrayitem_gc_pure_any(self, arraybox, indexbox, arraydescr):
  391. if isinstance(arraybox, ConstPtr) and isinstance(indexbox, ConstInt):
  392. # if the arguments are directly constants, bypass the heapcache
  393. # completely
  394. resbox = executor.execute(self.metainterp.cpu, self.metainterp,
  395. rop.GETARRAYITEM_GC_PURE, arraydescr,
  396. arraybox, indexbox)
  397. return resbox.constbox()
  398. return self._do_getarrayitem_gc_any(rop.GETARRAYITEM_GC_PURE, arraybox,
  399. indexbox, arraydescr)
  400. opimpl_getarrayitem_gc_i_pure = _opimpl_getarrayitem_gc_pure_any
  401. opimpl_getarrayitem_gc_r_pure = _opimpl_getarrayitem_gc_pure_any
  402. opimpl_getarrayitem_gc_f_pure = _opimpl_getarrayitem_gc_pure_any
  403. @arguments("box", "box", "box", "descr")
  404. def _opimpl_setarrayitem_gc_any(self, arraybox, indexbox, itembox,
  405. arraydescr):
  406. self.execute_with_descr(rop.SETARRAYITEM_GC, arraydescr, arraybox,
  407. indexbox, itembox)
  408. self.metainterp.heapcache.setarrayitem(
  409. arraybox, indexbox, itembox, arraydescr)
  410. opimpl_setarrayitem_gc_i = _opimpl_setarrayitem_gc_any
  411. opimpl_setarrayitem_gc_r = _opimpl_setarrayitem_gc_any
  412. opimpl_setarrayitem_gc_f = _opimpl_setarrayitem_gc_any
  413. @arguments("box", "box", "box", "descr")
  414. def _opimpl_setarrayitem_raw_any(self, arraybox, indexbox, itembox,
  415. arraydescr):
  416. self.execute_with_descr(rop.SETARRAYITEM_RAW, arraydescr, arraybox,
  417. indexbox, itembox)
  418. opimpl_setarrayitem_raw_i = _opimpl_setarrayitem_raw_any
  419. opimpl_setarrayitem_raw_f = _opimpl_setarrayitem_raw_any
  420. @arguments("box", "descr")
  421. def opimpl_arraylen_gc(self, arraybox, arraydescr):
  422. lengthbox = self.metainterp.heapcache.arraylen(arraybox)
  423. if lengthbox is None:
  424. lengthbox = self.execute_with_descr(
  425. rop.ARRAYLEN_GC, arraydescr, arraybox)
  426. self.metainterp.heapcache.arraylen_now_known(arraybox, lengthbox)
  427. return lengthbox
  428. @arguments("box", "box", "descr", "orgpc")
  429. def opimpl_check_neg_index(self, arraybox, indexbox, arraydescr, orgpc):
  430. negbox = self.metainterp.execute_and_record(
  431. rop.INT_LT, None, indexbox, history.CONST_FALSE)
  432. negbox = self.implement_guard_value(negbox, orgpc)
  433. if negbox.getint():
  434. # the index is < 0; add the array length to it
  435. lengthbox = self.opimpl_arraylen_gc(arraybox, arraydescr)
  436. indexbox = self.metainterp.execute_and_record(
  437. rop.INT_ADD, None, indexbox, lengthbox)
  438. return indexbox
  439. @arguments("box", "descr", "descr", "descr", "descr")
  440. def opimpl_newlist(self, sizebox, structdescr, lengthdescr,
  441. itemsdescr, arraydescr):
  442. sbox = self.opimpl_new(structdescr)
  443. self._opimpl_setfield_gc_any(sbox, sizebox, lengthdescr)
  444. abox = self.opimpl_new_array(sizebox, arraydescr)
  445. self._opimpl_setfield_gc_any(sbox, abox, itemsdescr)
  446. return sbox
  447. @arguments("box", "descr", "descr", "descr", "descr")
  448. def opimpl_newlist_hint(self, sizehintbox, structdescr, lengthdescr,
  449. itemsdescr, arraydescr):
  450. sbox = self.opimpl_new(structdescr)
  451. self._opimpl_setfield_gc_any(sbox, history.CONST_FALSE, lengthdescr)
  452. abox = self.opimpl_new_array(sizehintbox, arraydescr)
  453. self._opimpl_setfield_gc_any(sbox, abox, itemsdescr)
  454. return sbox
  455. @arguments("box", "box", "descr", "descr")
  456. def _opimpl_getlistitem_gc_any(self, listbox, indexbox,
  457. itemsdescr, arraydescr):
  458. arraybox = self._opimpl_getfield_gc_any(listbox, itemsdescr)
  459. return self._opimpl_getarrayitem_gc_any(arraybox, indexbox, arraydescr)
  460. opimpl_getlistitem_gc_i = _opimpl_getlistitem_gc_any
  461. opimpl_getlistitem_gc_r = _opimpl_getlistitem_gc_any
  462. opimpl_getlistitem_gc_f = _opimpl_getlistitem_gc_any
  463. @arguments("box", "box", "box", "descr", "descr")
  464. def _opimpl_setlistitem_gc_any(self, listbox, indexbox, valuebox,
  465. itemsdescr, arraydescr):
  466. arraybox = self._opimpl_getfield_gc_any(listbox, itemsdescr)
  467. self._opimpl_setarrayitem_gc_any(arraybox, indexbox, valuebox,
  468. arraydescr)
  469. opimpl_setlistitem_gc_i = _opimpl_setlistitem_gc_any
  470. opimpl_setlistitem_gc_r = _opimpl_setlistitem_gc_any
  471. opimpl_setlistitem_gc_f = _opimpl_setlistitem_gc_any
  472. @arguments("box", "box", "descr", "orgpc")
  473. def opimpl_check_resizable_neg_index(self, listbox, indexbox,
  474. lengthdescr, orgpc):
  475. negbox = self.metainterp.execute_and_record(
  476. rop.INT_LT, None, indexbox, history.CONST_FALSE)
  477. negbox = self.implement_guard_value(negbox, orgpc)
  478. if negbox.getint():
  479. # the index is < 0; add the array length to it
  480. lenbox = self.metainterp.execute_and_record(
  481. rop.GETFIELD_GC, lengthdescr, listbox)
  482. indexbox = self.metainterp.execute_and_record(
  483. rop.INT_ADD, None, indexbox, lenbox)
  484. return indexbox
  485. @arguments("box", "descr")
  486. def _opimpl_getfield_gc_any(self, box, fielddescr):
  487. return self._opimpl_getfield_gc_any_pureornot(
  488. rop.GETFIELD_GC, box, fielddescr)
  489. opimpl_getfield_gc_i = _opimpl_getfield_gc_any
  490. opimpl_getfield_gc_r = _opimpl_getfield_gc_any
  491. opimpl_getfield_gc_f = _opimpl_getfield_gc_any
  492. @arguments("box", "descr")
  493. def _opimpl_getfield_gc_pure_any(self, box, fielddescr):
  494. if isinstance(box, ConstPtr):
  495. # if 'box' is directly a ConstPtr, bypass the heapcache completely
  496. resbox = executor.execute(self.metainterp.cpu, self.metainterp,
  497. rop.GETFIELD_GC_PURE, fielddescr, box)
  498. return resbox.constbox()
  499. return self._opimpl_getfield_gc_any_pureornot(
  500. rop.GETFIELD_GC_PURE, box, fielddescr)
  501. opimpl_getfield_gc_i_pure = _opimpl_getfield_gc_pure_any
  502. opimpl_getfield_gc_r_pure = _opimpl_getfield_gc_pure_any
  503. opimpl_getfield_gc_f_pure = _opimpl_getfield_gc_pure_any
  504. @arguments("box", "box", "descr")
  505. def _opimpl_getinteriorfield_gc_any(self, array, index, descr):
  506. return self.execute_with_descr(rop.GETINTERIORFIELD_GC, descr,
  507. array, index)
  508. opimpl_getinteriorfield_gc_i = _opimpl_getinteriorfield_gc_any
  509. opimpl_getinteriorfield_gc_f = _opimpl_getinteriorfield_gc_any
  510. opimpl_getinteriorfield_gc_r = _opimpl_getinteriorfield_gc_any
  511. @specialize.arg(1)
  512. def _opimpl_getfield_gc_any_pureornot(self, opnum, box, fielddescr):
  513. tobox = self.metainterp.heapcache.getfield(box, fielddescr)
  514. if tobox is not None:
  515. # sanity check: see whether the current struct value
  516. # corresponds to what the cache thinks the value is
  517. #resbox = executor.execute(self.metainterp.cpu, self.metainterp,
  518. # rop.GETFIELD_GC, fielddescr, box)
  519. # XXX the sanity check does not seem to do anything, remove?
  520. return tobox
  521. resbox = self.execute_with_descr(opnum, fielddescr, box)
  522. self.metainterp.heapcache.getfield_now_known(box, fielddescr, resbox)
  523. return resbox
  524. @arguments("box", "descr", "orgpc")
  525. def _opimpl_getfield_gc_greenfield_any(self, box, fielddescr, pc):
  526. ginfo = self.metainterp.jitdriver_sd.greenfield_info
  527. if (ginfo is not None and fielddescr in ginfo.green_field_descrs
  528. and not self._nonstandard_virtualizable(pc, box)):
  529. # fetch the result, but consider it as a Const box and don't
  530. # record any operation
  531. resbox = executor.execute(self.metainterp.cpu, self.metainterp,
  532. rop.GETFIELD_GC_PURE, fielddescr, box)
  533. return resbox.constbox()
  534. # fall-back
  535. return self.execute_with_descr(rop.GETFIELD_GC_PURE, fielddescr, box)
  536. opimpl_getfield_gc_i_greenfield = _opimpl_getfield_gc_greenfield_any
  537. opimpl_getfield_gc_r_greenfield = _opimpl_getfield_gc_greenfield_any
  538. opimpl_getfield_gc_f_greenfield = _opimpl_getfield_gc_greenfield_any
  539. @arguments("box", "box", "descr")
  540. def _opimpl_setfield_gc_any(self, box, valuebox, fielddescr):
  541. tobox = self.metainterp.heapcache.getfield(box, fielddescr)
  542. if tobox is valuebox:
  543. return
  544. self.execute_with_descr(rop.SETFIELD_GC, fielddescr, box, valuebox)
  545. self.metainterp.heapcache.setfield(box, valuebox, fielddescr)
  546. opimpl_setfield_gc_i = _opimpl_setfield_gc_any
  547. opimpl_setfield_gc_r = _opimpl_setfield_gc_any
  548. opimpl_setfield_gc_f = _opimpl_setfield_gc_any
  549. @arguments("box", "box", "box", "descr")
  550. def _opimpl_setinteriorfield_gc_any(self, array, index, value, descr):
  551. self.execute_with_descr(rop.SETINTERIORFIELD_GC, descr,
  552. array, index, value)
  553. opimpl_setinteriorfield_gc_i = _opimpl_setinteriorfield_gc_any
  554. opimpl_setinteriorfield_gc_f = _opimpl_setinteriorfield_gc_any
  555. opimpl_setinteriorfield_gc_r = _opimpl_setinteriorfield_gc_any
  556. @arguments("box", "descr")
  557. def _opimpl_getfield_raw_any(self, box, fielddescr):
  558. return self.execute_with_descr(rop.GETFIELD_RAW, fielddescr, box)
  559. opimpl_getfield_raw_i = _opimpl_getfield_raw_any
  560. opimpl_getfield_raw_r = _opimpl_getfield_raw_any
  561. opimpl_getfield_raw_f = _opimpl_getfield_raw_any
  562. @arguments("box", "descr")
  563. def _opimpl_getfield_raw_pure_any(self, box, fielddescr):
  564. return self.execute_with_descr(rop.GETFIELD_RAW_PURE, fielddescr, box)
  565. opimpl_getfield_raw_i_pure = _opimpl_getfield_raw_pure_any
  566. opimpl_getfield_raw_r_pure = _opimpl_getfield_raw_pure_any
  567. opimpl_getfield_raw_f_pure = _opimpl_getfield_raw_pure_any
  568. @arguments("box", "box", "descr")
  569. def _opimpl_setfield_raw_any(self, box, valuebox, fielddescr):
  570. self.execute_with_descr(rop.SETFIELD_RAW, fielddescr, box, valuebox)
  571. opimpl_setfield_raw_i = _opimpl_setfield_raw_any
  572. opimpl_setfield_raw_r = _opimpl_setfield_raw_any
  573. opimpl_setfield_raw_f = _opimpl_setfield_raw_any
  574. @arguments("box", "box", "box", "descr")
  575. def _opimpl_raw_store(self, addrbox, offsetbox, valuebox, arraydescr):
  576. self.execute_with_descr(rop.RAW_STORE, arraydescr,
  577. addrbox, offsetbox, valuebox)
  578. opimpl_raw_store_i = _opimpl_raw_store
  579. opimpl_raw_store_f = _opimpl_raw_store
  580. @arguments("box", "box", "descr")
  581. def _opimpl_raw_load(self, addrbox, offsetbox, arraydescr):
  582. return self.execute_with_descr(rop.RAW_LOAD, arraydescr,
  583. addrbox, offsetbox)
  584. opimpl_raw_load_i = _opimpl_raw_load
  585. opimpl_raw_load_f = _opimpl_raw_load
  586. @arguments("box", "descr", "descr", "orgpc")
  587. def opimpl_record_quasiimmut_field(self, box, fielddescr,
  588. mutatefielddescr, orgpc):
  589. from pypy.jit.metainterp.quasiimmut import QuasiImmutDescr
  590. cpu = self.metainterp.cpu
  591. descr = QuasiImmutDescr(cpu, box, fielddescr, mutatefielddescr)
  592. self.metainterp.history.record(rop.QUASIIMMUT_FIELD, [box],
  593. None, descr=descr)
  594. self.generate_guard(rop.GUARD_NOT_INVALIDATED, resumepc=orgpc)
  595. @arguments("box", "descr", "orgpc")
  596. def opimpl_jit_force_quasi_immutable(self, box, mutatefielddescr, orgpc):
  597. # During tracing, a 'jit_force_quasi_immutable' usually turns into
  598. # the operations that check that the content of 'mutate_xxx' is null.
  599. # If it is actually not null already now, then we abort tracing.
  600. # The idea is that if we use 'jit_force_quasi_immutable' on a freshly
  601. # allocated object, then the GETFIELD_GC will know that the answer is
  602. # null, and the guard will be removed. So the fact that the field is
  603. # quasi-immutable will have no effect, and instead it will work as a
  604. # regular, probably virtual, structure.
  605. mutatebox = self.execute_with_descr(rop.GETFIELD_GC,
  606. mutatefielddescr, box)
  607. if mutatebox.nonnull():
  608. from pypy.jit.metainterp.quasiimmut import do_force_quasi_immutable
  609. do_force_quasi_immutable(self.metainterp.cpu, box.getref_base(),
  610. mutatefielddescr)
  611. raise SwitchToBlackhole(Counters.ABORT_FORCE_QUASIIMMUT)
  612. self.generate_guard(rop.GUARD_ISNULL, mutatebox, resumepc=orgpc)
  613. def _nonstandard_virtualizable(self, pc, box):
  614. # returns True if 'box' is actually not the "standard" virtualizable
  615. # that is stored in metainterp.virtualizable_boxes[-1]
  616. if (self.metainterp.jitdriver_sd.virtualizable_info is None and
  617. self.metainterp.jitdriver_sd.greenfield_info is None):
  618. return True # can occur in case of multiple JITs
  619. standard_box = self.metainterp.virtualizable_boxes[-1]
  620. if standard_box is box:
  621. return False
  622. if self.metainterp.heapcache.is_nonstandard_virtualizable(box):
  623. return True
  624. eqbox = self.metainterp.execute_and_record(rop.PTR_EQ, None,
  625. box, standard_box)
  626. eqbox = self.implement_guard_value(eqbox, pc)
  627. isstandard = eqbox.getint()
  628. if isstandard:
  629. self.metainterp.replace_box(box, standard_box)
  630. else:
  631. self.metainterp.heapcache.nonstandard_virtualizables_now_known(box)
  632. return not isstandard
  633. def _get_virtualizable_field_index(self, fielddescr):
  634. # Get the index of a fielddescr. Must only be called for
  635. # the "standard" virtualizable.
  636. vinfo = self.metainterp.jitdriver_sd.virtualizable_info
  637. return vinfo.static_field_by_descrs[fielddescr]
  638. @arguments("box", "descr", "orgpc")
  639. def _opimpl_getfield_vable(self, box, fielddescr, pc):
  640. if self._nonstandard_virtualizable(pc, box):
  641. return self._opimpl_getfield_gc_any(box, fielddescr)
  642. self.metainterp.check_synchronized_virtualizable()
  643. index = self._get_virtualizable_field_index(fielddescr)
  644. return self.metainterp.virtualizable_boxes[index]
  645. opimpl_getfield_vable_i = _opimpl_getfield_vable
  646. opimpl_getfield_vable_r = _opimpl_getfield_vable
  647. opimpl_getfield_vable_f = _opimpl_getfield_vable
  648. @arguments("box", "box", "descr", "orgpc")
  649. def _opimpl_setfield_vable(self, box, valuebox, fielddescr, pc):
  650. if self._nonstandard_virtualizable(pc, box):
  651. return self._opimpl_setfield_gc_any(box, valuebox, fielddescr)
  652. index = self._get_virtualizable_field_index(fielddescr)
  653. self.metainterp.virtualizable_boxes[index] = valuebox
  654. self.metainterp.synchronize_virtualizable()
  655. # XXX only the index'th field needs to be synchronized, really
  656. opimpl_setfield_vable_i = _opimpl_setfield_vable
  657. opimpl_setfield_vable_r = _opimpl_setfield_vable
  658. opimpl_setfield_vable_f = _opimpl_setfield_vable
  659. def _get_arrayitem_vable_index(self, pc, arrayfielddescr, indexbox):
  660. # Get the index of an array item: the index'th of the array
  661. # described by arrayfielddescr. Must only be called for
  662. # the "standard" virtualizable.
  663. indexbox = self.implement_guard_value(indexbox, pc)
  664. vinfo = self.metainterp.jitdriver_sd.virtualizable_info
  665. virtualizable_box = self.metainterp.virtualizable_boxes[-1]
  666. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  667. arrayindex = vinfo.array_field_by_descrs[arrayfielddescr]
  668. index = indexbox.getint()
  669. # Support for negative index: disabled
  670. # (see codewriter/jtransform.py, _check_no_vable_array).
  671. #if index < 0:
  672. # index += vinfo.get_array_length(virtualizable, arrayindex)
  673. assert 0 <= index < vinfo.get_array_length(virtualizable, arrayindex)
  674. return vinfo.get_index_in_array(virtualizable, arrayindex, index)
  675. @arguments("box", "box", "descr", "descr", "orgpc")
  676. def _opimpl_getarrayitem_vable(self, box, indexbox, fdescr, adescr, pc):
  677. if self._nonstandard_virtualizable(pc, box):
  678. arraybox = self._opimpl_getfield_gc_any(box, fdescr)
  679. return self._opimpl_getarrayitem_gc_any(arraybox, indexbox, adescr)
  680. self.metainterp.check_synchronized_virtualizable()
  681. index = self._get_arrayitem_vable_index(pc, fdescr, indexbox)
  682. return self.metainterp.virtualizable_boxes[index]
  683. opimpl_getarrayitem_vable_i = _opimpl_getarrayitem_vable
  684. opimpl_getarrayitem_vable_r = _opimpl_getarrayitem_vable
  685. opimpl_getarrayitem_vable_f = _opimpl_getarrayitem_vable
  686. @arguments("box", "box", "box", "descr", "descr", "orgpc")
  687. def _opimpl_setarrayitem_vable(self, box, indexbox, valuebox,
  688. fdescr, adescr, pc):
  689. if self._nonstandard_virtualizable(pc, box):
  690. arraybox = self._opimpl_getfield_gc_any(box, fdescr)
  691. self._opimpl_setarrayitem_gc_any(arraybox, indexbox, valuebox,
  692. adescr)
  693. return
  694. index = self._get_arrayitem_vable_index(pc, fdescr, indexbox)
  695. self.metainterp.virtualizable_boxes[index] = valuebox
  696. self.metainterp.synchronize_virtualizable()
  697. # XXX only the index'th field needs to be synchronized, really
  698. opimpl_setarrayitem_vable_i = _opimpl_setarrayitem_vable
  699. opimpl_setarrayitem_vable_r = _opimpl_setarrayitem_vable
  700. opimpl_setarrayitem_vable_f = _opimpl_setarrayitem_vable
  701. @arguments("box", "descr", "descr", "orgpc")
  702. def opimpl_arraylen_vable(self, box, fdescr, adescr, pc):
  703. if self._nonstandard_virtualizable(pc, box):
  704. arraybox = self._opimpl_getfield_gc_any(box, fdescr)
  705. return self.opimpl_arraylen_gc(arraybox, adescr)
  706. vinfo = self.metainterp.jitdriver_sd.virtualizable_info
  707. virtualizable_box = self.metainterp.virtualizable_boxes[-1]
  708. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  709. arrayindex = vinfo.array_field_by_descrs[fdescr]
  710. result = vinfo.get_array_length(virtualizable, arrayindex)
  711. return ConstInt(result)
  712. @arguments("jitcode", "boxes")
  713. def _opimpl_inline_call1(self, jitcode, argboxes):
  714. return self.metainterp.perform_call(jitcode, argboxes)
  715. @arguments("jitcode", "boxes2")
  716. def _opimpl_inline_call2(self, jitcode, argboxes):
  717. return self.metainterp.perform_call(jitcode, argboxes)
  718. @arguments("jitcode", "boxes3")
  719. def _opimpl_inline_call3(self, jitcode, argboxes):
  720. return self.metainterp.perform_call(jitcode, argboxes)
  721. opimpl_inline_call_r_i = _opimpl_inline_call1
  722. opimpl_inline_call_r_r = _opimpl_inline_call1
  723. opimpl_inline_call_r_v = _opimpl_inline_call1
  724. opimpl_inline_call_ir_i = _opimpl_inline_call2
  725. opimpl_inline_call_ir_r = _opimpl_inline_call2
  726. opimpl_inline_call_ir_v = _opimpl_inline_call2
  727. opimpl_inline_call_irf_i = _opimpl_inline_call3
  728. opimpl_inline_call_irf_r = _opimpl_inline_call3
  729. opimpl_inline_call_irf_f = _opimpl_inline_call3
  730. opimpl_inline_call_irf_v = _opimpl_inline_call3
  731. @arguments("box", "boxes", "descr")
  732. def _opimpl_residual_call1(self, funcbox, argboxes, calldescr):
  733. return self.do_residual_or_indirect_call(funcbox, argboxes, calldescr)
  734. @arguments("box", "boxes2", "descr")
  735. def _opimpl_residual_call2(self, funcbox, argboxes, calldescr):
  736. return self.do_residual_or_indirect_call(funcbox, argboxes, calldescr)
  737. @arguments("box", "boxes3", "descr")
  738. def _opimpl_residual_call3(self, funcbox, argboxes, calldescr):
  739. return self.do_residual_or_indirect_call(funcbox, argboxes, calldescr)
  740. opimpl_residual_call_r_i = _opimpl_residual_call1
  741. opimpl_residual_call_r_r = _opimpl_residual_call1
  742. opimpl_residual_call_r_v = _opimpl_residual_call1
  743. opimpl_residual_call_ir_i = _opimpl_residual_call2
  744. opimpl_residual_call_ir_r = _opimpl_residual_call2
  745. opimpl_residual_call_ir_v = _opimpl_residual_call2
  746. opimpl_residual_call_irf_i = _opimpl_residual_call3
  747. opimpl_residual_call_irf_r = _opimpl_residual_call3
  748. opimpl_residual_call_irf_f = _opimpl_residual_call3
  749. opimpl_residual_call_irf_v = _opimpl_residual_call3
  750. @arguments("int", "boxes3", "boxes3")
  751. def _opimpl_recursive_call(self, jdindex, greenboxes, redboxes):
  752. targetjitdriver_sd = self.metainterp.staticdata.jitdrivers_sd[jdindex]
  753. allboxes = greenboxes + redboxes
  754. warmrunnerstate = targetjitdriver_sd.warmstate
  755. assembler_call = False
  756. if warmrunnerstate.inlining:
  757. if warmrunnerstate.can_inline_callable(greenboxes):
  758. portal_code = targetjitdriver_sd.mainjitcode
  759. return self.metainterp.perform_call(portal_code, allboxes,
  760. greenkey=greenboxes)
  761. assembler_call = True
  762. # verify that we have all green args, needed to make sure
  763. # that assembler that we call is still correct
  764. self.verify_green_args(targetjitdriver_sd, greenboxes)
  765. #
  766. return self.do_recursive_call(targetjitdriver_sd, allboxes,
  767. assembler_call)
  768. def do_recursive_call(self, targetjitdriver_sd, allboxes,
  769. assembler_call=False):
  770. portal_code = targetjitdriver_sd.mainjitcode
  771. k = targetjitdriver_sd.portal_runner_adr
  772. funcbox = ConstInt(heaptracker.adr2int(k))
  773. return self.do_residual_call(funcbox, allboxes, portal_code.calldescr,
  774. assembler_call=assembler_call,
  775. assembler_call_jd=targetjitdriver_sd)
  776. opimpl_recursive_call_i = _opimpl_recursive_call
  777. opimpl_recursive_call_r = _opimpl_recursive_call
  778. opimpl_recursive_call_f = _opimpl_recursive_call
  779. opimpl_recursive_call_v = _opimpl_recursive_call
  780. @arguments("box")
  781. def opimpl_strlen(self, strbox):
  782. return self.execute(rop.STRLEN, strbox)
  783. @arguments("box")
  784. def opimpl_unicodelen(self, unicodebox):
  785. return self.execute(rop.UNICODELEN, unicodebox)
  786. @arguments("box", "box")
  787. def opimpl_strgetitem(self, strbox, indexbox):
  788. return self.execute(rop.STRGETITEM, strbox, indexbox)
  789. @arguments("box", "box")
  790. def opimpl_unicodegetitem(self, unicodebox, indexbox):
  791. return self.execute(rop.UNICODEGETITEM, unicodebox, indexbox)
  792. @arguments("box", "box", "box")
  793. def opimpl_strsetitem(self, strbox, indexbox, newcharbox):
  794. return self.execute(rop.STRSETITEM, strbox, indexbox, newcharbox)
  795. @arguments("box", "box", "box")
  796. def opimpl_unicodesetitem(self, unicodebox, indexbox, newcharbox):
  797. self.execute(rop.UNICODESETITEM, unicodebox, indexbox, newcharbox)
  798. @arguments("box")
  799. def opimpl_newstr(self, lengthbox):
  800. return self.execute(rop.NEWSTR, lengthbox)
  801. @arguments("box")
  802. def opimpl_newunicode(self, lengthbox):
  803. return self.execute(rop.NEWUNICODE, lengthbox)
  804. @arguments("box", "box", "box", "box", "box")
  805. def opimpl_copystrcontent(self, srcbox, dstbox, srcstartbox, dststartbox, lengthbox):
  806. return self.execute(rop.COPYSTRCONTENT, srcbox, dstbox, srcstartbox, dststartbox, lengthbox)
  807. @arguments("box", "box", "box", "box", "box")
  808. def opimpl_copyunicodecontent(self, srcbox, dstbox, srcstartbox, dststartbox, lengthbox):
  809. return self.execute(rop.COPYUNICODECONTENT, srcbox, dstbox, srcstartbox, dststartbox, lengthbox)
  810. @arguments("box", "orgpc")
  811. def _opimpl_guard_value(self, box, orgpc):
  812. self.implement_guard_value(box, orgpc)
  813. @arguments("box", "box", "descr", "orgpc")
  814. def opimpl_str_guard_value(self, box, funcbox, descr, orgpc):
  815. if isinstance(box, Const):
  816. return box # no promotion needed, already a Const
  817. else:
  818. constbox = box.constbox()
  819. resbox = self.do_residual_call(funcbox, [box, constbox], descr)
  820. promoted_box = resbox.constbox()
  821. # This is GUARD_VALUE because GUARD_TRUE assumes the existance
  822. # of a label when computing resumepc
  823. self.generate_guard(rop.GUARD_VALUE, resbox, [promoted_box],
  824. resumepc=orgpc)
  825. self.metainterp.replace_box(box, constbox)
  826. return constbox
  827. opimpl_int_guard_value = _opimpl_guard_value
  828. opimpl_ref_guard_value = _opimpl_guard_value
  829. opimpl_float_guard_value = _opimpl_guard_value
  830. @arguments("box", "orgpc")
  831. def opimpl_guard_class(self, box, orgpc):
  832. clsbox = self.cls_of_box(box)
  833. if not self.metainterp.heapcache.is_class_known(box):
  834. self.generate_guard(rop.GUARD_CLASS, box, [clsbox], resumepc=orgpc)
  835. self.metainterp.heapcache.class_now_known(box)
  836. return clsbox
  837. @arguments("int", "orgpc")
  838. def opimpl_loop_header(self, jdindex, orgpc):
  839. self.metainterp.seen_loop_header_for_jdindex = jdindex
  840. def verify_green_args(self, jitdriver_sd, varargs):
  841. num_green_args = jitdriver_sd.num_green_args
  842. assert len(varargs) == num_green_args
  843. for i in range(num_green_args):
  844. assert isinstance(varargs[i], Const)
  845. @arguments("int", "boxes3", "jitcode_position", "boxes3", "orgpc")
  846. def opimpl_jit_merge_point(self, jdindex, greenboxes,
  847. jcposition, redboxes, orgpc):
  848. resumedescr = compile.ResumeAtPositionDescr()
  849. self.capture_resumedata(resumedescr, orgpc)
  850. any_operation = len(self.metainterp.history.operations) > 0
  851. jitdriver_sd = self.metainterp.staticdata.jitdrivers_sd[jdindex]
  852. self.verify_green_args(jitdriver_sd, greenboxes)
  853. self.debug_merge_point(jitdriver_sd, jdindex,
  854. self.metainterp.portal_call_depth,
  855. self.metainterp.call_ids[-1],
  856. greenboxes)
  857. if self.metainterp.seen_loop_header_for_jdindex < 0:
  858. if not any_operation:
  859. return
  860. if self.metainterp.portal_call_depth or not self.metainterp.get_procedure_token(greenboxes, True):
  861. if not jitdriver_sd.no_loop_header:
  862. return
  863. # automatically add a loop_header if there is none
  864. self.metainterp.seen_loop_header_for_jdindex = jdindex
  865. #
  866. assert self.metainterp.seen_loop_header_for_jdindex == jdindex, (
  867. "found a loop_header for a JitDriver that does not match "
  868. "the following jit_merge_point's")
  869. self.metainterp.seen_loop_header_for_jdindex = -1
  870. #
  871. if not self.metainterp.portal_call_depth:
  872. assert jitdriver_sd is self.metainterp.jitdriver_sd
  873. # Set self.pc to point to jit_merge_point instead of just after:
  874. # if reached_loop_header() raises SwitchToBlackhole, then the
  875. # pc is still at the jit_merge_point, which is a point that is
  876. # much less expensive to blackhole out of.
  877. saved_pc = self.pc
  878. self.pc = orgpc
  879. self.metainterp.reached_loop_header(greenboxes, redboxes, resumedescr)
  880. self.pc = saved_pc
  881. # no exception, which means that the jit_merge_point did not
  882. # close the loop. We have to put the possibly-modified list
  883. # 'redboxes' back into the registers where it comes from.
  884. put_back_list_of_boxes3(self, jcposition, redboxes)
  885. else:
  886. if jitdriver_sd.warmstate.should_unroll_one_iteration(greenboxes):
  887. if self.unroll_iterations > 0:
  888. self.unroll_iterations -= 1
  889. return
  890. # warning! careful here. We have to return from the current
  891. # frame containing the jit_merge_point, and then use
  892. # do_recursive_call() to follow the recursive call. This is
  893. # needed because do_recursive_call() will write its result
  894. # with make_result_of_lastop(), so the lastop must be right:
  895. # it must be the call to 'self', and not the jit_merge_point
  896. # itself, which has no result at all.
  897. assert len(self.metainterp.framestack) >= 2
  898. try:
  899. self.metainterp.finishframe(None)
  900. except ChangeFrame:
  901. pass
  902. frame = self.metainterp.framestack[-1]
  903. frame.do_recursive_call(jitdriver_sd, greenboxes + redboxes,
  904. assembler_call=True)
  905. raise ChangeFrame
  906. def debug_merge_point(self, jitdriver_sd, jd_index, portal_call_depth, current_call_id, greenkey):
  907. # debugging: produce a DEBUG_MERGE_POINT operation
  908. loc = jitdriver_sd.warmstate.get_location_str(greenkey)
  909. debug_print(loc)
  910. args = [ConstInt(jd_index), ConstInt(portal_call_depth), ConstInt(current_call_id)] + greenkey
  911. self.metainterp.history.record(rop.DEBUG_MERGE_POINT, args, None)
  912. @arguments("box", "label")
  913. def opimpl_goto_if_exception_mismatch(self, vtablebox, next_exc_target):
  914. metainterp = self.metainterp
  915. last_exc_value_box = metainterp.last_exc_value_box
  916. assert last_exc_value_box is not None
  917. assert metainterp.class_of_last_exc_is_const
  918. if not metainterp.cpu.ts.instanceOf(last_exc_value_box, vtablebox):
  919. self.pc = next_exc_target
  920. @arguments("box", "orgpc")
  921. def opimpl_raise(self, exc_value_box, orgpc):
  922. # xxx hack
  923. clsbox = self.cls_of_box(exc_value_box)
  924. self.generate_guard(rop.GUARD_CLASS, exc_value_box, [clsbox],
  925. resumepc=orgpc)
  926. self.metainterp.class_of_last_exc_is_const = True
  927. self.metainterp.last_exc_value_box = exc_value_box
  928. self.metainterp.popframe()
  929. self.metainterp.finishframe_exception()
  930. @arguments()
  931. def opimpl_reraise(self):
  932. assert self.metainterp.last_exc_value_box is not None
  933. self.metainterp.popframe()
  934. self.metainterp.finishframe_exception()
  935. @arguments()
  936. def opimpl_last_exception(self):
  937. # Same comment as in opimpl_goto_if_exception_mismatch().
  938. exc_value_box = self.metainterp.last_exc_value_box
  939. assert exc_value_box is not None
  940. assert self.metainterp.class_of_last_exc_is_const
  941. return self.metainterp.cpu.ts.cls_of_box(exc_value_box)
  942. @arguments()
  943. def opimpl_last_exc_value(self):
  944. exc_value_box = self.metainterp.last_exc_value_box
  945. assert exc_value_box is not None
  946. return exc_value_box
  947. @arguments("box")
  948. def opimpl_debug_fatalerror(self, box):
  949. from pypy.rpython.lltypesystem import rstr, lloperation
  950. msg = box.getref(lltype.Ptr(rstr.STR))
  951. lloperation.llop.debug_fatalerror(msg)
  952. @arguments("box", "box", "box", "box", "box")
  953. def opimpl_jit_debug(self, stringbox, arg1box, arg2box, arg3box, arg4box):
  954. from pypy.rpython.lltypesystem import rstr
  955. from pypy.rpython.annlowlevel import hlstr
  956. msg = stringbox.getref(lltype.Ptr(rstr.STR))
  957. debug_print('jit_debug:', hlstr(msg),
  958. arg1box.getint(),

Large files files are truncated, but you can click here to view the full file