def __init__(self, rtyper, stats, opts, translate_support_code=False,
gcdescr=None):
assert type(opts) is not bool
self.opts = opts
from pypy.jit.backend.llsupport.gc import get_ll_description
AbstractCPU.__init__(self)
self.rtyper = rtyper
self.stats = stats
self.translate_support_code = translate_support_code
if translate_support_code:
translator = rtyper.annotator.translator
else:
translator = None
self.gc_ll_descr = get_ll_description(gcdescr, translator, rtyper)
if translator and translator.config.translation.gcremovetypeptr:
self.vtable_offset = None
else:
self.vtable_offset, _ = symbolic.get_field_token(rclass.OBJECT,
'typeptr',
translate_support_code)
self._setup_prebuilt_error('ovf', OverflowError)
self._setup_prebuilt_error('zer', ZeroDivisionError)
if translate_support_code:
self._setup_exception_handling_translated()
else:
self._setup_exception_handling_untranslated()
self.saved_exc_value = lltype.nullptr(llmemory.GCREF.TO)
self.asmmemmgr = AsmMemoryManager()
self.setup()
if translate_support_code:
self._setup_on_leave_jitted_translated()
else:
self._setup_on_leave_jitted_untranslated()
def test_get_index():
memmgr = AsmMemoryManager(min_fragment=8, num_indices=5)
index = 0
for length in range(100):
assert memmgr._get_index(length) == index
if length in [8, 11, 15, 21]:
index += 1
def test_get_index_default_values():
memmgr = AsmMemoryManager()
jumps = [64, 86, 115, 154, 206, 275, 367, 490, 654, 873, 1165,
1554, 2073, 2765, 3687, 4917, 6557, 8743, 11658, 15545,
20727, 27637, 36850, 49134, 65513, 87351, 116469, 155293,
207058, 276078, 368105]
for i, jump in enumerate(jumps):
assert memmgr._get_index(jump) == i
assert memmgr._get_index(jump + 1) == i + 1
def test_allocate_block():
memmgr = AsmMemoryManager(min_fragment=8, num_indices=5)
memmgr._add_free_block(10, 18)
memmgr._add_free_block(20, 30)
(start, stop) = memmgr._allocate_block(4)
assert (start, stop) == (10, 18)
assert memmgr.free_blocks == {20: 30}
assert memmgr.free_blocks_end == {30: 20}
assert memmgr.blocks_by_size == [[], [20], [], [], []]
(start, stop) = memmgr._allocate_block(4)
assert (start, stop) == (20, 30)
assert memmgr.free_blocks == {}
assert memmgr.free_blocks_end == {}
assert memmgr.blocks_by_size == [[], [], [], [], []]
def test_add_free_block():
memmgr = AsmMemoryManager(min_fragment=8, num_indices=5)
memmgr._add_free_block(10, 18)
assert memmgr.free_blocks == {10: 18}
assert memmgr.free_blocks_end == {18: 10}
assert memmgr.blocks_by_size == [[10], [], [], [], []]
memmgr._add_free_block(20, 30)
assert memmgr.free_blocks == {10: 18, 20: 30}
assert memmgr.free_blocks_end == {18: 10, 30: 20}
assert memmgr.blocks_by_size == [[10], [20], [], [], []]
memmgr._add_free_block(18, 20) # merge both left and right
assert memmgr.free_blocks == {10: 30}
assert memmgr.free_blocks_end == {30: 10}
assert memmgr.blocks_by_size == [[], [], [], [10], []]
def test_malloc_with_fragment():
for reqsize in range(1, 33):
memmgr = AsmMemoryManager(min_fragment=8, num_indices=5)
memmgr._add_free_block(12, 44)
(start, stop) = memmgr.malloc(reqsize, reqsize)
if reqsize + 8 <= 32:
assert (start, stop) == (12, 12 + reqsize)
assert memmgr.free_blocks == {stop: 44}
assert memmgr.free_blocks_end == {44: stop}
assert [stop] in memmgr.blocks_by_size
else:
assert (start, stop) == (12, 44)
assert memmgr.free_blocks == {}
assert memmgr.free_blocks_end == {}
assert memmgr.blocks_by_size == [[], [], [], [], []]
def test_malloc_without_fragment():
memmgr = AsmMemoryManager(min_fragment=8, num_indices=5)
memmgr._add_free_block(10, 18)
memmgr._add_free_block(20, 30)
for minsize in range(1, 11):
for maxsize in range(minsize, 14):
(start, stop) = memmgr.malloc(minsize, maxsize)
if minsize <= 8:
assert (start, stop) == (10, 18)
else:
assert (start, stop) == (20, 30)
memmgr._add_free_block(start, stop)
memmgr._add_free_block(40, 49)
(start, stop) = memmgr.malloc(10, 10)
assert (start, stop) == (20, 30)
def setup_method(self, _):
self.memmgr = AsmMemoryManager(min_fragment=8,
num_indices=10,
large_alloc_size=8192)
