PageRenderTime 42ms CodeModel.GetById 22ms RepoModel.GetById 0ms app.codeStats 1ms

/rpython/jit/metainterp/pyjitpl.py

https://bitbucket.org/pypy/pypy/
Python | 3361 lines | 3074 code | 141 blank | 146 comment | 143 complexity | 97877d59e379df1dde9641100abb1af8 MD5 | raw file
Possible License(s): AGPL-3.0, BSD-3-Clause, Apache-2.0
  1. import sys
  2. import py
  3. from rpython.jit.codewriter import heaptracker, longlong
  4. from rpython.jit.codewriter.effectinfo import EffectInfo
  5. from rpython.jit.codewriter.jitcode import JitCode, SwitchDictDescr
  6. from rpython.jit.metainterp import history, compile, resume, executor, jitexc
  7. from rpython.jit.metainterp.heapcache import HeapCache
  8. from rpython.jit.metainterp.history import (Const, ConstInt, ConstPtr,
  9. ConstFloat, TargetToken, MissingValue, SwitchToBlackhole)
  10. from rpython.jit.metainterp.jitprof import EmptyProfiler
  11. from rpython.jit.metainterp.logger import Logger
  12. from rpython.jit.metainterp.optimizeopt.util import args_dict
  13. from rpython.jit.metainterp.resoperation import rop, OpHelpers, GuardResOp
  14. from rpython.rlib.rjitlog import rjitlog as jl
  15. from rpython.rlib import nonconst, rstack
  16. from rpython.rlib.debug import debug_start, debug_stop, debug_print
  17. from rpython.rlib.debug import have_debug_prints, make_sure_not_resized
  18. from rpython.rlib.jit import Counters
  19. from rpython.rlib.objectmodel import we_are_translated, specialize
  20. from rpython.rlib.unroll import unrolling_iterable
  21. from rpython.rtyper.lltypesystem import lltype, rffi, llmemory
  22. from rpython.rtyper import rclass
  23. from rpython.rlib.objectmodel import compute_unique_id
  24. # ____________________________________________________________
  25. def arguments(*args):
  26. def decorate(func):
  27. func.argtypes = args
  28. return func
  29. return decorate
  30. # ____________________________________________________________
  31. FASTPATHS_SAME_BOXES = {
  32. "ne": "history.CONST_FALSE",
  33. "eq": "history.CONST_TRUE",
  34. "lt": "history.CONST_FALSE",
  35. "le": "history.CONST_TRUE",
  36. "gt": "history.CONST_FALSE",
  37. "ge": "history.CONST_TRUE",
  38. }
  39. class MIFrame(object):
  40. debug = False
  41. def __init__(self, metainterp):
  42. self.metainterp = metainterp
  43. self.registers_i = [None] * 256
  44. self.registers_r = [None] * 256
  45. self.registers_f = [None] * 256
  46. def setup(self, jitcode, greenkey=None):
  47. # if not translated, fill the registers with MissingValue()
  48. if not we_are_translated():
  49. self.registers_i = [MissingValue()] * 256
  50. self.registers_r = [MissingValue()] * 256
  51. self.registers_f = [MissingValue()] * 256
  52. assert isinstance(jitcode, JitCode)
  53. self.jitcode = jitcode
  54. self.bytecode = jitcode.code
  55. # this is not None for frames that are recursive portal calls
  56. self.greenkey = greenkey
  57. # copy the constants in place
  58. self.copy_constants(self.registers_i, jitcode.constants_i, ConstInt)
  59. self.copy_constants(self.registers_r, jitcode.constants_r, ConstPtr)
  60. self.copy_constants(self.registers_f, jitcode.constants_f, ConstFloat)
  61. self._result_argcode = 'v'
  62. # for resume.py operation
  63. self.parent_snapshot = None
  64. # counter for unrolling inlined loops
  65. self.unroll_iterations = 1
  66. @specialize.arg(3)
  67. def copy_constants(self, registers, constants, ConstClass):
  68. """Copy jitcode.constants[0] to registers[255],
  69. jitcode.constants[1] to registers[254],
  70. jitcode.constants[2] to registers[253], etc."""
  71. if nonconst.NonConstant(0): # force the right type
  72. constants[0] = ConstClass.value # (useful for small tests)
  73. i = len(constants) - 1
  74. while i >= 0:
  75. j = 255 - i
  76. assert j >= 0
  77. registers[j] = ConstClass(constants[i])
  78. i -= 1
  79. def cleanup_registers(self):
  80. # To avoid keeping references alive, this cleans up the registers_r.
  81. # It does not clear the references set by copy_constants(), but
  82. # these are all prebuilt constants anyway.
  83. for i in range(self.jitcode.num_regs_r()):
  84. self.registers_r[i] = None
  85. # ------------------------------
  86. # Decoding of the JitCode
  87. @specialize.arg(4)
  88. def prepare_list_of_boxes(self, outvalue, startindex, position, argcode):
  89. assert argcode in 'IRF'
  90. code = self.bytecode
  91. length = ord(code[position])
  92. position += 1
  93. for i in range(length):
  94. index = ord(code[position+i])
  95. if argcode == 'I': reg = self.registers_i[index]
  96. elif argcode == 'R': reg = self.registers_r[index]
  97. elif argcode == 'F': reg = self.registers_f[index]
  98. else: raise AssertionError(argcode)
  99. outvalue[startindex+i] = reg
  100. def _put_back_list_of_boxes(self, outvalue, startindex, position):
  101. code = self.bytecode
  102. length = ord(code[position])
  103. position += 1
  104. for i in range(length):
  105. index = ord(code[position+i])
  106. box = outvalue[startindex+i]
  107. if box.type == history.INT: self.registers_i[index] = box
  108. elif box.type == history.REF: self.registers_r[index] = box
  109. elif box.type == history.FLOAT: self.registers_f[index] = box
  110. else: raise AssertionError(box.type)
  111. def get_current_position_info(self):
  112. return self.jitcode.get_live_vars_info(self.pc)
  113. def get_list_of_active_boxes(self, in_a_call, new_array, encode):
  114. if in_a_call:
  115. # If we are not the topmost frame, self._result_argcode contains
  116. # the type of the result of the call instruction in the bytecode.
  117. # We use it to clear the box that will hold the result: this box
  118. # is not defined yet.
  119. argcode = self._result_argcode
  120. index = ord(self.bytecode[self.pc - 1])
  121. if argcode == 'i': self.registers_i[index] = history.CONST_FALSE
  122. elif argcode == 'r': self.registers_r[index] = history.CONST_NULL
  123. elif argcode == 'f': self.registers_f[index] = history.CONST_FZERO
  124. self._result_argcode = '?' # done
  125. #
  126. info = self.get_current_position_info()
  127. start_i = 0
  128. start_r = start_i + info.get_register_count_i()
  129. start_f = start_r + info.get_register_count_r()
  130. total = start_f + info.get_register_count_f()
  131. # allocate a list of the correct size
  132. env = new_array(total)
  133. make_sure_not_resized(env)
  134. # fill it now
  135. for i in range(info.get_register_count_i()):
  136. index = info.get_register_index_i(i)
  137. env[start_i + i] = encode(self.registers_i[index])
  138. for i in range(info.get_register_count_r()):
  139. index = info.get_register_index_r(i)
  140. env[start_r + i] = encode(self.registers_r[index])
  141. for i in range(info.get_register_count_f()):
  142. index = info.get_register_index_f(i)
  143. env[start_f + i] = encode(self.registers_f[index])
  144. return env
  145. def replace_active_box_in_frame(self, oldbox, newbox):
  146. if oldbox.type == 'i':
  147. count = self.jitcode.num_regs_i()
  148. registers = self.registers_i
  149. elif oldbox.type == 'r':
  150. count = self.jitcode.num_regs_r()
  151. registers = self.registers_r
  152. elif oldbox.type == 'f':
  153. count = self.jitcode.num_regs_f()
  154. registers = self.registers_f
  155. else:
  156. assert 0, oldbox
  157. for i in range(count):
  158. if registers[i] is oldbox:
  159. registers[i] = newbox
  160. if not we_are_translated():
  161. for b in registers[count:]:
  162. assert isinstance(b, (MissingValue, Const))
  163. def make_result_of_lastop(self, resultbox):
  164. got_type = resultbox.type
  165. if not we_are_translated():
  166. typeof = {'i': history.INT,
  167. 'r': history.REF,
  168. 'f': history.FLOAT}
  169. assert typeof[self.jitcode._resulttypes[self.pc]] == got_type
  170. target_index = ord(self.bytecode[self.pc-1])
  171. if got_type == history.INT:
  172. self.registers_i[target_index] = resultbox
  173. elif got_type == history.REF:
  174. #debug_print(' ->',
  175. # llmemory.cast_ptr_to_adr(resultbox.getref_base()))
  176. self.registers_r[target_index] = resultbox
  177. elif got_type == history.FLOAT:
  178. self.registers_f[target_index] = resultbox
  179. else:
  180. raise AssertionError("bad result box type")
  181. # ------------------------------
  182. for _opimpl in ['int_add', 'int_sub', 'int_mul',
  183. 'int_and', 'int_or', 'int_xor', 'int_signext',
  184. 'int_rshift', 'int_lshift', 'uint_rshift',
  185. 'uint_lt', 'uint_le', 'uint_gt', 'uint_ge',
  186. 'float_add', 'float_sub', 'float_mul', 'float_truediv',
  187. 'float_lt', 'float_le', 'float_eq',
  188. 'float_ne', 'float_gt', 'float_ge',
  189. ]:
  190. exec py.code.Source('''
  191. @arguments("box", "box")
  192. def opimpl_%s(self, b1, b2):
  193. return self.execute(rop.%s, b1, b2)
  194. ''' % (_opimpl, _opimpl.upper())).compile()
  195. for _opimpl in ['int_eq', 'int_ne', 'int_lt', 'int_le', 'int_gt', 'int_ge',
  196. 'ptr_eq', 'ptr_ne',
  197. 'instance_ptr_eq', 'instance_ptr_ne']:
  198. exec py.code.Source('''
  199. @arguments("box", "box")
  200. def opimpl_%s(self, b1, b2):
  201. if b1 is b2: # crude fast check
  202. return %s
  203. return self.execute(rop.%s, b1, b2)
  204. ''' % (_opimpl, FASTPATHS_SAME_BOXES[_opimpl.split("_")[-1]], _opimpl.upper())
  205. ).compile()
  206. for (_opimpl, resop) in [
  207. ('int_add_jump_if_ovf', 'INT_ADD_OVF'),
  208. ('int_sub_jump_if_ovf', 'INT_SUB_OVF'),
  209. ('int_mul_jump_if_ovf', 'INT_MUL_OVF')]:
  210. exec py.code.Source('''
  211. @arguments("label", "box", "box", "orgpc")
  212. def opimpl_%s(self, lbl, b1, b2, orgpc):
  213. self.metainterp.ovf_flag = False
  214. resbox = self.execute(rop.%s, b1, b2)
  215. if not isinstance(resbox, Const):
  216. return self.handle_possible_overflow_error(lbl, orgpc,
  217. resbox)
  218. elif self.metainterp.ovf_flag:
  219. self.pc = lbl
  220. return None # but don't emit GUARD_OVERFLOW
  221. return resbox
  222. ''' % (_opimpl, resop)).compile()
  223. for _opimpl in ['int_is_true', 'int_is_zero', 'int_neg', 'int_invert',
  224. 'cast_float_to_int', 'cast_int_to_float',
  225. 'cast_float_to_singlefloat', 'cast_singlefloat_to_float',
  226. 'float_neg', 'float_abs',
  227. 'cast_ptr_to_int', 'cast_int_to_ptr',
  228. 'convert_float_bytes_to_longlong',
  229. 'convert_longlong_bytes_to_float', 'int_force_ge_zero',
  230. ]:
  231. exec py.code.Source('''
  232. @arguments("box")
  233. def opimpl_%s(self, b):
  234. return self.execute(rop.%s, b)
  235. ''' % (_opimpl, _opimpl.upper())).compile()
  236. @arguments("box")
  237. def opimpl_int_same_as(self, box):
  238. # for tests only: emits a same_as, forcing the result to be in a Box
  239. resbox = self.metainterp._record_helper_nonpure_varargs(
  240. rop.SAME_AS_I, box.getint(), None, [box])
  241. return resbox
  242. @arguments("box")
  243. def opimpl_ptr_nonzero(self, box):
  244. return self.execute(rop.PTR_NE, box, history.CONST_NULL)
  245. @arguments("box")
  246. def opimpl_ptr_iszero(self, box):
  247. return self.execute(rop.PTR_EQ, box, history.CONST_NULL)
  248. @arguments("box", "box")
  249. def opimpl_record_exact_class(self, box, clsbox):
  250. from rpython.rtyper.lltypesystem import llmemory
  251. if self.metainterp.heapcache.is_class_known(box):
  252. return
  253. adr = clsbox.getaddr()
  254. self.execute(rop.RECORD_EXACT_CLASS, box, clsbox)
  255. self.metainterp.heapcache.class_now_known(box)
  256. @arguments("box")
  257. def _opimpl_any_return(self, box):
  258. self.metainterp.finishframe(box)
  259. opimpl_int_return = _opimpl_any_return
  260. opimpl_ref_return = _opimpl_any_return
  261. opimpl_float_return = _opimpl_any_return
  262. @arguments()
  263. def opimpl_void_return(self):
  264. self.metainterp.finishframe(None)
  265. @arguments("box")
  266. def _opimpl_any_copy(self, box):
  267. return box
  268. opimpl_int_copy = _opimpl_any_copy
  269. opimpl_ref_copy = _opimpl_any_copy
  270. opimpl_float_copy = _opimpl_any_copy
  271. @arguments("box")
  272. def _opimpl_any_push(self, box):
  273. self.pushed_box = box
  274. opimpl_int_push = _opimpl_any_push
  275. opimpl_ref_push = _opimpl_any_push
  276. opimpl_float_push = _opimpl_any_push
  277. @arguments()
  278. def _opimpl_any_pop(self):
  279. box = self.pushed_box
  280. self.pushed_box = None
  281. return box
  282. opimpl_int_pop = _opimpl_any_pop
  283. opimpl_ref_pop = _opimpl_any_pop
  284. opimpl_float_pop = _opimpl_any_pop
  285. @arguments("label")
  286. def opimpl_catch_exception(self, target):
  287. """This is a no-op when run normally. We can check that
  288. last_exc_value is a null ptr; it should have been set to None
  289. by the previous instruction. If the previous instruction
  290. raised instead, finishframe_exception() should have been
  291. called and we would not be there."""
  292. assert not self.metainterp.last_exc_value
  293. @arguments("label")
  294. def opimpl_goto(self, target):
  295. self.pc = target
  296. @arguments("box", "label", "orgpc")
  297. def opimpl_goto_if_not(self, box, target, orgpc):
  298. switchcase = box.getint()
  299. if switchcase:
  300. opnum = rop.GUARD_TRUE
  301. else:
  302. opnum = rop.GUARD_FALSE
  303. self.metainterp.generate_guard(opnum, box, resumepc=orgpc)
  304. if not switchcase:
  305. self.pc = target
  306. @arguments("box", "label", "orgpc")
  307. def opimpl_goto_if_not_int_is_true(self, box, target, orgpc):
  308. condbox = self.execute(rop.INT_IS_TRUE, box)
  309. self.opimpl_goto_if_not(condbox, target, orgpc)
  310. @arguments("box", "label", "orgpc")
  311. def opimpl_goto_if_not_int_is_zero(self, box, target, orgpc):
  312. condbox = self.execute(rop.INT_IS_ZERO, box)
  313. self.opimpl_goto_if_not(condbox, target, orgpc)
  314. for _opimpl in ['int_lt', 'int_le', 'int_eq', 'int_ne', 'int_gt', 'int_ge',
  315. 'ptr_eq', 'ptr_ne', 'float_lt', 'float_le', 'float_eq',
  316. 'float_ne', 'float_gt', 'float_ge']:
  317. exec py.code.Source('''
  318. @arguments("box", "box", "label", "orgpc")
  319. def opimpl_goto_if_not_%s(self, b1, b2, target, orgpc):
  320. if %s and b1 is b2:
  321. condbox = %s
  322. else:
  323. condbox = self.execute(rop.%s, b1, b2)
  324. self.opimpl_goto_if_not(condbox, target, orgpc)
  325. ''' % (_opimpl, not _opimpl.startswith('float_'),
  326. FASTPATHS_SAME_BOXES[_opimpl.split("_")[-1]], _opimpl.upper())
  327. ).compile()
  328. def _establish_nullity(self, box, orgpc):
  329. heapcache = self.metainterp.heapcache
  330. value = box.nonnull()
  331. if heapcache.is_nullity_known(box):
  332. return value
  333. if value:
  334. if not self.metainterp.heapcache.is_class_known(box):
  335. self.metainterp.generate_guard(rop.GUARD_NONNULL, box,
  336. resumepc=orgpc)
  337. else:
  338. if not isinstance(box, Const):
  339. self.metainterp.generate_guard(rop.GUARD_ISNULL, box,
  340. resumepc=orgpc)
  341. promoted_box = executor.constant_from_op(box)
  342. self.metainterp.replace_box(box, promoted_box)
  343. heapcache.nullity_now_known(box)
  344. return value
  345. @arguments("box", "label", "orgpc")
  346. def opimpl_goto_if_not_ptr_nonzero(self, box, target, orgpc):
  347. if not self._establish_nullity(box, orgpc):
  348. self.pc = target
  349. @arguments("box", "label", "orgpc")
  350. def opimpl_goto_if_not_ptr_iszero(self, box, target, orgpc):
  351. if self._establish_nullity(box, orgpc):
  352. self.pc = target
  353. @arguments("box", "box", "box")
  354. def opimpl_int_between(self, b1, b2, b3):
  355. b5 = self.execute(rop.INT_SUB, b3, b1)
  356. if isinstance(b5, ConstInt) and b5.getint() == 1:
  357. # the common case of int_between(a, b, a+1) turns into just INT_EQ
  358. return self.execute(rop.INT_EQ, b2, b1)
  359. else:
  360. b4 = self.execute(rop.INT_SUB, b2, b1)
  361. return self.execute(rop.UINT_LT, b4, b5)
  362. @arguments("box", "descr", "orgpc")
  363. def opimpl_switch(self, valuebox, switchdict, orgpc):
  364. search_value = valuebox.getint()
  365. assert isinstance(switchdict, SwitchDictDescr)
  366. try:
  367. target = switchdict.dict[search_value]
  368. except KeyError:
  369. # None of the cases match. Fall back to generating a chain
  370. # of 'int_eq'.
  371. # xxx as a minor optimization, if that's a bridge, then we would
  372. # not need the cases that we already tested (and failed) with
  373. # 'guard_value'. How to do it is not very clear though.
  374. for const1 in switchdict.const_keys_in_order:
  375. box = self.execute(rop.INT_EQ, valuebox, const1)
  376. assert box.getint() == 0
  377. target = switchdict.dict[const1.getint()]
  378. self.metainterp.generate_guard(rop.GUARD_FALSE, box,
  379. resumepc=orgpc)
  380. else:
  381. # found one of the cases
  382. self.implement_guard_value(valuebox, orgpc)
  383. self.pc = target
  384. @arguments()
  385. def opimpl_unreachable(self):
  386. raise AssertionError("unreachable")
  387. @arguments("descr")
  388. def opimpl_new(self, sizedescr):
  389. return self.metainterp.execute_new(sizedescr)
  390. @arguments("descr")
  391. def opimpl_new_with_vtable(self, sizedescr):
  392. return self.metainterp.execute_new_with_vtable(descr=sizedescr)
  393. @arguments("box", "descr")
  394. def opimpl_new_array(self, lengthbox, itemsizedescr):
  395. return self.metainterp.execute_new_array(itemsizedescr, lengthbox)
  396. @arguments("box", "descr")
  397. def opimpl_new_array_clear(self, lengthbox, itemsizedescr):
  398. return self.metainterp.execute_new_array_clear(itemsizedescr, lengthbox)
  399. @specialize.arg(1)
  400. def _do_getarrayitem_gc_any(self, op, arraybox, indexbox, arraydescr):
  401. tobox = self.metainterp.heapcache.getarrayitem(
  402. arraybox, indexbox, arraydescr)
  403. if tobox:
  404. # sanity check: see whether the current array value
  405. # corresponds to what the cache thinks the value is
  406. resvalue = executor.execute(self.metainterp.cpu, self.metainterp,
  407. op, arraydescr, arraybox, indexbox)
  408. if op == 'i':
  409. assert resvalue == tobox.getint()
  410. elif op == 'r':
  411. assert resvalue == tobox.getref_base()
  412. elif op == 'f':
  413. assert resvalue == tobox.getfloat()
  414. return tobox
  415. resop = self.execute_with_descr(op, arraydescr, arraybox, indexbox)
  416. self.metainterp.heapcache.getarrayitem_now_known(
  417. arraybox, indexbox, resop, arraydescr)
  418. return resop
  419. @arguments("box", "box", "descr")
  420. def opimpl_getarrayitem_gc_i(self, arraybox, indexbox, arraydescr):
  421. return self._do_getarrayitem_gc_any(rop.GETARRAYITEM_GC_I, arraybox,
  422. indexbox, arraydescr)
  423. @arguments("box", "box", "descr")
  424. def opimpl_getarrayitem_gc_r(self, arraybox, indexbox, arraydescr):
  425. return self._do_getarrayitem_gc_any(rop.GETARRAYITEM_GC_R, arraybox,
  426. indexbox, arraydescr)
  427. @arguments("box", "box", "descr")
  428. def opimpl_getarrayitem_gc_f(self, arraybox, indexbox, arraydescr):
  429. return self._do_getarrayitem_gc_any(rop.GETARRAYITEM_GC_F, arraybox,
  430. indexbox, arraydescr)
  431. @arguments("box", "box", "descr")
  432. def opimpl_getarrayitem_raw_i(self, arraybox, indexbox, arraydescr):
  433. return self.execute_with_descr(rop.GETARRAYITEM_RAW_I,
  434. arraydescr, arraybox, indexbox)
  435. @arguments("box", "box", "descr")
  436. def opimpl_getarrayitem_raw_f(self, arraybox, indexbox, arraydescr):
  437. return self.execute_with_descr(rop.GETARRAYITEM_RAW_F,
  438. arraydescr, arraybox, indexbox)
  439. @arguments("box", "box", "descr")
  440. def opimpl_getarrayitem_gc_i_pure(self, arraybox, indexbox, arraydescr):
  441. if isinstance(arraybox, ConstPtr) and isinstance(indexbox, ConstInt):
  442. # if the arguments are directly constants, bypass the heapcache
  443. # completely
  444. val = executor.execute(self.metainterp.cpu, self.metainterp,
  445. rop.GETARRAYITEM_GC_PURE_I, arraydescr,
  446. arraybox, indexbox)
  447. return executor.wrap_constant(val)
  448. return self._do_getarrayitem_gc_any(rop.GETARRAYITEM_GC_PURE_I,
  449. arraybox, indexbox, arraydescr)
  450. @arguments("box", "box", "descr")
  451. def opimpl_getarrayitem_gc_f_pure(self, arraybox, indexbox, arraydescr):
  452. if isinstance(arraybox, ConstPtr) and isinstance(indexbox, ConstInt):
  453. # if the arguments are directly constants, bypass the heapcache
  454. # completely
  455. resval = executor.execute(self.metainterp.cpu, self.metainterp,
  456. rop.GETARRAYITEM_GC_PURE_F, arraydescr,
  457. arraybox, indexbox)
  458. return executor.wrap_constant(resval)
  459. return self._do_getarrayitem_gc_any(rop.GETARRAYITEM_GC_PURE_F,
  460. arraybox, indexbox, arraydescr)
  461. @arguments("box", "box", "descr")
  462. def opimpl_getarrayitem_gc_r_pure(self, arraybox, indexbox, arraydescr):
  463. if isinstance(arraybox, ConstPtr) and isinstance(indexbox, ConstInt):
  464. # if the arguments are directly constants, bypass the heapcache
  465. # completely
  466. val = executor.execute(self.metainterp.cpu, self.metainterp,
  467. rop.GETARRAYITEM_GC_PURE_R, arraydescr,
  468. arraybox, indexbox)
  469. return executor.wrap_constant(val)
  470. return self._do_getarrayitem_gc_any(rop.GETARRAYITEM_GC_PURE_R,
  471. arraybox, indexbox, arraydescr)
  472. @arguments("box", "box", "box", "descr")
  473. def _opimpl_setarrayitem_gc_any(self, arraybox, indexbox, itembox,
  474. arraydescr):
  475. self.metainterp.execute_setarrayitem_gc(arraydescr, arraybox,
  476. indexbox, itembox)
  477. opimpl_setarrayitem_gc_i = _opimpl_setarrayitem_gc_any
  478. opimpl_setarrayitem_gc_r = _opimpl_setarrayitem_gc_any
  479. opimpl_setarrayitem_gc_f = _opimpl_setarrayitem_gc_any
  480. @arguments("box", "box", "box", "descr")
  481. def _opimpl_setarrayitem_raw_any(self, arraybox, indexbox, itembox,
  482. arraydescr):
  483. self.execute_with_descr(rop.SETARRAYITEM_RAW, arraydescr, arraybox,
  484. indexbox, itembox)
  485. opimpl_setarrayitem_raw_i = _opimpl_setarrayitem_raw_any
  486. opimpl_setarrayitem_raw_f = _opimpl_setarrayitem_raw_any
  487. @arguments("box", "descr")
  488. def opimpl_arraylen_gc(self, arraybox, arraydescr):
  489. lengthbox = self.metainterp.heapcache.arraylen(arraybox)
  490. if lengthbox is None:
  491. lengthbox = self.execute_with_descr(
  492. rop.ARRAYLEN_GC, arraydescr, arraybox)
  493. self.metainterp.heapcache.arraylen_now_known(arraybox, lengthbox)
  494. return lengthbox
  495. @arguments("box", "box", "descr", "orgpc")
  496. def opimpl_check_neg_index(self, arraybox, indexbox, arraydescr, orgpc):
  497. negbox = self.metainterp.execute_and_record(
  498. rop.INT_LT, None, indexbox, history.CONST_FALSE)
  499. negbox = self.implement_guard_value(negbox, orgpc)
  500. if negbox.getint():
  501. # the index is < 0; add the array length to it
  502. lengthbox = self.opimpl_arraylen_gc(arraybox, arraydescr)
  503. indexbox = self.metainterp.execute_and_record(
  504. rop.INT_ADD, None, indexbox, lengthbox)
  505. return indexbox
  506. @arguments("box", "descr", "descr", "descr", "descr")
  507. def opimpl_newlist(self, sizebox, structdescr, lengthdescr,
  508. itemsdescr, arraydescr):
  509. sbox = self.opimpl_new(structdescr)
  510. self._opimpl_setfield_gc_any(sbox, sizebox, lengthdescr)
  511. if (arraydescr.is_array_of_structs() or
  512. arraydescr.is_array_of_pointers()):
  513. abox = self.opimpl_new_array_clear(sizebox, arraydescr)
  514. else:
  515. abox = self.opimpl_new_array(sizebox, arraydescr)
  516. self._opimpl_setfield_gc_any(sbox, abox, itemsdescr)
  517. return sbox
  518. @arguments("box", "descr", "descr", "descr", "descr")
  519. def opimpl_newlist_clear(self, sizebox, structdescr, lengthdescr,
  520. itemsdescr, arraydescr):
  521. sbox = self.opimpl_new(structdescr)
  522. self._opimpl_setfield_gc_any(sbox, sizebox, lengthdescr)
  523. abox = self.opimpl_new_array_clear(sizebox, arraydescr)
  524. self._opimpl_setfield_gc_any(sbox, abox, itemsdescr)
  525. return sbox
  526. @arguments("box", "descr", "descr", "descr", "descr")
  527. def opimpl_newlist_hint(self, sizehintbox, structdescr, lengthdescr,
  528. itemsdescr, arraydescr):
  529. sbox = self.opimpl_new(structdescr)
  530. self._opimpl_setfield_gc_any(sbox, history.CONST_FALSE, lengthdescr)
  531. if (arraydescr.is_array_of_structs() or
  532. arraydescr.is_array_of_pointers()):
  533. abox = self.opimpl_new_array_clear(sizehintbox, arraydescr)
  534. else:
  535. abox = self.opimpl_new_array(sizehintbox, arraydescr)
  536. self._opimpl_setfield_gc_any(sbox, abox, itemsdescr)
  537. return sbox
  538. @arguments("box", "box", "descr", "descr")
  539. def opimpl_getlistitem_gc_i(self, listbox, indexbox,
  540. itemsdescr, arraydescr):
  541. arraybox = self.opimpl_getfield_gc_r(listbox, itemsdescr)
  542. return self.opimpl_getarrayitem_gc_i(arraybox, indexbox, arraydescr)
  543. @arguments("box", "box", "descr", "descr")
  544. def opimpl_getlistitem_gc_r(self, listbox, indexbox,
  545. itemsdescr, arraydescr):
  546. arraybox = self.opimpl_getfield_gc_r(listbox, itemsdescr)
  547. return self.opimpl_getarrayitem_gc_r(arraybox, indexbox, arraydescr)
  548. @arguments("box", "box", "descr", "descr")
  549. def opimpl_getlistitem_gc_f(self, listbox, indexbox,
  550. itemsdescr, arraydescr):
  551. arraybox = self.opimpl_getfield_gc_r(listbox, itemsdescr)
  552. return self.opimpl_getarrayitem_gc_f(arraybox, indexbox, arraydescr)
  553. @arguments("box", "box", "box", "descr", "descr")
  554. def _opimpl_setlistitem_gc_any(self, listbox, indexbox, valuebox,
  555. itemsdescr, arraydescr):
  556. arraybox = self.opimpl_getfield_gc_r(listbox, itemsdescr)
  557. self._opimpl_setarrayitem_gc_any(arraybox, indexbox, valuebox,
  558. arraydescr)
  559. opimpl_setlistitem_gc_i = _opimpl_setlistitem_gc_any
  560. opimpl_setlistitem_gc_r = _opimpl_setlistitem_gc_any
  561. opimpl_setlistitem_gc_f = _opimpl_setlistitem_gc_any
  562. @arguments("box", "box", "descr", "orgpc")
  563. def opimpl_check_resizable_neg_index(self, listbox, indexbox,
  564. lengthdescr, orgpc):
  565. negbox = self.metainterp.execute_and_record(
  566. rop.INT_LT, None, indexbox, history.CONST_FALSE)
  567. negbox = self.implement_guard_value(negbox, orgpc)
  568. if negbox.getint():
  569. # the index is < 0; add the array length to it
  570. lenbox = self.metainterp.execute_and_record(
  571. rop.GETFIELD_GC, lengthdescr, listbox)
  572. indexbox = self.metainterp.execute_and_record(
  573. rop.INT_ADD, None, indexbox, lenbox)
  574. return indexbox
  575. @arguments("box", "descr")
  576. def opimpl_getfield_gc_i(self, box, fielddescr):
  577. if fielddescr.is_always_pure() and isinstance(box, ConstPtr):
  578. # if 'box' is directly a ConstPtr, bypass the heapcache completely
  579. resbox = executor.execute(self.metainterp.cpu, self.metainterp,
  580. rop.GETFIELD_GC_I, fielddescr, box)
  581. return ConstInt(resbox)
  582. return self._opimpl_getfield_gc_any_pureornot(
  583. rop.GETFIELD_GC_I, box, fielddescr, 'i')
  584. @arguments("box", "descr")
  585. def opimpl_getfield_gc_f(self, box, fielddescr):
  586. if fielddescr.is_always_pure() and isinstance(box, ConstPtr):
  587. # if 'box' is directly a ConstPtr, bypass the heapcache completely
  588. resvalue = executor.execute(self.metainterp.cpu, self.metainterp,
  589. rop.GETFIELD_GC_F, fielddescr, box)
  590. return ConstFloat(resvalue)
  591. return self._opimpl_getfield_gc_any_pureornot(
  592. rop.GETFIELD_GC_F, box, fielddescr, 'f')
  593. @arguments("box", "descr")
  594. def opimpl_getfield_gc_r(self, box, fielddescr):
  595. if fielddescr.is_always_pure() and isinstance(box, ConstPtr):
  596. # if 'box' is directly a ConstPtr, bypass the heapcache completely
  597. val = executor.execute(self.metainterp.cpu, self.metainterp,
  598. rop.GETFIELD_GC_R, fielddescr, box)
  599. return ConstPtr(val)
  600. return self._opimpl_getfield_gc_any_pureornot(
  601. rop.GETFIELD_GC_R, box, fielddescr, 'r')
  602. opimpl_getfield_gc_i_pure = opimpl_getfield_gc_i
  603. opimpl_getfield_gc_r_pure = opimpl_getfield_gc_r
  604. opimpl_getfield_gc_f_pure = opimpl_getfield_gc_f
  605. @arguments("box", "box", "descr")
  606. def opimpl_getinteriorfield_gc_i(self, array, index, descr):
  607. return self.execute_with_descr(rop.GETINTERIORFIELD_GC_I, descr,
  608. array, index)
  609. @arguments("box", "box", "descr")
  610. def opimpl_getinteriorfield_gc_r(self, array, index, descr):
  611. return self.execute_with_descr(rop.GETINTERIORFIELD_GC_R, descr,
  612. array, index)
  613. @arguments("box", "box", "descr")
  614. def opimpl_getinteriorfield_gc_f(self, array, index, descr):
  615. return self.execute_with_descr(rop.GETINTERIORFIELD_GC_F, descr,
  616. array, index)
  617. @specialize.arg(1, 4)
  618. def _opimpl_getfield_gc_any_pureornot(self, opnum, box, fielddescr, type):
  619. upd = self.metainterp.heapcache.get_field_updater(box, fielddescr)
  620. if upd.currfieldbox is not None:
  621. # sanity check: see whether the current struct value
  622. # corresponds to what the cache thinks the value is
  623. resvalue = executor.execute(self.metainterp.cpu, self.metainterp,
  624. opnum, fielddescr, box)
  625. if type == 'i':
  626. assert resvalue == upd.currfieldbox.getint()
  627. elif type == 'r':
  628. assert resvalue == upd.currfieldbox.getref_base()
  629. else:
  630. assert type == 'f'
  631. # make the comparison more robust again NaNs
  632. # see ConstFloat.same_constant
  633. assert ConstFloat(resvalue).same_constant(
  634. upd.currfieldbox.constbox())
  635. return upd.currfieldbox
  636. resbox = self.execute_with_descr(opnum, fielddescr, box)
  637. upd.getfield_now_known(resbox)
  638. return resbox
  639. @arguments("box", "descr", "orgpc")
  640. def _opimpl_getfield_gc_greenfield_any(self, box, fielddescr, pc):
  641. ginfo = self.metainterp.jitdriver_sd.greenfield_info
  642. opnum = OpHelpers.getfield_for_descr(fielddescr)
  643. if (ginfo is not None and fielddescr in ginfo.green_field_descrs
  644. and not self._nonstandard_virtualizable(pc, box, fielddescr)):
  645. # fetch the result, but consider it as a Const box and don't
  646. # record any operation
  647. return executor.execute_nonspec_const(self.metainterp.cpu,
  648. self.metainterp, opnum, [box], fielddescr)
  649. # fall-back
  650. if fielddescr.is_pointer_field():
  651. return self.execute_with_descr(rop.GETFIELD_GC_R, fielddescr, box)
  652. elif fielddescr.is_float_field():
  653. return self.execute_with_descr(rop.GETFIELD_GC_F, fielddescr, box)
  654. else:
  655. return self.execute_with_descr(rop.GETFIELD_GC_I, fielddescr, box)
  656. opimpl_getfield_gc_i_greenfield = _opimpl_getfield_gc_greenfield_any
  657. opimpl_getfield_gc_r_greenfield = _opimpl_getfield_gc_greenfield_any
  658. opimpl_getfield_gc_f_greenfield = _opimpl_getfield_gc_greenfield_any
  659. @arguments("box", "box", "descr")
  660. def _opimpl_setfield_gc_any(self, box, valuebox, fielddescr):
  661. upd = self.metainterp.heapcache.get_field_updater(box, fielddescr)
  662. if upd.currfieldbox is valuebox:
  663. return
  664. self.metainterp.execute_and_record(rop.SETFIELD_GC, fielddescr, box, valuebox)
  665. upd.setfield(valuebox)
  666. # The following logic is disabled because buggy. It is supposed
  667. # to be: not(we're writing null into a freshly allocated object)
  668. # but the bug is that is_unescaped() can be True even after the
  669. # field cache is cleared --- see test_ajit:test_unescaped_write_zero
  670. #
  671. # if tobox is not None or not self.metainterp.heapcache.is_unescaped(box) or not isinstance(valuebox, Const) or valuebox.nonnull():
  672. # self.execute_with_descr(rop.SETFIELD_GC, fielddescr, box, valuebox)
  673. # self.metainterp.heapcache.setfield(box, valuebox, fielddescr)
  674. opimpl_setfield_gc_i = _opimpl_setfield_gc_any
  675. opimpl_setfield_gc_r = _opimpl_setfield_gc_any
  676. opimpl_setfield_gc_f = _opimpl_setfield_gc_any
  677. @arguments("box", "box", "box", "descr")
  678. def _opimpl_setinteriorfield_gc_any(self, array, index, value, descr):
  679. self.metainterp.execute_setinteriorfield_gc(descr, array, index, value)
  680. opimpl_setinteriorfield_gc_i = _opimpl_setinteriorfield_gc_any
  681. opimpl_setinteriorfield_gc_f = _opimpl_setinteriorfield_gc_any
  682. opimpl_setinteriorfield_gc_r = _opimpl_setinteriorfield_gc_any
  683. @arguments("box", "descr")
  684. def opimpl_getfield_raw_i(self, box, fielddescr):
  685. return self.execute_with_descr(rop.GETFIELD_RAW_I, fielddescr, box)
  686. @arguments("box", "descr")
  687. def opimpl_getfield_raw_r(self, box, fielddescr): # for pure only
  688. return self.execute_with_descr(rop.GETFIELD_RAW_R, fielddescr, box)
  689. @arguments("box", "descr")
  690. def opimpl_getfield_raw_f(self, box, fielddescr):
  691. return self.execute_with_descr(rop.GETFIELD_RAW_F, fielddescr, box)
  692. @arguments("box", "box", "descr")
  693. def _opimpl_setfield_raw_any(self, box, valuebox, fielddescr):
  694. self.execute_with_descr(rop.SETFIELD_RAW, fielddescr, box, valuebox)
  695. opimpl_setfield_raw_i = _opimpl_setfield_raw_any
  696. opimpl_setfield_raw_f = _opimpl_setfield_raw_any
  697. @arguments("box", "box", "box", "descr")
  698. def _opimpl_raw_store(self, addrbox, offsetbox, valuebox, arraydescr):
  699. self.metainterp.execute_raw_store(arraydescr,
  700. addrbox, offsetbox, valuebox)
  701. opimpl_raw_store_i = _opimpl_raw_store
  702. opimpl_raw_store_f = _opimpl_raw_store
  703. @arguments("box", "box", "descr")
  704. def opimpl_raw_load_i(self, addrbox, offsetbox, arraydescr):
  705. return self.execute_with_descr(rop.RAW_LOAD_I, arraydescr,
  706. addrbox, offsetbox)
  707. @arguments("box", "box", "descr")
  708. def opimpl_raw_load_f(self, addrbox, offsetbox, arraydescr):
  709. return self.execute_with_descr(rop.RAW_LOAD_F, arraydescr,
  710. addrbox, offsetbox)
  711. def _remove_symbolics(self, c):
  712. if not we_are_translated():
  713. from rpython.rtyper.lltypesystem import ll2ctypes
  714. assert isinstance(c, ConstInt)
  715. c = ConstInt(ll2ctypes.lltype2ctypes(c.value))
  716. return c
  717. @arguments("box", "box", "box", "box", "box")
  718. def opimpl_gc_load_indexed_i(self, addrbox, indexbox,
  719. scalebox, baseofsbox, bytesbox):
  720. return self.execute(rop.GC_LOAD_INDEXED_I, addrbox, indexbox,
  721. self._remove_symbolics(scalebox),
  722. self._remove_symbolics(baseofsbox), bytesbox)
  723. @arguments("box", "box", "box", "box", "box")
  724. def opimpl_gc_load_indexed_f(self, addrbox, indexbox,
  725. scalebox, baseofsbox, bytesbox):
  726. return self.execute(rop.GC_LOAD_INDEXED_F, addrbox, indexbox,
  727. self._remove_symbolics(scalebox),
  728. self._remove_symbolics(baseofsbox), bytesbox)
  729. @arguments("box")
  730. def opimpl_hint_force_virtualizable(self, box):
  731. self.metainterp.gen_store_back_in_vable(box)
  732. @arguments("box", "descr", "descr", "orgpc")
  733. def opimpl_record_quasiimmut_field(self, box, fielddescr,
  734. mutatefielddescr, orgpc):
  735. from rpython.jit.metainterp.quasiimmut import QuasiImmutDescr
  736. cpu = self.metainterp.cpu
  737. if self.metainterp.heapcache.is_quasi_immut_known(fielddescr, box):
  738. return
  739. descr = QuasiImmutDescr(cpu, box.getref_base(), fielddescr,
  740. mutatefielddescr)
  741. self.metainterp.heapcache.quasi_immut_now_known(fielddescr, box)
  742. self.metainterp.history.record(rop.QUASIIMMUT_FIELD, [box],
  743. None, descr=descr)
  744. self.metainterp.generate_guard(rop.GUARD_NOT_INVALIDATED,
  745. resumepc=orgpc)
  746. @arguments("box", "descr", "orgpc")
  747. def opimpl_jit_force_quasi_immutable(self, box, mutatefielddescr, orgpc):
  748. # During tracing, a 'jit_force_quasi_immutable' usually turns into
  749. # the operations that check that the content of 'mutate_xxx' is null.
  750. # If it is actually not null already now, then we abort tracing.
  751. # The idea is that if we use 'jit_force_quasi_immutable' on a freshly
  752. # allocated object, then the GETFIELD_GC will know that the answer is
  753. # null, and the guard will be removed. So the fact that the field is
  754. # quasi-immutable will have no effect, and instead it will work as a
  755. # regular, probably virtual, structure.
  756. if mutatefielddescr.is_pointer_field():
  757. mutatebox = self.execute_with_descr(rop.GETFIELD_GC_R,
  758. mutatefielddescr, box)
  759. elif mutatefielddescr.is_float_field():
  760. mutatebox = self.execute_with_descr(rop.GETFIELD_GC_R,
  761. mutatefielddescr, box)
  762. else:
  763. mutatebox = self.execute_with_descr(rop.GETFIELD_GC_I,
  764. mutatefielddescr, box)
  765. if mutatebox.nonnull():
  766. from rpython.jit.metainterp.quasiimmut import do_force_quasi_immutable
  767. do_force_quasi_immutable(self.metainterp.cpu, box.getref_base(),
  768. mutatefielddescr)
  769. raise SwitchToBlackhole(Counters.ABORT_FORCE_QUASIIMMUT)
  770. self.metainterp.generate_guard(rop.GUARD_ISNULL, mutatebox,
  771. resumepc=orgpc)
  772. def _nonstandard_virtualizable(self, pc, box, fielddescr):
  773. # returns True if 'box' is actually not the "standard" virtualizable
  774. # that is stored in metainterp.virtualizable_boxes[-1]
  775. if self.metainterp.heapcache.is_nonstandard_virtualizable(box):
  776. return True
  777. if box is self.metainterp.forced_virtualizable:
  778. self.metainterp.forced_virtualizable = None
  779. if (self.metainterp.jitdriver_sd.virtualizable_info is not None or
  780. self.metainterp.jitdriver_sd.greenfield_info is not None):
  781. standard_box = self.metainterp.virtualizable_boxes[-1]
  782. if standard_box is box:
  783. return False
  784. vinfo = self.metainterp.jitdriver_sd.virtualizable_info
  785. if vinfo is fielddescr.get_vinfo():
  786. eqbox = self.metainterp.execute_and_record(rop.PTR_EQ, None,
  787. box, standard_box)
  788. eqbox = self.implement_guard_value(eqbox, pc)
  789. isstandard = eqbox.getint()
  790. if isstandard:
  791. if box.type == 'r':
  792. self.metainterp.replace_box(box, standard_box)
  793. return False
  794. if not self.metainterp.heapcache.is_unescaped(box):
  795. self.emit_force_virtualizable(fielddescr, box)
  796. self.metainterp.heapcache.nonstandard_virtualizables_now_known(box)
  797. return True
  798. def emit_force_virtualizable(self, fielddescr, box):
  799. vinfo = fielddescr.get_vinfo()
  800. assert vinfo is not None
  801. token_descr = vinfo.vable_token_descr
  802. mi = self.metainterp
  803. tokenbox = mi.execute_and_record(rop.GETFIELD_GC_R, token_descr, box)
  804. condbox = mi.execute_and_record(rop.PTR_NE, None, tokenbox,
  805. history.CONST_NULL)
  806. funcbox = ConstInt(rffi.cast(lltype.Signed, vinfo.clear_vable_ptr))
  807. calldescr = vinfo.clear_vable_descr
  808. self.execute_varargs(rop.COND_CALL, [condbox, funcbox, box],
  809. calldescr, False, False)
  810. def _get_virtualizable_field_index(self, fielddescr):
  811. # Get the index of a fielddescr. Must only be called for
  812. # the "standard" virtualizable.
  813. vinfo = self.metainterp.jitdriver_sd.virtualizable_info
  814. return vinfo.static_field_by_descrs[fielddescr]
  815. @arguments("box", "descr", "orgpc")
  816. def opimpl_getfield_vable_i(self, box, fielddescr, pc):
  817. if self._nonstandard_virtualizable(pc, box, fielddescr):
  818. return self.opimpl_getfield_gc_i(box, fielddescr)
  819. self.metainterp.check_synchronized_virtualizable()
  820. index = self._get_virtualizable_field_index(fielddescr)
  821. return self.metainterp.virtualizable_boxes[index]
  822. @arguments("box", "descr", "orgpc")
  823. def opimpl_getfield_vable_r(self, box, fielddescr, pc):
  824. if self._nonstandard_virtualizable(pc, box, fielddescr):
  825. return self.opimpl_getfield_gc_r(box, fielddescr)
  826. self.metainterp.check_synchronized_virtualizable()
  827. index = self._get_virtualizable_field_index(fielddescr)
  828. return self.metainterp.virtualizable_boxes[index]
  829. @arguments("box", "descr", "orgpc")
  830. def opimpl_getfield_vable_f(self, box, fielddescr, pc):
  831. if self._nonstandard_virtualizable(pc, box, fielddescr):
  832. return self.opimpl_getfield_gc_f(box, fielddescr)
  833. self.metainterp.check_synchronized_virtualizable()
  834. index = self._get_virtualizable_field_index(fielddescr)
  835. return self.metainterp.virtualizable_boxes[index]
  836. @arguments("box", "box", "descr", "orgpc")
  837. def _opimpl_setfield_vable(self, box, valuebox, fielddescr, pc):
  838. if self._nonstandard_virtualizable(pc, box, fielddescr):
  839. return self._opimpl_setfield_gc_any(box, valuebox, fielddescr)
  840. index = self._get_virtualizable_field_index(fielddescr)
  841. self.metainterp.virtualizable_boxes[index] = valuebox
  842. self.metainterp.synchronize_virtualizable()
  843. # XXX only the index'th field needs to be synchronized, really
  844. opimpl_setfield_vable_i = _opimpl_setfield_vable
  845. opimpl_setfield_vable_r = _opimpl_setfield_vable
  846. opimpl_setfield_vable_f = _opimpl_setfield_vable
  847. def _get_arrayitem_vable_index(self, pc, arrayfielddescr, indexbox):
  848. # Get the index of an array item: the index'th of the array
  849. # described by arrayfielddescr. Must only be called for
  850. # the "standard" virtualizable.
  851. indexbox = self.implement_guard_value(indexbox, pc)
  852. vinfo = self.metainterp.jitdriver_sd.virtualizable_info
  853. virtualizable_box = self.metainterp.virtualizable_boxes[-1]
  854. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  855. arrayindex = vinfo.array_field_by_descrs[arrayfielddescr]
  856. index = indexbox.getint()
  857. # Support for negative index: disabled
  858. # (see codewriter/jtransform.py, _check_no_vable_array).
  859. #if index < 0:
  860. # index += vinfo.get_array_length(virtualizable, arrayindex)
  861. assert 0 <= index < vinfo.get_array_length(virtualizable, arrayindex)
  862. return vinfo.get_index_in_array(virtualizable, arrayindex, index)
  863. @arguments("box", "box", "descr", "descr", "orgpc")
  864. def _opimpl_getarrayitem_vable(self, box, indexbox, fdescr, adescr, pc):
  865. if self._nonstandard_virtualizable(pc, box, fdescr):
  866. arraybox = self.opimpl_getfield_gc_r(box, fdescr)
  867. if adescr.is_array_of_pointers():
  868. return self.opimpl_getarrayitem_gc_r(arraybox, indexbox, adescr)
  869. elif adescr.is_array_of_floats():
  870. return self.opimpl_getarrayitem_gc_f(arraybox, indexbox, adescr)
  871. else:
  872. return self.opimpl_getarrayitem_gc_i(arraybox, indexbox, adescr)
  873. self.metainterp.check_synchronized_virtualizable()
  874. index = self._get_arrayitem_vable_index(pc, fdescr, indexbox)
  875. return self.metainterp.virtualizable_boxes[index]
  876. opimpl_getarrayitem_vable_i = _opimpl_getarrayitem_vable
  877. opimpl_getarrayitem_vable_r = _opimpl_getarrayitem_vable
  878. opimpl_getarrayitem_vable_f = _opimpl_getarrayitem_vable
  879. @arguments("box", "box", "box", "descr", "descr", "orgpc")
  880. def _opimpl_setarrayitem_vable(self, box, indexbox, valuebox,
  881. fdescr, adescr, pc):
  882. if self._nonstandard_virtualizable(pc, box, fdescr):
  883. arraybox = self.opimpl_getfield_gc_r(box, fdescr)
  884. self._opimpl_setarrayitem_gc_any(arraybox, indexbox, valuebox,
  885. adescr)
  886. return
  887. index = self._get_arrayitem_vable_index(pc, fdescr, indexbox)
  888. self.metainterp.virtualizable_boxes[index] = valuebox
  889. self.metainterp.synchronize_virtualizable()
  890. # XXX only the index'th field needs to be synchronized, really
  891. opimpl_setarrayitem_vable_i = _opimpl_setarrayitem_vable
  892. opimpl_setarrayitem_vable_r = _opimpl_setarrayitem_vable
  893. opimpl_setarrayitem_vable_f = _opimpl_setarrayitem_vable
  894. @arguments("box", "descr", "descr", "orgpc")
  895. def opimpl_arraylen_vable(self, box, fdescr, adescr, pc):
  896. if self._nonstandard_virtualizable(pc, box, fdescr):
  897. arraybox = self.opimpl_getfield_gc_r(box, fdescr)
  898. return self.opimpl_arraylen_gc(arraybox, adescr)
  899. vinfo = self.metainterp.jitdriver_sd.virtualizable_info
  900. virtualizable_box = self.metainterp.virtualizable_boxes[-1]
  901. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  902. arrayindex = vinfo.array_field_by_descrs[fdescr]
  903. result = vinfo.get_array_length(virtualizable, arrayindex)
  904. return ConstInt(result)
  905. @arguments("jitcode", "boxes")
  906. def _opimpl_inline_call1(self, jitcode, argboxes):
  907. return self.metainterp.perform_call(jitcode, argboxes)
  908. @arguments("jitcode", "boxes2")
  909. def _opimpl_inline_call2(self, jitcode, argboxes):
  910. return self.metainterp.perform_call(jitcode, argboxes)
  911. @arguments("jitcode", "boxes3")
  912. def _opimpl_inline_call3(self, jitcode, argboxes):
  913. return self.metainterp.perform_call(jitcode, argboxes)
  914. opimpl_inline_call_r_i = _opimpl_inline_call1
  915. opimpl_inline_call_r_r = _opimpl_inline_call1
  916. opimpl_inline_call_r_v = _opimpl_inline_call1
  917. opimpl_inline_call_ir_i = _opimpl_inline_call2
  918. opimpl_inline_call_ir_r = _opimpl_inline_call2
  919. opimpl_inline_call_ir_v = _opimpl_inline_call2
  920. opimpl_inline_call_irf_i = _opimpl_inline_call3
  921. opimpl_inline_call_irf_r = _opimpl_inline_call3
  922. opimpl_inline_call_irf_f = _opimpl_inline_call3
  923. opimpl_inline_call_irf_v = _opimpl_inline_call3
  924. @arguments("box", "boxes", "descr", "orgpc")
  925. def _opimpl_residual_call1(self, funcbox, argboxes, calldescr, pc):
  926. return self.do_residual_or_indirect_call(funcbox, argboxes, calldescr, pc)
  927. @arguments("box", "boxes2", "descr", "orgpc")
  928. def _opimpl_residual_call2(self, funcbox, argboxes, calldescr, pc):
  929. return self.do_residual_or_indirect_call(funcbox, argboxes, calldescr, pc)
  930. @arguments("box", "boxes3", "descr", "orgpc")
  931. def _opimpl_residual_call3(self, funcbox, argboxes, calldescr, pc):
  932. return self.do_residual_or_indirect_call(funcbox, argboxes, calldescr, pc)
  933. opimpl_residual_call_r_i = _opimpl_residual_call1
  934. opimpl_residual_call_r_r = _opimpl_residual_call1
  935. opimpl_residual_call_r_v = _opimpl_residual_call1
  936. opimpl_residual_call_ir_i = _opimpl_residual_call2
  937. opimpl_residual_call_ir_r = _opimpl_residual_call2
  938. opimpl_residual_call_ir_v = _opimpl_residual_call2
  939. opimpl_residual_call_irf_i = _opimpl_residual_call3
  940. opimpl_residual_call_irf_r = _opimpl_residual_call3
  941. opimpl_residual_call_irf_f = _opimpl_residual_call3
  942. opimpl_residual_call_irf_v = _opimpl_residual_call3
  943. @arguments("box", "box", "boxes", "descr", "orgpc")
  944. def opimpl_conditional_call_i_v(self, condbox, funcbox, argboxes, calldescr,
  945. pc):
  946. self.do_conditional_call(condbox, funcbox, argboxes, calldescr, pc)
  947. opimpl_conditional_call_r_v = opimpl_conditional_call_i_v
  948. @arguments("box", "box", "boxes2", "descr", "orgpc")
  949. def opimpl_conditional_call_ir_v(self, condbox, funcbox, argboxes,
  950. calldescr, pc):
  951. self.do_conditional_call(condbox, funcbox, argboxes, calldescr, pc)
  952. @arguments("box", "box", "boxes3", "descr", "orgpc")
  953. def opimpl_conditional_call_irf_v(self, condbox, funcbox, argboxes,
  954. calldescr, pc):
  955. self.do_conditional_call(condbox, funcbox, argboxes, calldescr, pc)
  956. @arguments("int", "boxes3", "boxes3", "orgpc")
  957. def _opimpl_recursive_call(self, jdindex, greenboxes, redboxes, pc):
  958. targetjitdriver_sd = self.metainterp.staticdata.jitdrivers_sd[jdindex]
  959. allboxes = greenboxes + redboxes
  960. warmrunnerstate = targetjitdriver_sd.warmstate
  961. assembler_call = False
  962. if warmrunnerstate.inlining:
  963. if warmrunnerstate.can_inline_callable(greenboxes):
  964. # We've found a potentially inlinable function; now we need to
  965. # see if it's already on the stack. In other words: are we about
  966. # to enter recursion? If so, we don't want to inline the
  967. # recursion, which would be equivalent to unrolling a while
  968. # loop.
  969. portal_code = targetjitdriver_sd.mainjitcode
  970. count = 0
  971. for f in self.metainterp.framestack:
  972. if f.jitcode is not portal_code:
  973. continue
  974. gk = f.greenkey
  975. if gk is None:
  976. continue
  977. assert len(gk) == len(greenboxes)
  978. i = 0
  979. for i in range(len(gk)):
  980. if not gk[i].same_constant(greenboxes[i]):
  981. break
  982. else:
  983. count += 1
  984. memmgr = self.metainterp.staticdata.warmrunnerdesc.memory_manager
  985. if count >= memmgr.max_unroll_recursion:
  986. # This function is recursive and has exceeded the
  987. # maximum number of unrollings we allow. We want to stop
  988. # inlining it further and to make sure that, if it
  989. # hasn't happened already, the function is traced
  990. # separately as soon as possible.
  991. if have_debug_prints():
  992. loc = targetjitdriver_sd.warmstate.get_location_str(greenboxes)
  993. debug_print("recursive function (not inlined):", loc)
  994. warmrunnerstate.dont_trace_here(greenboxes)
  995. else:
  996. return self.metainterp.perform_call(portal_code, allboxes,
  997. greenkey=greenboxes)
  998. assembler_call = True
  999. # verify that we have all green args, needed to make sure
  1000. # that assembler that we call is still correct
  1001. self.verify_green_args(targetjitdriver_sd, greenboxes)
  1002. #
  1003. return self.do_recursive_call(targetjitdriver_sd, allboxes, pc,
  1004. assembler_call)
  1005. def do_recursive_call(self, targetjitdriver_sd, allboxes, pc,
  1006. assembler_call=False):
  1007. portal_code = targetjitdriver_sd.mainjitcode
  1008. k = targetjitdriver_sd.portal_runner_adr
  1009. funcbox = ConstInt(heaptracker.adr2int(k))
  1010. return self.do_residual_call(funcbox, allboxes, portal_code.calldescr, pc,
  1011. assembler_call=assembler_call,
  1012. assembler_call_jd=targetjitdriver_sd)
  1013. opimpl_recursive_call_i = _opimpl_recursive_call
  1014. opimpl_recursive_call_r = _opimpl_recursive_call
  1015. opimpl_recursive_call_f = _opimpl_recursive_call
  1016. opimpl_recursive_call_v = _opimpl_recursive_call
  1017. @arguments("box")
  1018. def opimpl_strlen(self, strbox):
  1019. return self.execute(rop.STRLEN, strbox)
  1020. @arguments("box")
  1021. def opimpl_unicodelen(self, unicodebox):
  1022. return self.execute(rop.UNICODELEN, unicodebox)
  1023. @arguments("box", "box")
  1024. def opimpl_strgetitem(self, strbox, indexbox):
  1025. return self.execute(rop.STRGETITEM, strbox, indexbox)
  1026. @arguments("box", "box")
  1027. def opimpl_unicodegetitem(self, unicodebox, indexbox):
  1028. return self.execute(rop.UNICODEGETITEM, unicodebox, indexbox)
  1029. @arguments("box", "box", "box")
  1030. def opimpl_strsetitem(self, strbox, indexbox, newcharbox):
  1031. return self.execute(rop.STRSETITEM, strbox, indexbox, newcharbox)
  1032. @arguments("box", "box", "box")
  1033. def opimpl_unicodesetitem(self, unicodebox, indexbox, newcharbox):
  1034. self.execute(rop.UNICODESETITEM, unicodebox, indexbox, newcharbox)
  1035. @arguments("box")
  1036. def opimpl_newstr(self, lengthbox):
  1037. return self.execute(rop.NEWSTR, lengthbox)
  1038. @arguments("box")
  1039. def opimpl_newunicode(self, lengthbox):
  1040. return self.execute(rop.NEWUNICODE, lengthbox)
  1041. @arguments("box", "box", "box", "box", "box")
  1042. def opimpl_copystrcontent(self, srcbox, dstbox, srcstartbox, dststartbox, lengthbox):
  1043. return self.execute(rop.COPYSTRCONTENT, srcbox, dstbox, srcstartbox, dststartbox, lengthbox)
  1044. @arguments("box", "box", "box", "box", "box")
  1045. def opimpl_copyunicodecontent(self, srcbox, dstbox, srcstartbox, dststartbox, lengthbox):
  1046. return self.execute(rop.COPYUNICODECONTENT, srcbox, dstbox, srcstartbox, dststartbox, lengthbox)
  1047. @arguments("box", "orgpc")
  1048. def _opimpl_guard_value(self, box, orgpc):
  1049. self.implement_guard_value(box, orgpc)
  1050. @arguments("box", "box", "descr", "orgpc")
  1051. def opimpl_str_guard_value(self, box, funcbox, descr, orgpc):
  1052. if isinstance(box, Const):
  1053. return box # no promotion needed, already a Const
  1054. else:
  1055. constbox = ConstPtr(box.getref_base())
  1056. resbox = self.do_residual_call(funcbox, [box, constbox], descr, orgpc)
  1057. promoted_box = ConstInt(resbox.getint())
  1058. # This is GUARD_VALUE because GUARD_TRUE assumes the existance
  1059. # of a label when computing resumepc
  1060. self.metainterp.generate_guard(rop.GUARD_VALUE, resbox,
  1061. [promoted_box],
  1062. resumepc=orgpc)
  1063. self.metainterp.replace_box(box, constbox)
  1064. return constbox
  1065. opimpl_int_guard_value = _opimpl_guard_value
  1066. opimpl_ref_guard_value = _opimpl_guard_value
  1067. opimpl_float_guard_value = _opimpl_guard_value
  1068. @arguments("box", "orgpc")
  1069. def opimpl_guard_class(self, box, orgpc):
  1070. clsbox = self.cls_of_box(box)
  1071. if not self.metainterp.heapcache.is_class_known(box):
  1072. self.metainterp.generate_guard(rop.GUARD_CLASS, box, [clsbox],
  1073. resumepc=orgpc)
  1074. self.metainterp.heapcache.class_now_known(box)
  1075. return clsbox
  1076. @arguments("int", "orgpc")
  1077. def opimpl_loop_header(self, jdindex, orgpc):
  1078. self.metainterp.seen_loop_header_for_jdindex = jdindex
  1079. def verify_green_args(self, jitdriver_sd, varargs):
  1080. num_green_args = jitdriver_sd.num_green_args
  1081. assert len(varargs) == num_green_args
  1082. for i in range(num_green_args):
  1083. assert isinstance(varargs[i], Const)
  1084. @arguments("int", "boxes3", "jitcode_position", "boxes3", "orgpc")
  1085. def opimpl_jit_merge_point(self, jdindex, greenboxes,
  1086. jcposition, redboxes, orgpc):
  1087. any_operation = self.metainterp.history.any_operation()
  1088. jitdriver_sd = self.metainterp.staticdata.jitdrivers_sd[jdindex]
  1089. self.verify_green_args(jitdriver_sd, greenboxes)
  1090. self.debug_merge_point(jitdriver_sd, jdindex,
  1091. self.metainterp.portal_call_depth,
  1092. self.metainterp.call_ids[-1],
  1093. greenboxes)
  1094. if self.metainterp.seen_loop_header_for_jdindex < 0:
  1095. if not any_operation:
  1096. return
  1097. if self.metainterp.portal_call_depth or not self.metainterp.get_procedure_token(greenboxes, True):
  1098. if not jitdriver_sd.no_loop_header:
  1099. return
  1100. # automatically add a loop_header if there is none
  1101. self.metainterp.seen_loop_header_for_jdindex = jdindex
  1102. #
  1103. assert self.metainterp.seen_loop_header_for_jdindex == jdindex, (
  1104. "found a loop_header for a JitDriver that does not match "
  1105. "the following jit_merge_point's")
  1106. self.metainterp.seen_loop_header_for_jdindex = -1
  1107. #
  1108. if not self.metainterp.portal_call_depth:
  1109. assert jitdriver_sd is self.metainterp.jitdriver_sd
  1110. # Set self.pc to point to jit_merge_point instead of just after:
  1111. # if reached_loop_header() raises SwitchToBlackhole, then the
  1112. # pc is still at the jit_merge_point, which is a point that is
  1113. # much less expensive to blackhole out of.
  1114. saved_pc = self.pc
  1115. self.pc = orgpc
  1116. self.metainterp.reached_loop_header(greenboxes, redboxes)
  1117. self.pc = saved_pc
  1118. # no exception, which means that the jit_merge_point did not
  1119. # close the loop. We have to put the possibly-modified list
  1120. # 'redboxes' back into the registers where it comes from.
  1121. put_back_list_of_boxes3(self, jcposition, redboxes)
  1122. else:
  1123. if jitdriver_sd.warmstate.should_unroll_one_iteration(greenboxes):
  1124. if self.unroll_iterations > 0:
  1125. self.unroll_iterations -= 1
  1126. return
  1127. # warning! careful here. We have to return from the current
  1128. # frame containing the jit_merge_point, and then use
  1129. # do_recursive_call() to follow the recursive call. This is
  1130. # needed because do_recursive_call() will write its result
  1131. # with make_result_of_lastop(), so the lastop must be right:
  1132. # it must be the call to 'self', and not the jit_merge_point
  1133. # itself, which has no result at all.
  1134. assert len(self.metainterp.framestack) >= 2
  1135. old_frame = self.metainterp.framestack[-1]
  1136. try:
  1137. self.metainterp.finishframe(None, leave_portal_frame=False)
  1138. except ChangeFrame:
  1139. pass
  1140. frame = self.metainterp.framestack[-1]
  1141. frame.do_recursive_call(jitdriver_sd, greenboxes + redboxes, orgpc,
  1142. assembler_call=True)
  1143. jd_no = old_frame.jitcode.jitdriver_sd.index
  1144. self.metainterp.leave_portal_frame(jd_no)
  1145. raise ChangeFrame
  1146. def debug_merge_point(self, jitdriver_sd, jd_index, portal_call_depth, current_call_id, greenkey):
  1147. # debugging: produce a DEBUG_MERGE_POINT operation
  1148. if have_debug_prints():
  1149. loc = jitdriver_sd.warmstate.get_location_str(greenkey)
  1150. debug_print(loc)
  1151. #
  1152. # Note: the logger hides the jd_index argument, so we see in the logs:
  1153. # debug_merge_point(portal_call_depth, current_call_id, 'location')
  1154. #
  1155. args = [ConstInt(jd_index), ConstInt(portal_call_depth),
  1156. ConstInt(current_call_id)] + greenkey
  1157. self.metainterp.history.record(rop.DEBUG_MERGE_POINT, args, None)
  1158. @arguments("box", "label")
  1159. def opimpl_goto_if_exception_mismatch(self, vtablebox, next_exc_target):
  1160. metainterp = self.metainterp
  1161. last_exc_value = metainterp.last_exc_value
  1162. assert last_exc_value
  1163. assert metainterp.class_of_last_exc_is_const
  1164. if not metainterp.cpu.ts.instanceOf(ConstPtr(lltype.cast_opaque_ptr(llmemory.GCREF, last_exc_value)), vtablebox):
  1165. self.pc = next_exc_target
  1166. @arguments("box", "orgpc")
  1167. def opimpl_raise(self, exc_value_box, orgpc):
  1168. # xxx hack
  1169. if not self.metainterp.heapcache.is_class_known(exc_value_box):
  1170. clsbox = self.cls_of_box(exc_value_box)
  1171. self.metainterp.generate_guard(rop.GUARD_CLASS, exc_value_box,
  1172. [clsbox], resumepc=orgpc)
  1173. self.metainterp.class_of_last_exc_is_const = True
  1174. self.metainterp.last_exc_value = exc_value_box.getref(rclass.OBJECTPTR)
  1175. self.metainterp.last_exc_box = exc_value_box
  1176. self.metainterp.popframe()
  1177. self.metainterp.finishframe_exception()
  1178. @arguments()
  1179. def opimpl_reraise(self):
  1180. assert self.metainterp.last_exc_value
  1181. self.metainterp.popframe()
  1182. self.metainterp.finishframe_exception()
  1183. @arguments()
  1184. def opimpl_last_exception(self):
  1185. # Same comment as in opimpl_goto_if_exception_mismatch().
  1186. exc_value = self.metainterp.last_exc_value
  1187. assert exc_value
  1188. assert self.metainterp.class_of_last_exc_is_const
  1189. return self.metainterp.cpu.ts.cls_of_box(ConstPtr(lltype.cast_opaque_ptr(llmemory.GCREF, exc_value)))
  1190. @arguments()
  1191. def opimpl_last_exc_value(self):
  1192. exc_value = self.metainterp.last_exc_value
  1193. assert exc_value
  1194. return self.metainterp.last_exc_box
  1195. @arguments("box")
  1196. def opimpl_debug_fatalerror(self, box):
  1197. from rpython.rtyper.lltypesystem import rstr, lloperation
  1198. msg = box.getref(lltype.Ptr(rstr.STR))
  1199. lloperation.llop.debug_fatalerror(lltype.Void, msg)
  1200. @arguments("box", "box", "box", "box", "box")
  1201. def opimpl_jit_debug(self, stringbox, arg1box, arg2box, arg3box, arg4box):
  1202. debug_print('jit_debug:', stringbox._get_str(),
  1203. arg1box.getint(), arg2box.getint(),
  1204. arg3box.getint(), arg4box.getint())
  1205. args = [stringbox, arg1box, arg2box, arg3box, arg4box]
  1206. i = 4
  1207. while i > 0 and args[i].getint() == -sys.maxint-1:
  1208. i -= 1
  1209. assert i >= 0
  1210. op = self.metainterp.history.record(rop.JIT_DEBUG, args[:i+1], None)
  1211. self.metainterp.attach_debug_info(op)
  1212. @arguments("box")
  1213. def opimpl_jit_enter_portal_frame(self, uniqueidbox):
  1214. unique_id = uniqueidbox.getint()
  1215. jd_no = self.metainterp.jitdriver_sd.mainjitcode.index # fish
  1216. self.metainterp.enter_portal_frame(jd_no, unique_id)
  1217. @arguments()
  1218. def opimpl_jit_leave_portal_frame(self):
  1219. jd_no = self.metainterp.jitdriver_sd.mainjitcode.index # fish
  1220. self.metainterp.leave_portal_frame(jd_no)
  1221. @arguments("box")
  1222. def _opimpl_assert_green(self, box):
  1223. if not isinstance(box, Const):
  1224. msg = "assert_green failed at %s:%d" % (
  1225. self.jitcode.name,
  1226. self.pc)
  1227. if we_are_translated():
  1228. from rpython.rtyper.annlowlevel import llstr
  1229. from rpython.rtyper.lltypesystem import lloperation
  1230. lloperation.llop.debug_fatalerror(lltype.Void, llstr(msg))
  1231. else:
  1232. from rpython.rlib.jit import AssertGreenFailed
  1233. raise AssertGreenFailed(msg)
  1234. opimpl_int_assert_green = _opimpl_assert_green
  1235. opimpl_ref_assert_green = _opimpl_assert_green
  1236. opimpl_float_assert_green = _opimpl_assert_green
  1237. @arguments()
  1238. def opimpl_current_trace_length(self):
  1239. trace_length = self.metainterp.history.length()
  1240. return ConstInt(trace_length)
  1241. @arguments("box")
  1242. def _opimpl_isconstant(self, box):
  1243. return ConstInt(isinstance(box, Const))
  1244. opimpl_int_isconstant = _opimpl_isconstant
  1245. opimpl_ref_isconstant = _opimpl_isconstant
  1246. opimpl_float_isconstant = _opimpl_isconstant
  1247. @arguments("box")
  1248. def _opimpl_isvirtual(self, box):
  1249. return ConstInt(self.metainterp.heapcache.is_likely_virtual(box))
  1250. opimpl_ref_isvirtual = _opimpl_isvirtual
  1251. @arguments("box")
  1252. def opimpl_virtual_ref(self, box):
  1253. # Details on the content of metainterp.virtualref_boxes:
  1254. #
  1255. # * it's a list whose items go two by two, containing first the
  1256. # virtual box (e.g. the PyFrame) and then the vref box (e.g.
  1257. # the 'virtual_ref(frame)').
  1258. #
  1259. # * if we detect that the virtual box escapes during tracing
  1260. # already (by generating a CALL_MAY_FORCE that marks the flags
  1261. # in the vref), then we replace the vref in the list with
  1262. # ConstPtr(NULL).
  1263. #
  1264. metainterp = self.metainterp
  1265. vrefinfo = metainterp.staticdata.virtualref_info
  1266. obj = box.getref_base()
  1267. vref = vrefinfo.virtual_ref_during_tracing(obj)
  1268. cindex = history.ConstInt(len(metainterp.virtualref_boxes) // 2)
  1269. resbox = metainterp.history.record(rop.VIRTUAL_REF, [box, cindex], vref)
  1270. self.metainterp.heapcache.new(resbox)
  1271. # Note: we allocate a JIT_VIRTUAL_REF here
  1272. # (in virtual_ref_during_tracing()), in order to detect when
  1273. # the virtual escapes during tracing already. We record it as a
  1274. # VIRTUAL_REF operation. Later, optimizeopt.py should either kill
  1275. # that operation or replace it with a NEW_WITH_VTABLE followed by
  1276. # SETFIELD_GCs.
  1277. metainterp.virtualref_boxes.append(box)
  1278. metainterp.virtualref_boxes.append(resbox)
  1279. return resbox
  1280. @arguments("box")
  1281. def opimpl_virtual_ref_finish(self, box):
  1282. # virtual_ref_finish() assumes that we have a stack-like, last-in
  1283. # first-out order.
  1284. metainterp = self.metainterp
  1285. vrefbox = metainterp.virtualref_boxes.pop()
  1286. lastbox = metainterp.virtualref_boxes.pop()
  1287. assert box.getref_base() == lastbox.getref_base()
  1288. vrefinfo = metainterp.staticdata.virtualref_info
  1289. vref = vrefbox.getref_base()
  1290. if vrefinfo.is_virtual_ref(vref):
  1291. # XXX write a comment about nullbox
  1292. nullbox = self.metainterp.cpu.ts.CONST_NULL
  1293. metainterp.history.record(rop.VIRTUAL_REF_FINISH,
  1294. [vrefbox, nullbox], None)
  1295. @arguments("int", "box")
  1296. def opimpl_rvmprof_code(self, leaving, box_unique_id):
  1297. from rpython.rlib.rvmprof import cintf
  1298. cintf.jit_rvmprof_code(leaving, box_unique_id.getint())
  1299. def handle_rvmprof_enter_on_resume(self):
  1300. code = self.bytecode
  1301. position = self.pc
  1302. opcode = ord(code[position])
  1303. if opcode == self.metainterp.staticdata.op_rvmprof_code:
  1304. arg1 = self.registers_i[ord(code[position + 1])].getint()
  1305. arg2 = self.registers_i[ord(code[position + 2])].getint()
  1306. if arg1 == 1:
  1307. # we are resuming at a position that will do a
  1308. # jit_rvmprof_code(1), when really executed. That's a
  1309. # hint for the need for a jit_rvmprof_code(0).
  1310. from rpython.rlib.rvmprof import cintf
  1311. cintf.jit_rvmprof_code(0, arg2)
  1312. # ------------------------------
  1313. def setup_call(self, argboxes):
  1314. self.pc = 0
  1315. count_i = count_r = count_f = 0
  1316. for box in argboxes:
  1317. if box.type == history.INT:
  1318. self.registers_i[count_i] = box
  1319. count_i += 1
  1320. elif box.type == history.REF:
  1321. self.registers_r[count_r] = box
  1322. count_r += 1
  1323. elif box.type == history.FLOAT:
  1324. self.registers_f[count_f] = box
  1325. count_f += 1
  1326. else:
  1327. raise AssertionError(box.type)
  1328. def setup_resume_at_op(self, pc):
  1329. self.pc = pc
  1330. def handle_possible_overflow_error(self, label, orgpc, resbox):
  1331. if self.metainterp.ovf_flag:
  1332. self.metainterp.generate_guard(rop.GUARD_OVERFLOW, None,
  1333. resumepc=orgpc)
  1334. self.pc = label
  1335. return None
  1336. else:
  1337. self.metainterp.generate_guard(rop.GUARD_NO_OVERFLOW, None,
  1338. resumepc=orgpc)
  1339. return resbox
  1340. def run_one_step(self):
  1341. # Execute the frame forward. This method contains a loop that leaves
  1342. # whenever the 'opcode_implementations' (which is one of the 'opimpl_'
  1343. # methods) raises ChangeFrame. This is the case when the current frame
  1344. # changes, due to a call or a return.
  1345. try:
  1346. staticdata = self.metainterp.staticdata
  1347. while True:
  1348. pc = self.pc
  1349. op = ord(self.bytecode[pc])
  1350. staticdata.opcode_implementations[op](self, pc)
  1351. except ChangeFrame:
  1352. pass
  1353. def implement_guard_value(self, box, orgpc):
  1354. """Promote the given Box into a Const. Note: be careful, it's a
  1355. bit unclear what occurs if a single opcode needs to generate
  1356. several ones and/or ones not near the beginning."""
  1357. if isinstance(box, Const):
  1358. return box # no promotion needed, already a Const
  1359. else:
  1360. promoted_box = executor.constant_from_op(box)
  1361. self.metainterp.generate_guard(rop.GUARD_VALUE, box, [promoted_box],
  1362. resumepc=orgpc)
  1363. self.metainterp.replace_box(box, promoted_box)
  1364. return promoted_box
  1365. def cls_of_box(self, box):
  1366. return self.metainterp.cpu.ts.cls_of_box(box)
  1367. @specialize.arg(1)
  1368. def execute(self, opnum, *argboxes):
  1369. return self.metainterp.execute_and_record(opnum, None, *argboxes)
  1370. @specialize.arg(1)
  1371. def execute_with_descr(self, opnum, descr, *argboxes):
  1372. return self.metainterp.execute_and_record(opnum, descr, *argboxes)
  1373. @specialize.arg(1)
  1374. def execute_varargs(self, opnum, argboxes, descr, exc, pure):
  1375. self.metainterp.clear_exception()
  1376. patch_pos = self.metainterp.history.get_trace_position()
  1377. op = self.metainterp.execute_and_record_varargs(opnum, argboxes,
  1378. descr=descr)
  1379. if pure and not self.metainterp.last_exc_value and op:
  1380. op = self.metainterp.record_result_of_call_pure(op, argboxes, descr,
  1381. patch_pos)
  1382. exc = exc and not isinstance(op, Const)
  1383. if exc:
  1384. if op is not None:
  1385. self.make_result_of_lastop(op)
  1386. # ^^^ this is done before handle_possible_exception() because we
  1387. # need the box to show up in get_list_of_active_boxes()
  1388. self.metainterp.handle_possible_exception()
  1389. else:
  1390. self.metainterp.assert_no_exception()
  1391. return op
  1392. def _build_allboxes(self, funcbox, argboxes, descr):
  1393. allboxes = [None] * (len(argboxes)+1)
  1394. allboxes[0] = funcbox
  1395. src_i = src_r = src_f = 0
  1396. i = 1
  1397. for kind in descr.get_arg_types():
  1398. if kind == history.INT or kind == 'S': # single float
  1399. while True:
  1400. box = argboxes[src_i]
  1401. src_i += 1
  1402. if box.type == history.INT:
  1403. break
  1404. elif kind == history.REF:
  1405. while True:
  1406. box = argboxes[src_r]
  1407. src_r += 1
  1408. if box.type == history.REF:
  1409. break
  1410. elif kind == history.FLOAT or kind == 'L': # long long
  1411. while True:
  1412. box = argboxes[src_f]
  1413. src_f += 1
  1414. if box.type == history.FLOAT:
  1415. break
  1416. else:
  1417. raise AssertionError
  1418. allboxes[i] = box
  1419. i += 1
  1420. assert i == len(allboxes)
  1421. return allboxes
  1422. def do_residual_call(self, funcbox, argboxes, descr, pc,
  1423. assembler_call=False,
  1424. assembler_call_jd=None):
  1425. # First build allboxes: it may need some reordering from the
  1426. # list provided in argboxes, depending on the order in which
  1427. # the arguments are expected by the function
  1428. #
  1429. allboxes = self._build_allboxes(funcbox, argboxes, descr)
  1430. effectinfo = descr.get_extra_info()
  1431. if effectinfo.oopspecindex == effectinfo.OS_NOT_IN_TRACE:
  1432. return self.metainterp.do_not_in_trace_call(allboxes, descr)
  1433. cut_pos = self.metainterp.history.get_trace_position()
  1434. if (assembler_call or
  1435. effectinfo.check_forces_virtual_or_virtualizable()):
  1436. # residual calls require attention to keep virtualizables in-sync
  1437. self.metainterp.clear_exception()
  1438. if effectinfo.oopspecindex == EffectInfo.OS_JIT_FORCE_VIRTUAL:
  1439. resbox = self._do_jit_force_virtual(allboxes, descr, pc)
  1440. if resbox is not None:
  1441. return resbox
  1442. # 1. preparation
  1443. self.metainterp.vable_and_vrefs_before_residual_call()
  1444. # 2. actually do the call now (we'll have cases later): the
  1445. # result is stored into 'c_result' for now, which is a Const
  1446. metainterp = self.metainterp
  1447. tp = descr.get_normalized_result_type()
  1448. if tp == 'i':
  1449. opnum1 = rop.CALL_MAY_FORCE_I
  1450. value = executor.execute_varargs(metainterp.cpu, metainterp,
  1451. opnum1, allboxes, descr)
  1452. c_result = ConstInt(value)
  1453. elif tp == 'r':
  1454. opnum1 = rop.CALL_MAY_FORCE_R
  1455. value = executor.execute_varargs(metainterp.cpu, metainterp,
  1456. opnum1, allboxes, descr)
  1457. c_result = ConstPtr(value)
  1458. elif tp == 'f':
  1459. opnum1 = rop.CALL_MAY_FORCE_F
  1460. value = executor.execute_varargs(metainterp.cpu, metainterp,
  1461. opnum1, allboxes, descr)
  1462. c_result = ConstFloat(value)
  1463. elif tp == 'v':
  1464. opnum1 = rop.CALL_MAY_FORCE_N
  1465. executor.execute_varargs(metainterp.cpu, metainterp,
  1466. opnum1, allboxes, descr)
  1467. c_result = None
  1468. else:
  1469. assert False
  1470. # 3. after this call, check the vrefs. If any have been
  1471. # forced by the call, then we record in the trace a
  1472. # VIRTUAL_REF_FINISH---before we record any CALL
  1473. self.metainterp.vrefs_after_residual_call()
  1474. # 4. figure out what kind of CALL we need to record
  1475. # from the effectinfo and the 'assembler_call' flag
  1476. if assembler_call:
  1477. vablebox, resbox = self.metainterp.direct_assembler_call(
  1478. allboxes, descr, assembler_call_jd)
  1479. else:
  1480. vablebox = None
  1481. resbox = None
  1482. if effectinfo.oopspecindex == effectinfo.OS_LIBFFI_CALL:
  1483. resbox = self.metainterp.direct_libffi_call(allboxes, descr)
  1484. # ^^^ may return None to mean "can't handle it myself"
  1485. if resbox is None:
  1486. if effectinfo.is_call_release_gil():
  1487. resbox = self.metainterp.direct_call_release_gil(
  1488. allboxes, descr)
  1489. else:
  1490. resbox = self.metainterp.direct_call_may_force(
  1491. allboxes, descr)
  1492. # 5. invalidate the heapcache based on the CALL_MAY_FORCE
  1493. # operation executed above in step 2
  1494. self.metainterp.heapcache.invalidate_caches(opnum1, descr, allboxes)
  1495. # 6. put 'c_result' back into the recorded operation
  1496. if resbox.type == 'v':
  1497. resbox = None # for void calls, must return None below
  1498. else:
  1499. resbox.copy_value_from(c_result)
  1500. self.make_result_of_lastop(resbox)
  1501. self.metainterp.vable_after_residual_call(funcbox)
  1502. self.metainterp.generate_guard(rop.GUARD_NOT_FORCED, None)
  1503. if vablebox is not None:
  1504. self.metainterp.history.record(rop.KEEPALIVE, [vablebox], None)
  1505. self.metainterp.handle_possible_exception()
  1506. return resbox
  1507. else:
  1508. effect = effectinfo.extraeffect
  1509. tp = descr.get_normalized_result_type()
  1510. if effect == effectinfo.EF_LOOPINVARIANT:
  1511. if tp == 'i':
  1512. return self.execute_varargs(rop.CALL_LOOPINVARIANT_I,
  1513. allboxes,
  1514. descr, False, False)
  1515. elif tp == 'r':
  1516. return self.execute_varargs(rop.CALL_LOOPINVARIANT_R,
  1517. allboxes,
  1518. descr, False, False)
  1519. elif tp == 'f':
  1520. return self.execute_varargs(rop.CALL_LOOPINVARIANT_F,
  1521. allboxes,
  1522. descr, False, False)
  1523. elif tp == 'v':
  1524. return self.execute_varargs(rop.CALL_LOOPINVARIANT_N,
  1525. allboxes,
  1526. descr, False, False)
  1527. else:
  1528. assert False
  1529. exc = effectinfo.check_can_raise()
  1530. pure = effectinfo.check_is_elidable()
  1531. if tp == 'i':
  1532. return self.execute_varargs(rop.CALL_I, allboxes, descr,
  1533. exc, pure)
  1534. elif tp == 'r':
  1535. return self.execute_varargs(rop.CALL_R, allboxes, descr,
  1536. exc, pure)
  1537. elif tp == 'f':
  1538. return self.execute_varargs(rop.CALL_F, allboxes, descr,
  1539. exc, pure)
  1540. elif tp == 'v':
  1541. return self.execute_varargs(rop.CALL_N, allboxes, descr,
  1542. exc, pure)
  1543. else:
  1544. assert False
  1545. def do_conditional_call(self, condbox, funcbox, argboxes, descr, pc):
  1546. if isinstance(condbox, ConstInt) and condbox.value == 0:
  1547. return # so that the heapcache can keep argboxes virtual
  1548. allboxes = self._build_allboxes(funcbox, argboxes, descr)
  1549. effectinfo = descr.get_extra_info()
  1550. assert not effectinfo.check_forces_virtual_or_virtualizable()
  1551. exc = effectinfo.check_can_raise()
  1552. pure = effectinfo.check_is_elidable()
  1553. return self.execute_varargs(rop.COND_CALL, [condbox] + allboxes, descr,
  1554. exc, pure)
  1555. def _do_jit_force_virtual(self, allboxes, descr, pc):
  1556. assert len(allboxes) == 2
  1557. if (self.metainterp.jitdriver_sd.virtualizable_info is None and
  1558. self.metainterp.jitdriver_sd.greenfield_info is None):
  1559. # can occur in case of multiple JITs
  1560. return None
  1561. vref_box = allboxes[1]
  1562. standard_box = self.metainterp.virtualizable_boxes[-1]
  1563. if standard_box is vref_box:
  1564. return vref_box
  1565. if self.metainterp.heapcache.is_nonstandard_virtualizable(vref_box):
  1566. return None
  1567. eqbox = self.metainterp.execute_and_record(rop.PTR_EQ, None, vref_box, standard_box)
  1568. eqbox = self.implement_guard_value(eqbox, pc)
  1569. isstandard = eqbox.getint()
  1570. if isstandard:
  1571. return standard_box
  1572. else:
  1573. return None
  1574. def do_residual_or_indirect_call(self, funcbox, argboxes, calldescr, pc):
  1575. """The 'residual_call' operation is emitted in two cases:
  1576. when we have to generate a residual CALL operation, but also
  1577. to handle an indirect_call that may need to be inlined."""
  1578. if isinstance(funcbox, Const):
  1579. sd = self.metainterp.staticdata
  1580. key = sd.cpu.ts.getaddr_for_box(funcbox)
  1581. jitcode = sd.bytecode_for_address(key)
  1582. if jitcode is not None:
  1583. # we should follow calls to this graph
  1584. return self.metainterp.perform_call(jitcode, argboxes)
  1585. # but we should not follow calls to that graph
  1586. return self.do_residual_call(funcbox, argboxes, calldescr, pc)
  1587. # ____________________________________________________________
  1588. class MetaInterpStaticData(object):
  1589. logger_noopt = None
  1590. logger_ops = None
  1591. def __init__(self, cpu, options,
  1592. ProfilerClass=EmptyProfiler, warmrunnerdesc=None):
  1593. self.cpu = cpu
  1594. self.stats = self.cpu.stats
  1595. self.options = options
  1596. self.jitlog = jl.JitLogger(self.cpu)
  1597. self.logger_noopt = Logger(self)
  1598. self.logger_ops = Logger(self, guard_number=True)
  1599. # legacy loggers
  1600. self.jitlog.logger_noopt = self.logger_noopt
  1601. self.jitlog.logger_ops = self.logger_ops
  1602. self.profiler = ProfilerClass()
  1603. self.profiler.cpu = cpu
  1604. self.warmrunnerdesc = warmrunnerdesc
  1605. if warmrunnerdesc:
  1606. self.config = warmrunnerdesc.translator.config
  1607. else:
  1608. from rpython.config.translationoption import get_combined_translation_config
  1609. self.config = get_combined_translation_config(translating=True)
  1610. backendmodule = self.cpu.__module__
  1611. backendmodule = backendmodule.split('.')[-2]
  1612. self.jit_starting_line = 'JIT starting (%s)' % backendmodule
  1613. self._addr2name_keys = []
  1614. self._addr2name_values = []
  1615. self.__dict__.update(compile.make_done_loop_tokens())
  1616. for val in ['int', 'float', 'ref', 'void']:
  1617. fullname = 'done_with_this_frame_descr_' + val
  1618. setattr(self.cpu, fullname, getattr(self, fullname))
  1619. d = self.exit_frame_with_exception_descr_ref
  1620. self.cpu.exit_frame_with_exception_descr_ref = d
  1621. def _freeze_(self):
  1622. return True
  1623. def setup_insns(self, insns):
  1624. self.opcode_names = ['?'] * len(insns)
  1625. self.opcode_implementations = [None] * len(insns)
  1626. for key, value in insns.items():
  1627. assert self.opcode_implementations[value] is None
  1628. self.opcode_names[value] = key
  1629. name, argcodes = key.split('/')
  1630. opimpl = _get_opimpl_method(name, argcodes)
  1631. self.opcode_implementations[value] = opimpl
  1632. self.op_catch_exception = insns.get('catch_exception/L', -1)
  1633. self.op_rvmprof_code = insns.get('rvmprof_code/ii', -1)
  1634. def setup_descrs(self, descrs):
  1635. self.opcode_descrs = descrs
  1636. def setup_indirectcalltargets(self, indirectcalltargets):
  1637. self.indirectcalltargets = list(indirectcalltargets)
  1638. def setup_list_of_addr2name(self, list_of_addr2name):
  1639. self._addr2name_keys = [key for key, value in list_of_addr2name]
  1640. self._addr2name_values = [value for key, value in list_of_addr2name]
  1641. def finish_setup(self, codewriter, optimizer=None):
  1642. from rpython.jit.metainterp.blackhole import BlackholeInterpBuilder
  1643. self.blackholeinterpbuilder = BlackholeInterpBuilder(codewriter, self)
  1644. #
  1645. asm = codewriter.assembler
  1646. self.setup_insns(asm.insns)
  1647. self.setup_descrs(asm.descrs)
  1648. self.setup_indirectcalltargets(asm.indirectcalltargets)
  1649. self.setup_list_of_addr2name(asm.list_of_addr2name)
  1650. #
  1651. self.jitdrivers_sd = codewriter.callcontrol.jitdrivers_sd
  1652. self.virtualref_info = codewriter.callcontrol.virtualref_info
  1653. self.callinfocollection = codewriter.callcontrol.callinfocollection
  1654. self.has_libffi_call = codewriter.callcontrol.has_libffi_call
  1655. #
  1656. # store this information for fastpath of call_assembler
  1657. # (only the paths that can actually be taken)
  1658. exc_descr = compile.PropagateExceptionDescr()
  1659. for jd in self.jitdrivers_sd:
  1660. name = {history.INT: 'int',
  1661. history.REF: 'ref',
  1662. history.FLOAT: 'float',
  1663. history.VOID: 'void'}[jd.result_type]
  1664. tokens = getattr(self, 'loop_tokens_done_with_this_frame_%s' % name)
  1665. jd.portal_finishtoken = tokens[0].finishdescr
  1666. jd.propagate_exc_descr = exc_descr
  1667. #
  1668. self.cpu.propagate_exception_descr = exc_descr
  1669. #
  1670. self.globaldata = MetaInterpGlobalData(self)
  1671. def finish_setup_descrs(self):
  1672. from rpython.jit.codewriter import effectinfo
  1673. self.all_descrs = self.cpu.setup_descrs()
  1674. effectinfo.compute_bitstrings(self.all_descrs)
  1675. def _setup_once(self):
  1676. """Runtime setup needed by the various components of the JIT."""
  1677. if not self.globaldata.initialized:
  1678. self.jitlog.setup_once()
  1679. debug_print(self.jit_starting_line)
  1680. self.cpu.setup_once()
  1681. if not self.profiler.initialized:
  1682. self.profiler.start()
  1683. self.profiler.initialized = True
  1684. self.globaldata.initialized = True
  1685. def get_name_from_address(self, addr):
  1686. # for debugging only
  1687. if we_are_translated():
  1688. d = self.globaldata.addr2name
  1689. if d is None:
  1690. # Build the dictionary at run-time. This is needed
  1691. # because the keys are function/class addresses, so they
  1692. # can change from run to run.
  1693. d = {}
  1694. keys = self._addr2name_keys
  1695. values = self._addr2name_values
  1696. for i in range(len(keys)):
  1697. d[keys[i]] = values[i]
  1698. self.globaldata.addr2name = d
  1699. return d.get(addr, '')
  1700. else:
  1701. for i in range(len(self._addr2name_keys)):
  1702. if addr == self._addr2name_keys[i]:
  1703. return self._addr2name_values[i]
  1704. return ''
  1705. def bytecode_for_address(self, fnaddress):
  1706. if we_are_translated():
  1707. d = self.globaldata.indirectcall_dict
  1708. if d is None:
  1709. # Build the dictionary at run-time. This is needed
  1710. # because the keys are function addresses, so they
  1711. # can change from run to run.
  1712. d = {}
  1713. for jitcode in self.indirectcalltargets:
  1714. assert jitcode.fnaddr not in d
  1715. d[jitcode.fnaddr] = jitcode
  1716. self.globaldata.indirectcall_dict = d
  1717. return d.get(fnaddress, None)
  1718. else:
  1719. for jitcode in self.indirectcalltargets:
  1720. if jitcode.fnaddr == fnaddress:
  1721. return jitcode
  1722. return None
  1723. def try_to_free_some_loops(self):
  1724. # Increase here the generation recorded by the memory manager.
  1725. if self.warmrunnerdesc is not None: # for tests
  1726. self.warmrunnerdesc.memory_manager.next_generation()
  1727. # ---------------- logging ------------------------
  1728. def log(self, msg):
  1729. debug_print(msg)
  1730. # ____________________________________________________________
  1731. class MetaInterpGlobalData(object):
  1732. """This object contains the JIT's global, mutable data.
  1733. Warning: for any data that you put here, think that there might be
  1734. multiple MetaInterps accessing it at the same time. As usual we are
  1735. safe from corruption thanks to the GIL, but keep in mind that any
  1736. MetaInterp might modify any of these fields while another MetaInterp
  1737. is, say, currently in a residual call to a function. Multiple
  1738. MetaInterps occur either with threads or, in single-threaded cases,
  1739. with recursion. This is a case that is not well-tested, so please
  1740. be careful :-( But thankfully this is one of the very few places
  1741. where multiple concurrent MetaInterps may interact with each other.
  1742. """
  1743. def __init__(self, staticdata):
  1744. self.initialized = False
  1745. self.indirectcall_dict = None
  1746. self.addr2name = None
  1747. # ____________________________________________________________
  1748. class MetaInterp(object):
  1749. portal_call_depth = 0
  1750. cancel_count = 0
  1751. exported_state = None
  1752. last_exc_box = None
  1753. _last_op = None
  1754. def __init__(self, staticdata, jitdriver_sd):
  1755. self.staticdata = staticdata
  1756. self.cpu = staticdata.cpu
  1757. self.jitdriver_sd = jitdriver_sd
  1758. # Note: self.jitdriver_sd is the JitDriverStaticData that corresponds
  1759. # to the current loop -- the outermost one. Be careful, because
  1760. # during recursion we can also see other jitdrivers.
  1761. self.portal_trace_positions = []
  1762. self.free_frames_list = []
  1763. self.last_exc_value = lltype.nullptr(rclass.OBJECT)
  1764. self.forced_virtualizable = None
  1765. self.partial_trace = None
  1766. self.retracing_from = (-1, -1, -1)
  1767. self.call_pure_results = args_dict()
  1768. self.heapcache = HeapCache()
  1769. self.call_ids = []
  1770. self.current_call_id = 0
  1771. self.box_names_memo = {}
  1772. self.aborted_tracing_jitdriver = None
  1773. self.aborted_tracing_greenkey = None
  1774. def retrace_needed(self, trace, exported_state):
  1775. self.partial_trace = trace
  1776. self.retracing_from = self.potential_retrace_position
  1777. self.exported_state = exported_state
  1778. self.heapcache.reset()
  1779. def perform_call(self, jitcode, boxes, greenkey=None):
  1780. # causes the metainterp to enter the given subfunction
  1781. f = self.newframe(jitcode, greenkey)
  1782. f.setup_call(boxes)
  1783. raise ChangeFrame
  1784. def is_main_jitcode(self, jitcode):
  1785. return (jitcode.jitdriver_sd is not None and
  1786. jitcode.jitdriver_sd.jitdriver.is_recursive)
  1787. #return self.jitdriver_sd is not None and jitcode is self.jitdriver_sd.mainjitcode
  1788. def newframe(self, jitcode, greenkey=None):
  1789. if jitcode.jitdriver_sd:
  1790. self.portal_call_depth += 1
  1791. self.call_ids.append(self.current_call_id)
  1792. unique_id = -1
  1793. if greenkey is not None:
  1794. unique_id = jitcode.jitdriver_sd.warmstate.get_unique_id(
  1795. greenkey)
  1796. jd_no = jitcode.jitdriver_sd.index
  1797. self.enter_portal_frame(jd_no, unique_id)
  1798. self.current_call_id += 1
  1799. if greenkey is not None and self.is_main_jitcode(jitcode):
  1800. self.portal_trace_positions.append(
  1801. (jitcode.jitdriver_sd, greenkey, self.history.get_trace_position()))
  1802. if len(self.free_frames_list) > 0:
  1803. f = self.free_frames_list.pop()
  1804. else:
  1805. f = MIFrame(self)
  1806. f.setup(jitcode, greenkey)
  1807. self.framestack.append(f)
  1808. return f
  1809. def enter_portal_frame(self, jd_no, unique_id):
  1810. self.history.record(rop.ENTER_PORTAL_FRAME,
  1811. [ConstInt(jd_no), ConstInt(unique_id)], None)
  1812. def leave_portal_frame(self, jd_no):
  1813. self.history.record(rop.LEAVE_PORTAL_FRAME, [ConstInt(jd_no)], None)
  1814. def popframe(self, leave_portal_frame=True):
  1815. frame = self.framestack.pop()
  1816. jitcode = frame.jitcode
  1817. if jitcode.jitdriver_sd:
  1818. self.portal_call_depth -= 1
  1819. if leave_portal_frame:
  1820. self.leave_portal_frame(jitcode.jitdriver_sd.index)
  1821. self.call_ids.pop()
  1822. if frame.greenkey is not None and self.is_main_jitcode(jitcode):
  1823. self.portal_trace_positions.append(
  1824. (jitcode.jitdriver_sd, None, self.history.get_trace_position()))
  1825. # we save the freed MIFrames to avoid needing to re-create new
  1826. # MIFrame objects all the time; they are a bit big, with their
  1827. # 3*256 register entries.
  1828. frame.cleanup_registers()
  1829. self.free_frames_list.append(frame)
  1830. def finishframe(self, resultbox, leave_portal_frame=True):
  1831. # handle a non-exceptional return from the current frame
  1832. self.last_exc_value = lltype.nullptr(rclass.OBJECT)
  1833. self.popframe(leave_portal_frame=leave_portal_frame)
  1834. if self.framestack:
  1835. if resultbox is not None:
  1836. self.framestack[-1].make_result_of_lastop(resultbox)
  1837. raise ChangeFrame
  1838. else:
  1839. try:
  1840. self.compile_done_with_this_frame(resultbox)
  1841. except SwitchToBlackhole as stb:
  1842. self.aborted_tracing(stb.reason)
  1843. sd = self.staticdata
  1844. result_type = self.jitdriver_sd.result_type
  1845. if result_type == history.VOID:
  1846. assert resultbox is None
  1847. raise jitexc.DoneWithThisFrameVoid()
  1848. elif result_type == history.INT:
  1849. raise jitexc.DoneWithThisFrameInt(int(resultbox.getint()))
  1850. elif result_type == history.REF:
  1851. raise jitexc.DoneWithThisFrameRef(self.cpu, resultbox.getref_base())
  1852. elif result_type == history.FLOAT:
  1853. raise jitexc.DoneWithThisFrameFloat(resultbox.getfloatstorage())
  1854. else:
  1855. assert False
  1856. def finishframe_exception(self):
  1857. excvalue = self.last_exc_value
  1858. while self.framestack:
  1859. frame = self.framestack[-1]
  1860. code = frame.bytecode
  1861. position = frame.pc # <-- just after the insn that raised
  1862. if position < len(code):
  1863. opcode = ord(code[position])
  1864. if opcode == self.staticdata.op_catch_exception:
  1865. # found a 'catch_exception' instruction;
  1866. # jump to the handler
  1867. target = ord(code[position+1]) | (ord(code[position+2])<<8)
  1868. frame.pc = target
  1869. raise ChangeFrame
  1870. if opcode == self.staticdata.op_rvmprof_code:
  1871. # call the 'jit_rvmprof_code(1)' for rvmprof, but then
  1872. # continue popping frames. Decode the 'rvmprof_code' insn
  1873. # manually here.
  1874. from rpython.rlib.rvmprof import cintf
  1875. arg1 = frame.registers_i[ord(code[position + 1])].getint()
  1876. arg2 = frame.registers_i[ord(code[position + 2])].getint()
  1877. assert arg1 == 1
  1878. cintf.jit_rvmprof_code(arg1, arg2)
  1879. self.popframe()
  1880. try:
  1881. self.compile_exit_frame_with_exception(self.last_exc_box)
  1882. except SwitchToBlackhole as stb:
  1883. self.aborted_tracing(stb.reason)
  1884. raise jitexc.ExitFrameWithExceptionRef(self.cpu, lltype.cast_opaque_ptr(llmemory.GCREF, excvalue))
  1885. def check_recursion_invariant(self):
  1886. portal_call_depth = -1
  1887. for frame in self.framestack:
  1888. jitcode = frame.jitcode
  1889. if jitcode.jitdriver_sd:
  1890. portal_call_depth += 1
  1891. if portal_call_depth != self.portal_call_depth:
  1892. print "portal_call_depth problem!!!"
  1893. print portal_call_depth, self.portal_call_depth
  1894. for frame in self.framestack:
  1895. jitcode = frame.jitcode
  1896. if jitcode.jitdriver_sd:
  1897. print "P",
  1898. else:
  1899. print " ",
  1900. print jitcode.name
  1901. raise AssertionError
  1902. def generate_guard(self, opnum, box=None, extraargs=[], resumepc=-1):
  1903. if isinstance(box, Const): # no need for a guard
  1904. return
  1905. if box is not None:
  1906. moreargs = [box] + extraargs
  1907. else:
  1908. moreargs = list(extraargs)
  1909. if opnum == rop.GUARD_EXCEPTION:
  1910. guard_op = self.history.record(opnum, moreargs,
  1911. lltype.nullptr(llmemory.GCREF.TO))
  1912. else:
  1913. guard_op = self.history.record(opnum, moreargs, None)
  1914. self.capture_resumedata(resumepc)
  1915. # ^^^ records extra to history
  1916. self.staticdata.profiler.count_ops(opnum, Counters.GUARDS)
  1917. # count
  1918. #self.attach_debug_info(guard_op)
  1919. return guard_op
  1920. def capture_resumedata(self, resumepc=-1):
  1921. virtualizable_boxes = None
  1922. if (self.jitdriver_sd.virtualizable_info is not None or
  1923. self.jitdriver_sd.greenfield_info is not None):
  1924. virtualizable_boxes = self.virtualizable_boxes
  1925. saved_pc = 0
  1926. if self.framestack:
  1927. frame = self.framestack[-1]
  1928. saved_pc = frame.pc
  1929. if resumepc >= 0:
  1930. frame.pc = resumepc
  1931. resume.capture_resumedata(self.framestack, virtualizable_boxes,
  1932. self.virtualref_boxes, self.history.trace)
  1933. if self.framestack:
  1934. self.framestack[-1].pc = saved_pc
  1935. def create_empty_history(self):
  1936. self.history = history.History()
  1937. self.staticdata.stats.set_history(self.history)
  1938. def _all_constants(self, *boxes):
  1939. if len(boxes) == 0:
  1940. return True
  1941. return isinstance(boxes[0], Const) and self._all_constants(*boxes[1:])
  1942. def _all_constants_varargs(self, boxes):
  1943. for box in boxes:
  1944. if not isinstance(box, Const):
  1945. return False
  1946. return True
  1947. @specialize.arg(1)
  1948. def execute_and_record(self, opnum, descr, *argboxes):
  1949. history.check_descr(descr)
  1950. assert not (rop._CANRAISE_FIRST <= opnum <= rop._CANRAISE_LAST)
  1951. # execute the operation
  1952. profiler = self.staticdata.profiler
  1953. profiler.count_ops(opnum)
  1954. resvalue = executor.execute(self.cpu, self, opnum, descr, *argboxes)
  1955. if OpHelpers.is_pure_with_descr(opnum, descr):
  1956. return self._record_helper_pure(opnum, resvalue, descr, *argboxes)
  1957. if rop._OVF_FIRST <= opnum <= rop._OVF_LAST:
  1958. return self._record_helper_ovf(opnum, resvalue, descr, *argboxes)
  1959. return self._record_helper_nonpure_varargs(opnum, resvalue, descr,
  1960. list(argboxes))
  1961. @specialize.arg(1)
  1962. def execute_and_record_varargs(self, opnum, argboxes, descr=None):
  1963. history.check_descr(descr)
  1964. # execute the operation
  1965. profiler = self.staticdata.profiler
  1966. profiler.count_ops(opnum)
  1967. resvalue = executor.execute_varargs(self.cpu, self,
  1968. opnum, argboxes, descr)
  1969. # check if the operation can be constant-folded away
  1970. argboxes = list(argboxes)
  1971. if rop._ALWAYS_PURE_FIRST <= opnum <= rop._ALWAYS_PURE_LAST:
  1972. return self._record_helper_pure_varargs(opnum, resvalue, descr,
  1973. argboxes)
  1974. return self._record_helper_nonpure_varargs(opnum, resvalue, descr,
  1975. argboxes)
  1976. @specialize.argtype(2)
  1977. def _record_helper_pure(self, opnum, resvalue, descr, *argboxes):
  1978. canfold = self._all_constants(*argboxes)
  1979. if canfold:
  1980. return history.newconst(resvalue)
  1981. else:
  1982. return self._record_helper_nonpure_varargs(opnum, resvalue, descr,
  1983. list(argboxes))
  1984. def _record_helper_ovf(self, opnum, resvalue, descr, *argboxes):
  1985. if (not self.last_exc_value and
  1986. self._all_constants(*argboxes)):
  1987. return history.newconst(resvalue)
  1988. return self._record_helper_nonpure_varargs(opnum, resvalue, descr, list(argboxes))
  1989. @specialize.argtype(2)
  1990. def _record_helper_pure_varargs(self, opnum, resvalue, descr, argboxes):
  1991. canfold = self._all_constants_varargs(argboxes)
  1992. if canfold:
  1993. return executor.wrap_constant(resvalue)
  1994. else:
  1995. return self._record_helper_nonpure_varargs(opnum, resvalue, descr,
  1996. argboxes)
  1997. @specialize.argtype(2)
  1998. def _record_helper_nonpure_varargs(self, opnum, resvalue, descr, argboxes):
  1999. # record the operation
  2000. profiler = self.staticdata.profiler
  2001. profiler.count_ops(opnum, Counters.RECORDED_OPS)
  2002. self.heapcache.invalidate_caches(opnum, descr, argboxes)
  2003. op = self.history.record(opnum, argboxes, resvalue, descr)
  2004. self.attach_debug_info(op)
  2005. if op.type != 'v':
  2006. return op
  2007. def execute_new_with_vtable(self, descr):
  2008. resbox = self.execute_and_record(rop.NEW_WITH_VTABLE, descr)
  2009. self.heapcache.new(resbox)
  2010. self.heapcache.class_now_known(resbox)
  2011. return resbox
  2012. def execute_new(self, typedescr):
  2013. resbox = self.execute_and_record(rop.NEW, typedescr)
  2014. self.heapcache.new(resbox)
  2015. return resbox
  2016. def execute_new_array(self, itemsizedescr, lengthbox):
  2017. resbox = self.execute_and_record(rop.NEW_ARRAY, itemsizedescr,
  2018. lengthbox)
  2019. self.heapcache.new_array(resbox, lengthbox)
  2020. return resbox
  2021. def execute_new_array_clear(self, itemsizedescr, lengthbox):
  2022. resbox = self.execute_and_record(rop.NEW_ARRAY_CLEAR, itemsizedescr,
  2023. lengthbox)
  2024. self.heapcache.new_array(resbox, lengthbox)
  2025. return resbox
  2026. def execute_setfield_gc(self, fielddescr, box, valuebox):
  2027. self.execute_and_record(rop.SETFIELD_GC, fielddescr, box, valuebox)
  2028. self.heapcache.setfield(box, valuebox, fielddescr)
  2029. def execute_setarrayitem_gc(self, arraydescr, arraybox, indexbox, itembox):
  2030. self.execute_and_record(rop.SETARRAYITEM_GC, arraydescr,
  2031. arraybox, indexbox, itembox)
  2032. self.heapcache.setarrayitem(arraybox, indexbox, itembox, arraydescr)
  2033. def execute_setinteriorfield_gc(self, descr, array, index, value):
  2034. self.execute_and_record(rop.SETINTERIORFIELD_GC, descr,
  2035. array, index, value)
  2036. def execute_raw_store(self, arraydescr, addrbox, offsetbox, valuebox):
  2037. self.execute_and_record(rop.RAW_STORE, arraydescr,
  2038. addrbox, offsetbox, valuebox)
  2039. def attach_debug_info(self, op):
  2040. if (not we_are_translated() and op is not None
  2041. and getattr(self, 'framestack', None)):
  2042. op.pc = self.framestack[-1].pc
  2043. op.name = self.framestack[-1].jitcode.name
  2044. def execute_raised(self, exception, constant=False):
  2045. if isinstance(exception, jitexc.JitException):
  2046. raise exception # go through
  2047. llexception = jitexc.get_llexception(self.cpu, exception)
  2048. self.execute_ll_raised(llexception, constant)
  2049. def execute_ll_raised(self, llexception, constant=False):
  2050. # Exception handling: when execute.do_call() gets an exception it
  2051. # calls metainterp.execute_raised(), which puts it into
  2052. # 'self.last_exc_value'. This is used shortly afterwards
  2053. # to generate either GUARD_EXCEPTION or GUARD_NO_EXCEPTION, and also
  2054. # to handle the following opcodes 'goto_if_exception_mismatch'.
  2055. self.last_exc_value = llexception
  2056. self.class_of_last_exc_is_const = constant
  2057. # 'class_of_last_exc_is_const' means that the class of the value
  2058. # stored in the exc_value Box can be assumed to be a Const. This
  2059. # is only True after a GUARD_EXCEPTION or GUARD_CLASS.
  2060. def clear_exception(self):
  2061. self.last_exc_value = lltype.nullptr(rclass.OBJECT)
  2062. def aborted_tracing(self, reason):
  2063. self.staticdata.profiler.count(reason)
  2064. debug_print('~~~ ABORTING TRACING %s' % Counters.counter_names[reason])
  2065. jd_sd = self.jitdriver_sd
  2066. if not self.current_merge_points:
  2067. greenkey = None # we're in the bridge
  2068. else:
  2069. greenkey = self.current_merge_points[0][0][:jd_sd.num_green_args]
  2070. self.staticdata.warmrunnerdesc.hooks.on_abort(reason,
  2071. jd_sd.jitdriver, greenkey,
  2072. jd_sd.warmstate.get_location_str(greenkey),
  2073. self.staticdata.logger_ops._make_log_operations(
  2074. self.box_names_memo),
  2075. self.history.trace.unpack()[1])
  2076. if self.aborted_tracing_jitdriver is not None:
  2077. jd_sd = self.aborted_tracing_jitdriver
  2078. greenkey = self.aborted_tracing_greenkey
  2079. self.staticdata.warmrunnerdesc.hooks.on_trace_too_long(
  2080. jd_sd.jitdriver, greenkey,
  2081. jd_sd.warmstate.get_location_str(greenkey))
  2082. # no ops for now
  2083. self.aborted_tracing_jitdriver = None
  2084. self.aborted_tracing_greenkey = None
  2085. self.staticdata.stats.aborted()
  2086. def blackhole_if_trace_too_long(self):
  2087. warmrunnerstate = self.jitdriver_sd.warmstate
  2088. if self.history.length() > warmrunnerstate.trace_limit:
  2089. jd_sd, greenkey_of_huge_function = self.find_biggest_function()
  2090. self.history.trace.done()
  2091. self.staticdata.stats.record_aborted(greenkey_of_huge_function)
  2092. self.portal_trace_positions = None
  2093. if greenkey_of_huge_function is not None:
  2094. jd_sd.warmstate.disable_noninlinable_function(
  2095. greenkey_of_huge_function)
  2096. self.aborted_tracing_jitdriver = jd_sd
  2097. self.aborted_tracing_greenkey = greenkey_of_huge_function
  2098. if self.current_merge_points:
  2099. jd_sd = self.jitdriver_sd
  2100. greenkey = self.current_merge_points[0][0][:jd_sd.num_green_args]
  2101. warmrunnerstate.JitCell.trace_next_iteration(greenkey)
  2102. raise SwitchToBlackhole(Counters.ABORT_TOO_LONG)
  2103. def _interpret(self):
  2104. # Execute the frames forward until we raise a DoneWithThisFrame,
  2105. # a ExitFrameWithException, or a ContinueRunningNormally exception.
  2106. self.staticdata.stats.entered()
  2107. while True:
  2108. self.framestack[-1].run_one_step()
  2109. self.blackhole_if_trace_too_long()
  2110. if not we_are_translated():
  2111. self.check_recursion_invariant()
  2112. def interpret(self):
  2113. if we_are_translated():
  2114. self._interpret()
  2115. else:
  2116. try:
  2117. self._interpret()
  2118. except:
  2119. import sys
  2120. if sys.exc_info()[0] is not None:
  2121. self.staticdata.log(sys.exc_info()[0].__name__)
  2122. raise
  2123. @specialize.arg(1)
  2124. def compile_and_run_once(self, jitdriver_sd, *args):
  2125. # NB. we pass explicity 'jitdriver_sd' around here, even though it
  2126. # is also available as 'self.jitdriver_sd', because we need to
  2127. # specialize this function and a few other ones for the '*args'.
  2128. debug_start('jit-tracing')
  2129. self.staticdata._setup_once()
  2130. self.staticdata.profiler.start_tracing()
  2131. assert jitdriver_sd is self.jitdriver_sd
  2132. self.staticdata.try_to_free_some_loops()
  2133. self.create_empty_history()
  2134. try:
  2135. original_boxes = self.initialize_original_boxes(jitdriver_sd, *args)
  2136. return self._compile_and_run_once(original_boxes)
  2137. finally:
  2138. self.staticdata.profiler.end_tracing()
  2139. debug_stop('jit-tracing')
  2140. def _compile_and_run_once(self, original_boxes):
  2141. self.initialize_state_from_start(original_boxes)
  2142. self.current_merge_points = [(original_boxes, (0, 0, 0))]
  2143. num_green_args = self.jitdriver_sd.num_green_args
  2144. original_greenkey = original_boxes[:num_green_args]
  2145. self.resumekey = compile.ResumeFromInterpDescr(original_greenkey)
  2146. self.history.set_inputargs(original_boxes[num_green_args:],
  2147. self.staticdata)
  2148. self.seen_loop_header_for_jdindex = -1
  2149. try:
  2150. self.interpret()
  2151. except SwitchToBlackhole as stb:
  2152. self.run_blackhole_interp_to_cancel_tracing(stb)
  2153. assert False, "should always raise"
  2154. def handle_guard_failure(self, resumedescr, deadframe):
  2155. debug_start('jit-tracing')
  2156. self.staticdata.profiler.start_tracing()
  2157. if isinstance(resumedescr, compile.ResumeGuardCopiedDescr):
  2158. key = resumedescr.prev
  2159. else:
  2160. key = resumedescr
  2161. assert isinstance(key, compile.ResumeGuardDescr)
  2162. # store the resumekey.wref_original_loop_token() on 'self' to make
  2163. # sure that it stays alive as long as this MetaInterp
  2164. self.resumekey_original_loop_token = resumedescr.rd_loop_token.loop_token_wref()
  2165. if self.resumekey_original_loop_token is None:
  2166. raise compile.giveup() # should be rare
  2167. self.staticdata.try_to_free_some_loops()
  2168. inputargs = self.initialize_state_from_guard_failure(key, deadframe)
  2169. try:
  2170. return self._handle_guard_failure(resumedescr, key, inputargs, deadframe)
  2171. finally:
  2172. self.resumekey_original_loop_token = None
  2173. self.staticdata.profiler.end_tracing()
  2174. debug_stop('jit-tracing')
  2175. def _handle_guard_failure(self, resumedescr, key, inputargs, deadframe):
  2176. self.current_merge_points = []
  2177. self.resumekey = resumedescr
  2178. self.seen_loop_header_for_jdindex = -1
  2179. if isinstance(key, compile.ResumeAtPositionDescr):
  2180. self.seen_loop_header_for_jdindex = self.jitdriver_sd.index
  2181. try:
  2182. self.prepare_resume_from_failure(deadframe, inputargs, resumedescr)
  2183. if self.resumekey_original_loop_token is None: # very rare case
  2184. raise SwitchToBlackhole(Counters.ABORT_BRIDGE)
  2185. self.interpret()
  2186. except SwitchToBlackhole as stb:
  2187. self.run_blackhole_interp_to_cancel_tracing(stb)
  2188. assert False, "should always raise"
  2189. def run_blackhole_interp_to_cancel_tracing(self, stb):
  2190. # We got a SwitchToBlackhole exception. Convert the framestack into
  2191. # a stack of blackhole interpreters filled with the same values, and
  2192. # run it.
  2193. from rpython.jit.metainterp.blackhole import convert_and_run_from_pyjitpl
  2194. self.aborted_tracing(stb.reason)
  2195. convert_and_run_from_pyjitpl(self, stb.raising_exception)
  2196. assert False # ^^^ must raise
  2197. def remove_consts_and_duplicates(self, boxes, endindex, duplicates):
  2198. for i in range(endindex):
  2199. box = boxes[i]
  2200. if isinstance(box, Const) or box in duplicates:
  2201. opnum = OpHelpers.same_as_for_type(box.type)
  2202. op = self.history.record_default_val(opnum, [box])
  2203. boxes[i] = op
  2204. else:
  2205. duplicates[box] = None
  2206. def reached_loop_header(self, greenboxes, redboxes):
  2207. self.heapcache.reset() #reset_virtuals=False)
  2208. #self.heapcache.reset_keep_likely_virtuals()
  2209. duplicates = {}
  2210. self.remove_consts_and_duplicates(redboxes, len(redboxes),
  2211. duplicates)
  2212. live_arg_boxes = greenboxes + redboxes
  2213. if self.jitdriver_sd.virtualizable_info is not None:
  2214. # we use pop() to remove the last item, which is the virtualizable
  2215. # itself
  2216. self.remove_consts_and_duplicates(self.virtualizable_boxes,
  2217. len(self.virtualizable_boxes)-1,
  2218. duplicates)
  2219. live_arg_boxes += self.virtualizable_boxes
  2220. live_arg_boxes.pop()
  2221. # generate a dummy guard just before the JUMP so that unroll can use it
  2222. # when it's creating artificial guards.
  2223. self.generate_guard(rop.GUARD_FUTURE_CONDITION)
  2224. assert len(self.virtualref_boxes) == 0, "missing virtual_ref_finish()?"
  2225. # Called whenever we reach the 'loop_header' hint.
  2226. # First, attempt to make a bridge:
  2227. # - if self.resumekey is a ResumeGuardDescr, it starts from a guard
  2228. # that failed;
  2229. # - if self.resumekey is a ResumeFromInterpDescr, it starts directly
  2230. # from the interpreter.
  2231. if not self.partial_trace:
  2232. # FIXME: Support a retrace to be a bridge as well as a loop
  2233. self.compile_trace(live_arg_boxes)
  2234. # raises in case it works -- which is the common case, hopefully,
  2235. # at least for bridges starting from a guard.
  2236. # Search in current_merge_points for original_boxes with compatible
  2237. # green keys, representing the beginning of the same loop as the one
  2238. # we end now.
  2239. num_green_args = self.jitdriver_sd.num_green_args
  2240. for j in range(len(self.current_merge_points)-1, -1, -1):
  2241. original_boxes, start = self.current_merge_points[j]
  2242. assert len(original_boxes) == len(live_arg_boxes)
  2243. for i in range(num_green_args):
  2244. box1 = original_boxes[i]
  2245. box2 = live_arg_boxes[i]
  2246. assert isinstance(box1, Const)
  2247. if not box1.same_constant(box2):
  2248. break
  2249. else:
  2250. if self.partial_trace:
  2251. if start != self.retracing_from:
  2252. raise SwitchToBlackhole(Counters.ABORT_BAD_LOOP) # For now
  2253. # Found! Compile it as a loop.
  2254. # raises in case it works -- which is the common case
  2255. self.compile_loop(original_boxes, live_arg_boxes, start,
  2256. exported_state=self.exported_state)
  2257. self.exported_state = None
  2258. # creation of the loop was cancelled!
  2259. self.cancel_count += 1
  2260. if self.staticdata.warmrunnerdesc:
  2261. memmgr = self.staticdata.warmrunnerdesc.memory_manager
  2262. if memmgr:
  2263. if self.cancel_count > memmgr.max_unroll_loops:
  2264. self.compile_loop_or_abort(original_boxes,
  2265. live_arg_boxes,
  2266. start)
  2267. self.staticdata.log('cancelled, tracing more...')
  2268. # Otherwise, no loop found so far, so continue tracing.
  2269. start = self.history.get_trace_position()
  2270. self.current_merge_points.append((live_arg_boxes, start))
  2271. def _unpack_boxes(self, boxes, start, stop):
  2272. ints = []; refs = []; floats = []
  2273. for i in range(start, stop):
  2274. box = boxes[i]
  2275. if box.type == history.INT: ints.append(box.getint())
  2276. elif box.type == history.REF: refs.append(box.getref_base())
  2277. elif box.type == history.FLOAT:floats.append(box.getfloatstorage())
  2278. else: assert 0
  2279. return ints[:], refs[:], floats[:]
  2280. def raise_continue_running_normally(self, live_arg_boxes, loop_token):
  2281. self.history.inputargs = None
  2282. self.history.operations = None
  2283. # For simplicity, we just raise ContinueRunningNormally here and
  2284. # ignore the loop_token passed in. It means that we go back to
  2285. # interpreted mode, but it should come back very quickly to the
  2286. # JIT, find probably the same 'loop_token', and execute it.
  2287. if we_are_translated():
  2288. num_green_args = self.jitdriver_sd.num_green_args
  2289. gi, gr, gf = self._unpack_boxes(live_arg_boxes, 0, num_green_args)
  2290. ri, rr, rf = self._unpack_boxes(live_arg_boxes, num_green_args,
  2291. len(live_arg_boxes))
  2292. CRN = jitexc.ContinueRunningNormally
  2293. raise CRN(gi, gr, gf, ri, rr, rf)
  2294. else:
  2295. # However, in order to keep the existing tests working
  2296. # (which are based on the assumption that 'loop_token' is
  2297. # directly used here), a bit of custom non-translatable code...
  2298. self._nontranslated_run_directly(live_arg_boxes, loop_token)
  2299. assert 0, "unreachable"
  2300. def _nontranslated_run_directly(self, live_arg_boxes, loop_token):
  2301. "NOT_RPYTHON"
  2302. args = []
  2303. num_green_args = self.jitdriver_sd.num_green_args
  2304. num_red_args = self.jitdriver_sd.num_red_args
  2305. for box in live_arg_boxes[num_green_args:num_green_args+num_red_args]:
  2306. if box.type == history.INT: args.append(box.getint())
  2307. elif box.type == history.REF: args.append(box.getref_base())
  2308. elif box.type == history.FLOAT: args.append(box.getfloatstorage())
  2309. else: assert 0
  2310. res = self.jitdriver_sd.warmstate.execute_assembler(loop_token, *args)
  2311. kind = history.getkind(lltype.typeOf(res))
  2312. if kind == 'void': raise jitexc.DoneWithThisFrameVoid()
  2313. if kind == 'int': raise jitexc.DoneWithThisFrameInt(res)
  2314. if kind == 'ref': raise jitexc.DoneWithThisFrameRef(self.cpu, res)
  2315. if kind == 'float': raise jitexc.DoneWithThisFrameFloat(res)
  2316. raise AssertionError(kind)
  2317. def prepare_resume_from_failure(self, deadframe, inputargs, resumedescr):
  2318. exception = self.cpu.grab_exc_value(deadframe)
  2319. if (isinstance(resumedescr, compile.ResumeGuardExcDescr) or
  2320. isinstance(resumedescr, compile.ResumeGuardCopiedExcDescr)):
  2321. # Add a GUARD_EXCEPTION or GUARD_NO_EXCEPTION at the start
  2322. # of the bridge---except it is not really the start, because
  2323. # the history aleady contains operations from resume.py.
  2324. # The optimizer should remove these operations. However,
  2325. # 'test_guard_no_exception_incorrectly_removed_from_bridge'
  2326. # shows a corner case in which just putting GuARD_NO_EXCEPTION
  2327. # here is a bad idea: the optimizer might remove it too.
  2328. # So we put a SAVE_EXCEPTION at the start, and a
  2329. # RESTORE_EXCEPTION just before the guard. (rewrite.py will
  2330. # remove the two if they end up consecutive.)
  2331. # XXX too much jumps between older and newer models; clean up
  2332. # by killing SAVE_EXC_CLASS, RESTORE_EXCEPTION and GUARD_EXCEPTION
  2333. exception_obj = lltype.cast_opaque_ptr(rclass.OBJECTPTR, exception)
  2334. if exception_obj:
  2335. exc_class = heaptracker.adr2int(
  2336. llmemory.cast_ptr_to_adr(exception_obj.typeptr))
  2337. else:
  2338. exc_class = 0
  2339. assert self.history.trace is None
  2340. i = len(self.history._cache)
  2341. op1 = self.history.record(rop.SAVE_EXC_CLASS, [], exc_class)
  2342. op2 = self.history.record(rop.SAVE_EXCEPTION, [], exception)
  2343. self.history._cache = self.history._cache[i:] + self.history._cache[:i]
  2344. self.history.record(rop.RESTORE_EXCEPTION, [op1, op2], None)
  2345. self.history.set_inputargs(inputargs, self.staticdata)
  2346. if exception_obj:
  2347. self.execute_ll_raised(exception_obj)
  2348. else:
  2349. self.clear_exception()
  2350. try:
  2351. self.handle_possible_exception()
  2352. except ChangeFrame:
  2353. pass
  2354. else:
  2355. self.history.set_inputargs(inputargs, self.staticdata)
  2356. assert not exception
  2357. def get_procedure_token(self, greenkey, with_compiled_targets=False):
  2358. JitCell = self.jitdriver_sd.warmstate.JitCell
  2359. cell = JitCell.get_jit_cell_at_key(greenkey)
  2360. if cell is None:
  2361. return None
  2362. token = cell.get_procedure_token()
  2363. if with_compiled_targets:
  2364. if not token:
  2365. return None
  2366. if not token.target_tokens:
  2367. return None
  2368. return token
  2369. def compile_loop(self, original_boxes, live_arg_boxes, start,
  2370. try_disabling_unroll=False, exported_state=None):
  2371. num_green_args = self.jitdriver_sd.num_green_args
  2372. greenkey = original_boxes[:num_green_args]
  2373. self.history.trace.done()
  2374. if not self.partial_trace:
  2375. ptoken = self.get_procedure_token(greenkey)
  2376. if ptoken is not None and ptoken.target_tokens is not None:
  2377. # XXX this path not tested, but shown to occur on pypy-c :-(
  2378. self.staticdata.log('cancelled: we already have a token now')
  2379. raise SwitchToBlackhole(Counters.ABORT_BAD_LOOP)
  2380. if self.partial_trace:
  2381. target_token = compile.compile_retrace(self, greenkey, start,
  2382. original_boxes[num_green_args:],
  2383. live_arg_boxes[num_green_args:],
  2384. self.partial_trace,
  2385. self.resumekey,
  2386. exported_state)
  2387. else:
  2388. target_token = compile.compile_loop(self, greenkey, start,
  2389. original_boxes[num_green_args:],
  2390. live_arg_boxes[num_green_args:],
  2391. try_disabling_unroll=try_disabling_unroll)
  2392. if target_token is not None:
  2393. assert isinstance(target_token, TargetToken)
  2394. self.jitdriver_sd.warmstate.attach_procedure_to_interp(greenkey, target_token.targeting_jitcell_token)
  2395. self.staticdata.stats.add_jitcell_token(target_token.targeting_jitcell_token)
  2396. if target_token is not None: # raise if it *worked* correctly
  2397. assert isinstance(target_token, TargetToken)
  2398. jitcell_token = target_token.targeting_jitcell_token
  2399. self.raise_continue_running_normally(live_arg_boxes, jitcell_token)
  2400. def compile_loop_or_abort(self, original_boxes, live_arg_boxes,
  2401. start):
  2402. """Called after we aborted more than 'max_unroll_loops' times.
  2403. As a last attempt, try to compile the loop with unrolling disabled.
  2404. """
  2405. if not self.partial_trace:
  2406. self.compile_loop(original_boxes, live_arg_boxes, start,
  2407. try_disabling_unroll=True)
  2408. #
  2409. self.staticdata.log('cancelled too many times!')
  2410. raise SwitchToBlackhole(Counters.ABORT_BAD_LOOP)
  2411. def compile_trace(self, live_arg_boxes):
  2412. num_green_args = self.jitdriver_sd.num_green_args
  2413. greenkey = live_arg_boxes[:num_green_args]
  2414. target_jitcell_token = self.get_procedure_token(greenkey, True)
  2415. if not target_jitcell_token:
  2416. return
  2417. cut_at = self.history.get_trace_position()
  2418. self.potential_retrace_position = cut_at
  2419. self.history.record(rop.JUMP, live_arg_boxes[num_green_args:], None,
  2420. descr=target_jitcell_token)
  2421. self.history.ends_with_jump = True
  2422. self.history.trace.done()
  2423. try:
  2424. target_token = compile.compile_trace(self, self.resumekey,
  2425. live_arg_boxes[num_green_args:])
  2426. finally:
  2427. self.history.cut(cut_at) # pop the jump
  2428. self.history.ends_with_jump = False
  2429. if target_token is not None: # raise if it *worked* correctly
  2430. assert isinstance(target_token, TargetToken)
  2431. jitcell_token = target_token.targeting_jitcell_token
  2432. self.raise_continue_running_normally(live_arg_boxes, jitcell_token)
  2433. def compile_done_with_this_frame(self, exitbox):
  2434. # temporarily put a JUMP to a pseudo-loop
  2435. self.store_token_in_vable()
  2436. sd = self.staticdata
  2437. result_type = self.jitdriver_sd.result_type
  2438. if result_type == history.VOID:
  2439. assert exitbox is None
  2440. exits = []
  2441. loop_tokens = sd.loop_tokens_done_with_this_frame_void
  2442. elif result_type == history.INT:
  2443. exits = [exitbox]
  2444. loop_tokens = sd.loop_tokens_done_with_this_frame_int
  2445. elif result_type == history.REF:
  2446. exits = [exitbox]
  2447. loop_tokens = sd.loop_tokens_done_with_this_frame_ref
  2448. elif result_type == history.FLOAT:
  2449. exits = [exitbox]
  2450. loop_tokens = sd.loop_tokens_done_with_this_frame_float
  2451. else:
  2452. assert False
  2453. # FIXME: kill TerminatingLoopToken?
  2454. # FIXME: can we call compile_trace?
  2455. token = loop_tokens[0].finishdescr
  2456. self.history.record(rop.FINISH, exits, None, descr=token)
  2457. self.history.trace.done()
  2458. target_token = compile.compile_trace(self, self.resumekey, exits)
  2459. if target_token is not token:
  2460. compile.giveup()
  2461. def store_token_in_vable(self):
  2462. vinfo = self.jitdriver_sd.virtualizable_info
  2463. if vinfo is None:
  2464. return
  2465. vbox = self.virtualizable_boxes[-1]
  2466. if vbox is self.forced_virtualizable:
  2467. return # we already forced it by hand
  2468. # in case the force_token has not been recorded, record it here
  2469. # to make sure we know the virtualizable can be broken. However, the
  2470. # contents of the virtualizable should be generally correct
  2471. force_token = self.history.record(rop.FORCE_TOKEN, [],
  2472. lltype.nullptr(llmemory.GCREF.TO))
  2473. self.history.record(rop.SETFIELD_GC, [vbox, force_token],
  2474. None, descr=vinfo.vable_token_descr)
  2475. self.generate_guard(rop.GUARD_NOT_FORCED_2, None)
  2476. def compile_exit_frame_with_exception(self, valuebox):
  2477. self.store_token_in_vable()
  2478. sd = self.staticdata
  2479. token = sd.loop_tokens_exit_frame_with_exception_ref[0].finishdescr
  2480. self.history.record(rop.FINISH, [valuebox], None, descr=token)
  2481. self.history.trace.done()
  2482. target_token = compile.compile_trace(self, self.resumekey, [valuebox])
  2483. if target_token is not token:
  2484. compile.giveup()
  2485. @specialize.arg(1)
  2486. def initialize_original_boxes(self, jitdriver_sd, *args):
  2487. original_boxes = []
  2488. self._fill_original_boxes(jitdriver_sd, original_boxes,
  2489. jitdriver_sd.num_green_args, *args)
  2490. return original_boxes
  2491. @specialize.arg(1)
  2492. def _fill_original_boxes(self, jitdriver_sd, original_boxes,
  2493. num_green_args, *args):
  2494. if args:
  2495. from rpython.jit.metainterp.warmstate import wrap
  2496. box = wrap(self.cpu, args[0], num_green_args > 0)
  2497. original_boxes.append(box)
  2498. self._fill_original_boxes(jitdriver_sd, original_boxes,
  2499. num_green_args-1, *args[1:])
  2500. def initialize_state_from_start(self, original_boxes):
  2501. # ----- make a new frame -----
  2502. self.portal_call_depth = -1 # always one portal around
  2503. self.framestack = []
  2504. f = self.newframe(self.jitdriver_sd.mainjitcode)
  2505. f.setup_call(original_boxes)
  2506. assert self.portal_call_depth == 0
  2507. self.virtualref_boxes = []
  2508. self.initialize_withgreenfields(original_boxes)
  2509. self.initialize_virtualizable(original_boxes)
  2510. def initialize_state_from_guard_failure(self, resumedescr, deadframe):
  2511. # guard failure: rebuild a complete MIFrame stack
  2512. # This is stack-critical code: it must not be interrupted by StackOverflow,
  2513. # otherwise the jit_virtual_refs are left in a dangling state.
  2514. rstack._stack_criticalcode_start()
  2515. try:
  2516. self.portal_call_depth = -1 # always one portal around
  2517. self.history = history.History()
  2518. inputargs_and_holes = self.rebuild_state_after_failure(resumedescr,
  2519. deadframe)
  2520. return [box for box in inputargs_and_holes if box]
  2521. finally:
  2522. rstack._stack_criticalcode_stop()
  2523. def initialize_virtualizable(self, original_boxes):
  2524. vinfo = self.jitdriver_sd.virtualizable_info
  2525. if vinfo is not None:
  2526. index = (self.jitdriver_sd.num_green_args +
  2527. self.jitdriver_sd.index_of_virtualizable)
  2528. virtualizable_box = original_boxes[index]
  2529. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  2530. # First force the virtualizable if needed!
  2531. vinfo.clear_vable_token(virtualizable)
  2532. # The field 'virtualizable_boxes' is not even present
  2533. # if 'virtualizable_info' is None. Check for that first.
  2534. self.virtualizable_boxes = vinfo.read_boxes(self.cpu,
  2535. virtualizable)
  2536. original_boxes += self.virtualizable_boxes
  2537. self.virtualizable_boxes.append(virtualizable_box)
  2538. self.check_synchronized_virtualizable()
  2539. def initialize_withgreenfields(self, original_boxes):
  2540. ginfo = self.jitdriver_sd.greenfield_info
  2541. if ginfo is not None:
  2542. assert self.jitdriver_sd.virtualizable_info is None
  2543. index = (self.jitdriver_sd.num_green_args +
  2544. ginfo.red_index)
  2545. self.virtualizable_boxes = [original_boxes[index]]
  2546. def vable_and_vrefs_before_residual_call(self):
  2547. vrefinfo = self.staticdata.virtualref_info
  2548. for i in range(1, len(self.virtualref_boxes), 2):
  2549. vrefbox = self.virtualref_boxes[i]
  2550. vref = vrefbox.getref_base()
  2551. vrefinfo.tracing_before_residual_call(vref)
  2552. # the FORCE_TOKEN is already set at runtime in each vref when
  2553. # it is created, by optimizeopt.py.
  2554. #
  2555. vinfo = self.jitdriver_sd.virtualizable_info
  2556. if vinfo is not None:
  2557. virtualizable_box = self.virtualizable_boxes[-1]
  2558. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  2559. vinfo.tracing_before_residual_call(virtualizable)
  2560. #
  2561. force_token = self.history.record(rop.FORCE_TOKEN, [],
  2562. lltype.nullptr(llmemory.GCREF.TO))
  2563. self.history.record(rop.SETFIELD_GC, [virtualizable_box,
  2564. force_token],
  2565. None, descr=vinfo.vable_token_descr)
  2566. def vrefs_after_residual_call(self):
  2567. vrefinfo = self.staticdata.virtualref_info
  2568. for i in range(0, len(self.virtualref_boxes), 2):
  2569. vrefbox = self.virtualref_boxes[i+1]
  2570. vref = vrefbox.getref_base()
  2571. if vrefinfo.tracing_after_residual_call(vref):
  2572. # this vref was really a virtual_ref, but it escaped
  2573. # during this CALL_MAY_FORCE. Mark this fact by
  2574. # generating a VIRTUAL_REF_FINISH on it and replacing
  2575. # it by ConstPtr(NULL).
  2576. self.stop_tracking_virtualref(i)
  2577. def vable_after_residual_call(self, funcbox):
  2578. vinfo = self.jitdriver_sd.virtualizable_info
  2579. if vinfo is not None:
  2580. virtualizable_box = self.virtualizable_boxes[-1]
  2581. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  2582. if vinfo.tracing_after_residual_call(virtualizable):
  2583. # the virtualizable escaped during CALL_MAY_FORCE.
  2584. self.load_fields_from_virtualizable()
  2585. target_name = self.staticdata.get_name_from_address(funcbox.getaddr())
  2586. if target_name:
  2587. target_name = "ConstClass(%s)" % target_name
  2588. else:
  2589. target_name = str(funcbox.getaddr())
  2590. debug_print('vable escaped during a call in %s to %s' % (
  2591. self.framestack[-1].jitcode.name, target_name
  2592. ))
  2593. raise SwitchToBlackhole(Counters.ABORT_ESCAPE,
  2594. raising_exception=True)
  2595. # ^^^ we set 'raising_exception' to True because we must still
  2596. # have the eventual exception raised (this is normally done
  2597. # after the call to vable_after_residual_call()).
  2598. def stop_tracking_virtualref(self, i):
  2599. virtualbox = self.virtualref_boxes[i]
  2600. vrefbox = self.virtualref_boxes[i+1]
  2601. # record VIRTUAL_REF_FINISH here, which is before the actual
  2602. # CALL_xxx is recorded
  2603. self.history.record(rop.VIRTUAL_REF_FINISH, [vrefbox, virtualbox], None)
  2604. # mark this situation by replacing the vrefbox with ConstPtr(NULL)
  2605. self.virtualref_boxes[i+1] = self.cpu.ts.CONST_NULL
  2606. def handle_possible_exception(self):
  2607. if self.last_exc_value:
  2608. exception_box = ConstInt(heaptracker.adr2int(
  2609. llmemory.cast_ptr_to_adr(self.last_exc_value.typeptr)))
  2610. op = self.generate_guard(rop.GUARD_EXCEPTION,
  2611. None, [exception_box])
  2612. val = lltype.cast_opaque_ptr(llmemory.GCREF, self.last_exc_value)
  2613. if self.class_of_last_exc_is_const:
  2614. self.last_exc_box = ConstPtr(val)
  2615. else:
  2616. self.last_exc_box = op
  2617. op.setref_base(val)
  2618. assert op is not None
  2619. self.class_of_last_exc_is_const = True
  2620. self.finishframe_exception()
  2621. else:
  2622. self.generate_guard(rop.GUARD_NO_EXCEPTION, None, [])
  2623. def assert_no_exception(self):
  2624. assert not self.last_exc_value
  2625. def rebuild_state_after_failure(self, resumedescr, deadframe):
  2626. vinfo = self.jitdriver_sd.virtualizable_info
  2627. ginfo = self.jitdriver_sd.greenfield_info
  2628. self.framestack = []
  2629. boxlists = resume.rebuild_from_resumedata(self, resumedescr, deadframe,
  2630. vinfo, ginfo)
  2631. inputargs_and_holes, virtualizable_boxes, virtualref_boxes = boxlists
  2632. #
  2633. # virtual refs: make the vrefs point to the freshly allocated virtuals
  2634. self.virtualref_boxes = virtualref_boxes
  2635. vrefinfo = self.staticdata.virtualref_info
  2636. for i in range(0, len(virtualref_boxes), 2):
  2637. virtualbox = virtualref_boxes[i]
  2638. vrefbox = virtualref_boxes[i+1]
  2639. vrefinfo.continue_tracing(vrefbox.getref_base(),
  2640. virtualbox.getref_base())
  2641. #
  2642. # virtualizable: synchronize the real virtualizable and the local
  2643. # boxes, in whichever direction is appropriate
  2644. if vinfo is not None:
  2645. self.virtualizable_boxes = virtualizable_boxes
  2646. # just jumped away from assembler (case 4 in the comment in
  2647. # virtualizable.py) into tracing (case 2); if we get the
  2648. # virtualizable from somewhere strange it might not be forced,
  2649. # do it
  2650. virtualizable_box = self.virtualizable_boxes[-1]
  2651. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  2652. if vinfo.is_token_nonnull_gcref(virtualizable):
  2653. vinfo.reset_token_gcref(virtualizable)
  2654. # fill the virtualizable with the local boxes
  2655. self.synchronize_virtualizable()
  2656. #
  2657. elif self.jitdriver_sd.greenfield_info:
  2658. self.virtualizable_boxes = virtualizable_boxes
  2659. else:
  2660. assert not virtualizable_boxes
  2661. #
  2662. return inputargs_and_holes
  2663. def check_synchronized_virtualizable(self):
  2664. if not we_are_translated():
  2665. vinfo = self.jitdriver_sd.virtualizable_info
  2666. virtualizable_box = self.virtualizable_boxes[-1]
  2667. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  2668. vinfo.check_boxes(virtualizable, self.virtualizable_boxes)
  2669. def synchronize_virtualizable(self):
  2670. vinfo = self.jitdriver_sd.virtualizable_info
  2671. virtualizable_box = self.virtualizable_boxes[-1]
  2672. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  2673. vinfo.write_boxes(virtualizable, self.virtualizable_boxes)
  2674. def load_fields_from_virtualizable(self):
  2675. # Force a reload of the virtualizable fields into the local
  2676. # boxes (called only in escaping cases). Only call this function
  2677. # just before SwitchToBlackhole.
  2678. vinfo = self.jitdriver_sd.virtualizable_info
  2679. if vinfo is not None:
  2680. virtualizable_box = self.virtualizable_boxes[-1]
  2681. virtualizable = vinfo.unwrap_virtualizable_box(virtualizable_box)
  2682. self.virtualizable_boxes = vinfo.read_boxes(self.cpu,
  2683. virtualizable)
  2684. self.virtualizable_boxes.append(virtualizable_box)
  2685. def gen_store_back_in_vable(self, box):
  2686. vinfo = self.jitdriver_sd.virtualizable_info
  2687. if vinfo is not None:
  2688. # xxx only write back the fields really modified
  2689. vbox = self.virtualizable_boxes[-1]
  2690. if vbox is not box:
  2691. # ignore the hint on non-standard virtualizable
  2692. # specifically, ignore it on a virtual
  2693. return
  2694. if self.forced_virtualizable is not None:
  2695. # this can happen only in strange cases, but we don't care
  2696. # it was already forced
  2697. return
  2698. self.forced_virtualizable = vbox
  2699. for i in range(vinfo.num_static_extra_boxes):
  2700. fieldbox = self.virtualizable_boxes[i]
  2701. descr = vinfo.static_field_descrs[i]
  2702. self.execute_and_record(rop.SETFIELD_GC, descr, vbox, fieldbox)
  2703. i = vinfo.num_static_extra_boxes
  2704. virtualizable = vinfo.unwrap_virtualizable_box(vbox)
  2705. for k in range(vinfo.num_arrays):
  2706. descr = vinfo.array_field_descrs[k]
  2707. abox = self.execute_and_record(rop.GETFIELD_GC_R, descr, vbox)
  2708. descr = vinfo.array_descrs[k]
  2709. for j in range(vinfo.get_array_length(virtualizable, k)):
  2710. itembox = self.virtualizable_boxes[i]
  2711. i += 1
  2712. self.execute_and_record(rop.SETARRAYITEM_GC, descr,
  2713. abox, ConstInt(j), itembox)
  2714. assert i + 1 == len(self.virtualizable_boxes)
  2715. # we're during tracing, so we should not execute it
  2716. self.history.record(rop.SETFIELD_GC, [vbox, self.cpu.ts.CONST_NULL],
  2717. None, descr=vinfo.vable_token_descr)
  2718. def replace_box(self, oldbox, newbox):
  2719. for frame in self.framestack:
  2720. frame.replace_active_box_in_frame(oldbox, newbox)
  2721. boxes = self.virtualref_boxes
  2722. for i in range(len(boxes)):
  2723. if boxes[i] is oldbox:
  2724. boxes[i] = newbox
  2725. if (self.jitdriver_sd.virtualizable_info is not None or
  2726. self.jitdriver_sd.greenfield_info is not None):
  2727. boxes = self.virtualizable_boxes
  2728. for i in range(len(boxes)):
  2729. if boxes[i] is oldbox:
  2730. boxes[i] = newbox
  2731. self.heapcache.replace_box(oldbox, newbox)
  2732. def find_biggest_function(self):
  2733. start_stack = []
  2734. max_size = 0
  2735. max_key = None
  2736. max_jdsd = None
  2737. r = ''
  2738. debug_start("jit-abort-longest-function")
  2739. for elem in self.portal_trace_positions:
  2740. jitdriver_sd, key, pos = elem
  2741. if key is not None:
  2742. start_stack.append(elem)
  2743. else:
  2744. jitdriver_sd, greenkey, startpos = start_stack.pop()
  2745. warmstate = jitdriver_sd.warmstate
  2746. size = pos[0] - startpos[0]
  2747. if size > max_size:
  2748. if warmstate is not None:
  2749. r = warmstate.get_location_str(greenkey)
  2750. debug_print("found new longest: %s %d" % (r, size))
  2751. max_size = size
  2752. max_jdsd = jitdriver_sd
  2753. max_key = greenkey
  2754. if start_stack:
  2755. jitdriver_sd, key, pos = start_stack[0]
  2756. warmstate = jitdriver_sd.warmstate
  2757. size = self.history.get_trace_position()[0] - pos[0]
  2758. if size > max_size:
  2759. if warmstate is not None:
  2760. r = warmstate.get_location_str(key)
  2761. debug_print("found new longest: %s %d" % (r, size))
  2762. max_size = size
  2763. max_jdsd = jitdriver_sd
  2764. max_key = key
  2765. if self.portal_trace_positions: # tests
  2766. self.staticdata.logger_ops.log_abort_loop(self.history.trace,
  2767. self.box_names_memo)
  2768. debug_stop("jit-abort-longest-function")
  2769. return max_jdsd, max_key
  2770. def record_result_of_call_pure(self, op, argboxes, descr, patch_pos):
  2771. """ Patch a CALL into a CALL_PURE.
  2772. """
  2773. resbox_as_const = executor.constant_from_op(op)
  2774. for argbox in argboxes:
  2775. if not isinstance(argbox, Const):
  2776. break
  2777. else:
  2778. # all-constants: remove the CALL operation now and propagate a
  2779. # constant result
  2780. self.history.cut(patch_pos)
  2781. return resbox_as_const
  2782. # not all constants (so far): turn CALL into CALL_PURE, which might
  2783. # be either removed later by optimizeopt or turned back into CALL.
  2784. arg_consts = [executor.constant_from_op(a) for a in argboxes]
  2785. self.call_pure_results[arg_consts] = resbox_as_const
  2786. opnum = OpHelpers.call_pure_for_descr(descr)
  2787. self.history.cut(patch_pos)
  2788. newop = self.history.record_nospec(opnum, argboxes, descr)
  2789. newop.copy_value_from(op)
  2790. return newop
  2791. def direct_call_may_force(self, argboxes, calldescr):
  2792. """ Common case: record in the history a CALL_MAY_FORCE with
  2793. 'c_result' as the result of that call. (The actual call has
  2794. already been done.)
  2795. """
  2796. opnum = rop.call_may_force_for_descr(calldescr)
  2797. return self.history.record_nospec(opnum, argboxes, calldescr)
  2798. def direct_assembler_call(self, arglist, calldescr, targetjitdriver_sd):
  2799. """ Record in the history a direct call to assembler for portal
  2800. entry point.
  2801. """
  2802. num_green_args = targetjitdriver_sd.num_green_args
  2803. greenargs = arglist[1:num_green_args+1]
  2804. args = arglist[num_green_args+1:]
  2805. assert len(args) == targetjitdriver_sd.num_red_args
  2806. warmrunnerstate = targetjitdriver_sd.warmstate
  2807. token = warmrunnerstate.get_assembler_token(greenargs)
  2808. opnum = OpHelpers.call_assembler_for_descr(calldescr)
  2809. op = self.history.record_nospec(opnum, args, descr=token)
  2810. #
  2811. # To fix an obscure issue, make sure the vable stays alive
  2812. # longer than the CALL_ASSEMBLER operation. We do it by
  2813. # inserting explicitly an extra KEEPALIVE operation.
  2814. jd = token.outermost_jitdriver_sd
  2815. if jd.index_of_virtualizable >= 0:
  2816. return args[jd.index_of_virtualizable], op
  2817. else:
  2818. return None, op
  2819. def direct_libffi_call(self, argboxes, orig_calldescr):
  2820. """Generate a direct call to C code using jit_ffi_call()
  2821. """
  2822. # an 'assert' that constant-folds away the rest of this function
  2823. # if the codewriter didn't produce any OS_LIBFFI_CALL at all.
  2824. assert self.staticdata.has_libffi_call
  2825. #
  2826. from rpython.rtyper.lltypesystem import llmemory
  2827. from rpython.rlib.jit_libffi import CIF_DESCRIPTION_P
  2828. from rpython.jit.backend.llsupport.ffisupport import get_arg_descr
  2829. #
  2830. box_cif_description = argboxes[1]
  2831. if not isinstance(box_cif_description, ConstInt):
  2832. return None # cannot be handled by direct_libffi_call()
  2833. cif_description = box_cif_description.getint()
  2834. cif_description = llmemory.cast_int_to_adr(cif_description)
  2835. cif_description = llmemory.cast_adr_to_ptr(cif_description,
  2836. CIF_DESCRIPTION_P)
  2837. extrainfo = orig_calldescr.get_extra_info()
  2838. calldescr = self.cpu.calldescrof_dynamic(cif_description, extrainfo)
  2839. if calldescr is None:
  2840. return None # cannot be handled by direct_libffi_call()
  2841. #
  2842. box_exchange_buffer = argboxes[3]
  2843. arg_boxes = []
  2844. for i in range(cif_description.nargs):
  2845. kind, descr, itemsize = get_arg_descr(self.cpu,
  2846. cif_description.atypes[i])
  2847. ofs = cif_description.exchange_args[i]
  2848. assert ofs % itemsize == 0 # alignment check
  2849. if kind == 'i':
  2850. box_arg = self.history.record(
  2851. rop.GETARRAYITEM_RAW_I,
  2852. [box_exchange_buffer,
  2853. ConstInt(ofs // itemsize)],
  2854. 0, descr)
  2855. elif kind == 'f':
  2856. box_arg = self.history.record(
  2857. rop.GETARRAYITEM_RAW_F,
  2858. [box_exchange_buffer,
  2859. ConstInt(ofs // itemsize)],
  2860. longlong.ZEROF, descr)
  2861. else:
  2862. assert kind == 'v'
  2863. continue
  2864. arg_boxes.append(box_arg)
  2865. #
  2866. # for now, any call via libffi saves and restores everything
  2867. # (that is, errno and SetLastError/GetLastError on Windows)
  2868. # Note these flags match the ones in clibffi.ll_callback
  2869. c_saveall = ConstInt(rffi.RFFI_ERR_ALL | rffi.RFFI_ALT_ERRNO)
  2870. opnum = rop.call_release_gil_for_descr(orig_calldescr)
  2871. assert opnum == rop.call_release_gil_for_descr(calldescr)
  2872. return self.history.record_nospec(opnum,
  2873. [c_saveall, argboxes[2]] + arg_boxes,
  2874. calldescr)
  2875. # note that the result is written back to the exchange_buffer by the
  2876. # following operation, which should be a raw_store
  2877. def direct_call_release_gil(self, argboxes, calldescr):
  2878. if not we_are_translated(): # for llgraph
  2879. calldescr._original_func_ = argboxes[0].getint()
  2880. effectinfo = calldescr.get_extra_info()
  2881. realfuncaddr, saveerr = effectinfo.call_release_gil_target
  2882. funcbox = ConstInt(heaptracker.adr2int(realfuncaddr))
  2883. savebox = ConstInt(saveerr)
  2884. opnum = rop.call_release_gil_for_descr(calldescr)
  2885. return self.history.record_nospec(opnum,
  2886. [savebox, funcbox] + argboxes[1:],
  2887. calldescr)
  2888. def do_not_in_trace_call(self, allboxes, descr):
  2889. self.clear_exception()
  2890. executor.execute_varargs(self.cpu, self, rop.CALL_N,
  2891. allboxes, descr)
  2892. if self.last_exc_value:
  2893. # cannot trace this! it raises, so we have to follow the
  2894. # exception-catching path, but the trace doesn't contain
  2895. # the call at all
  2896. raise SwitchToBlackhole(Counters.ABORT_ESCAPE,
  2897. raising_exception=True)
  2898. return None
  2899. # ____________________________________________________________
  2900. class ChangeFrame(jitexc.JitException):
  2901. """Raised after we mutated metainterp.framestack, in order to force
  2902. it to reload the current top-of-stack frame that gets interpreted."""
  2903. # ____________________________________________________________
  2904. def _get_opimpl_method(name, argcodes):
  2905. from rpython.jit.metainterp.blackhole import signedord
  2906. #
  2907. def handler(self, position):
  2908. assert position >= 0
  2909. args = ()
  2910. next_argcode = 0
  2911. code = self.bytecode
  2912. orgpc = position
  2913. position += 1
  2914. for argtype in argtypes:
  2915. if argtype == "box": # a box, of whatever type
  2916. argcode = argcodes[next_argcode]
  2917. next_argcode = next_argcode + 1
  2918. if argcode == 'i':
  2919. value = self.registers_i[ord(code[position])]
  2920. elif argcode == 'c':
  2921. value = ConstInt(signedord(code[position]))
  2922. elif argcode == 'r':
  2923. value = self.registers_r[ord(code[position])]
  2924. elif argcode == 'f':
  2925. value = self.registers_f[ord(code[position])]
  2926. else:
  2927. raise AssertionError("bad argcode")
  2928. position += 1
  2929. elif argtype == "descr" or argtype == "jitcode":
  2930. assert argcodes[next_argcode] == 'd'
  2931. next_argcode = next_argcode + 1
  2932. index = ord(code[position]) | (ord(code[position+1])<<8)
  2933. value = self.metainterp.staticdata.opcode_descrs[index]
  2934. if argtype == "jitcode":
  2935. assert isinstance(value, JitCode)
  2936. position += 2
  2937. elif argtype == "label":
  2938. assert argcodes[next_argcode] == 'L'
  2939. next_argcode = next_argcode + 1
  2940. value = ord(code[position]) | (ord(code[position+1])<<8)
  2941. position += 2
  2942. elif argtype == "boxes": # a list of boxes of some type
  2943. length = ord(code[position])
  2944. value = [None] * length
  2945. self.prepare_list_of_boxes(value, 0, position,
  2946. argcodes[next_argcode])
  2947. next_argcode = next_argcode + 1
  2948. position += 1 + length
  2949. elif argtype == "boxes2": # two lists of boxes merged into one
  2950. length1 = ord(code[position])
  2951. position2 = position + 1 + length1
  2952. length2 = ord(code[position2])
  2953. value = [None] * (length1 + length2)
  2954. self.prepare_list_of_boxes(value, 0, position,
  2955. argcodes[next_argcode])
  2956. self.prepare_list_of_boxes(value, length1, position2,
  2957. argcodes[next_argcode + 1])
  2958. next_argcode = next_argcode + 2
  2959. position = position2 + 1 + length2
  2960. elif argtype == "boxes3": # three lists of boxes merged into one
  2961. length1 = ord(code[position])
  2962. position2 = position + 1 + length1
  2963. length2 = ord(code[position2])
  2964. position3 = position2 + 1 + length2
  2965. length3 = ord(code[position3])
  2966. value = [None] * (length1 + length2 + length3)
  2967. self.prepare_list_of_boxes(value, 0, position,
  2968. argcodes[next_argcode])
  2969. self.prepare_list_of_boxes(value, length1, position2,
  2970. argcodes[next_argcode + 1])
  2971. self.prepare_list_of_boxes(value, length1 + length2, position3,
  2972. argcodes[next_argcode + 2])
  2973. next_argcode = next_argcode + 3
  2974. position = position3 + 1 + length3
  2975. elif argtype == "orgpc":
  2976. value = orgpc
  2977. elif argtype == "int":
  2978. argcode = argcodes[next_argcode]
  2979. next_argcode = next_argcode + 1
  2980. if argcode == 'i':
  2981. value = self.registers_i[ord(code[position])].getint()
  2982. elif argcode == 'c':
  2983. value = signedord(code[position])
  2984. else:
  2985. raise AssertionError("bad argcode")
  2986. position += 1
  2987. elif argtype == "jitcode_position":
  2988. value = position
  2989. else:
  2990. raise AssertionError("bad argtype: %r" % (argtype,))
  2991. args += (value,)
  2992. #
  2993. num_return_args = len(argcodes) - next_argcode
  2994. assert num_return_args == 0 or num_return_args == 2
  2995. if num_return_args:
  2996. # Save the type of the resulting box. This is needed if there is
  2997. # a get_list_of_active_boxes(). See comments there.
  2998. self._result_argcode = argcodes[next_argcode + 1]
  2999. position += 1
  3000. else:
  3001. self._result_argcode = 'v'
  3002. self.pc = position
  3003. #
  3004. if not we_are_translated():
  3005. if self.debug:
  3006. print '\tpyjitpl: %s(%s)' % (name, ', '.join(map(repr, args))),
  3007. try:
  3008. resultbox = unboundmethod(self, *args)
  3009. except Exception as e:
  3010. if self.debug:
  3011. print '-> %s!' % e.__class__.__name__
  3012. raise
  3013. if num_return_args == 0:
  3014. if self.debug:
  3015. print
  3016. assert resultbox is None
  3017. else:
  3018. if self.debug:
  3019. print '-> %r' % (resultbox,)
  3020. assert argcodes[next_argcode] == '>'
  3021. result_argcode = argcodes[next_argcode + 1]
  3022. if 'ovf' not in name:
  3023. assert resultbox.type == {'i': history.INT,
  3024. 'r': history.REF,
  3025. 'f': history.FLOAT}[result_argcode]
  3026. else:
  3027. resultbox = unboundmethod(self, *args)
  3028. #
  3029. if resultbox is not None:
  3030. self.make_result_of_lastop(resultbox)
  3031. elif not we_are_translated():
  3032. assert self._result_argcode in 'v?' or 'ovf' in name
  3033. #
  3034. unboundmethod = getattr(MIFrame, 'opimpl_' + name).im_func
  3035. argtypes = unrolling_iterable(unboundmethod.argtypes)
  3036. handler.func_name = 'handler_' + name
  3037. return handler
  3038. def put_back_list_of_boxes3(frame, position, newvalue):
  3039. code = frame.bytecode
  3040. length1 = ord(code[position])
  3041. position2 = position + 1 + length1
  3042. length2 = ord(code[position2])
  3043. position3 = position2 + 1 + length2
  3044. length3 = ord(code[position3])
  3045. assert len(newvalue) == length1 + length2 + length3
  3046. frame._put_back_list_of_boxes(newvalue, 0, position)
  3047. frame._put_back_list_of_boxes(newvalue, length1, position2)
  3048. frame._put_back_list_of_boxes(newvalue, length1 + length2, position3)