1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
import weakref
from rpython.rlib.jit import JitDriver, dont_look_inside
from rpython.jit.backend.llsupport.test.zrpy_gc_test import run, get_entry, compile
class X(object):
def __init__(self, x=0):
self.x = x
next = None
class CheckError(Exception):
pass
def check(flag):
if not flag:
raise CheckError
def compile_and_run(f, gc, **kwds):
cbuilder = compile(f, gc, **kwds)
return run(cbuilder)
def get_g(main):
main._dont_inline_ = True
def g(name, n):
x = X()
x.foo = 2
main(n, x)
x.foo = 5
return weakref.ref(x)
g._dont_inline_ = True
return g
def compile_boehm_test():
myjitdriver = JitDriver(greens = [], reds = ['n', 'x'])
@dont_look_inside
def see(lst, n):
assert len(lst) == 3
assert lst[0] == n+10
assert lst[1] == n+20
assert lst[2] == n+30
def main(n, x):
while n > 0:
myjitdriver.can_enter_jit(n=n, x=x)
myjitdriver.jit_merge_point(n=n, x=x)
y = X()
y.foo = x.foo
n -= y.foo
see([n+10, n+20, n+30], n)
res = compile_and_run(get_entry(get_g(main)), "boehm", jit=True)
assert int(res) >= 16
|
