diff options
Diffstat (limited to 'rpython/jit/metainterp/test/test_warmspot.py')
| -rw-r--r-- | rpython/jit/metainterp/test/test_warmspot.py | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/rpython/jit/metainterp/test/test_warmspot.py b/rpython/jit/metainterp/test/test_warmspot.py index a2a508b610..76e382339c 100644 --- a/rpython/jit/metainterp/test/test_warmspot.py +++ b/rpython/jit/metainterp/test/test_warmspot.py @@ -1,7 +1,7 @@ import py from rpython.jit.metainterp import jitexc from rpython.jit.metainterp.warmspot import get_stats -from rpython.rlib.jit import JitDriver, set_param, unroll_safe, jit_callback +from rpython.rlib.jit import JitDriver, set_param, unroll_safe, jit_callback, set_user_param from rpython.jit.backend.llgraph import runner from rpython.jit.metainterp.test.support import LLJitMixin @@ -575,6 +575,26 @@ class TestLLWarmspot(LLJitMixin): assert str(e.value) == ("there are multiple jit_merge_points " "with the same jitdriver") + def test_jit_off_returns_early(self): + from rpython.jit.metainterp.counter import DeterministicJitCounter + driver = JitDriver(greens = ['s'], reds = ['i'], name='jit') + + def loop(i, s): + set_user_param(driver, "off") + while i > s: + driver.jit_merge_point(i=i, s=s) + i -= 1 + + def main(s): + loop(30, s) + + fn = DeterministicJitCounter.lookup_chain + DeterministicJitCounter.lookup_chain = None + try: + self.meta_interp(main, [5]) # must not crash + finally: + DeterministicJitCounter.lookup_chain = fn + class TestWarmspotDirect(object): def setup_class(cls): |