def test_untranslated_storage_unaligned(monkeypatch): monkeypatch.setattr(rawstorage, 'misaligned_is_fine', False) r = alloc_raw_storage(15) raw_storage_setitem_unaligned(r, 3, 1 << 30) res = raw_storage_getitem_unaligned(lltype.Signed, r, 3) assert res == 1 << 30 raw_storage_setitem_unaligned(r, 3, 3.14) res = raw_storage_getitem_unaligned(lltype.Float, r, 3) assert res == 3.14 free_raw_storage(r)
def test_untranslated_storage_unaligned(monkeypatch): monkeypatch.setattr(rawstorage, 'misaligned_is_fine', False) r = alloc_raw_storage(15) raw_storage_setitem_unaligned(r, 3, 1<<30) res = raw_storage_getitem_unaligned(lltype.Signed, r, 3) assert res == 1<<30 raw_storage_setitem_unaligned(r, 3, 3.14) res = raw_storage_getitem_unaligned(lltype.Float, r, 3) assert res == 3.14 free_raw_storage(r)
def test_untranslated_storage(): r = alloc_raw_storage(37) raw_storage_setitem(r, 8, 1 << 30) res = raw_storage_getitem(lltype.Signed, r, 8) assert res == 1 << 30 raw_storage_setitem(r, 8, 3.14) res = raw_storage_getitem(lltype.Float, r, 8) assert res == 3.14 py.test.raises(AlignmentError, raw_storage_getitem, lltype.Signed, r, 3) py.test.raises(AlignmentError, raw_storage_setitem, r, 3, 42.5) free_raw_storage(r)
def test_untranslated_storage(): r = alloc_raw_storage(37) raw_storage_setitem(r, 8, 1<<30) res = raw_storage_getitem(lltype.Signed, r, 8) assert res == 1<<30 raw_storage_setitem(r, 8, 3.14) res = raw_storage_getitem(lltype.Float, r, 8) assert res == 3.14 py.test.raises(AlignmentError, raw_storage_getitem, lltype.Signed, r, 3) py.test.raises(AlignmentError, raw_storage_setitem, r, 3, 42.5) free_raw_storage(r)
def test_float_int32_casts(self): myjitdriver = JitDriver(greens = [], reds = 'auto', vectorize=True) def f(bytecount, va, vb, vc): i = 0 j = 0 while j < bytecount: myjitdriver.jit_merge_point() a = raw_storage_getitem(rffi.DOUBLE,va,j) b = raw_storage_getitem(rffi.INT,vb,i) c = a+rffi.cast(rffi.DOUBLE,b) raw_storage_setitem(vc, j, c) i += 4 j += 8 count = 32 va = alloc_raw_storage(8*count, zero=True) vb = alloc_raw_storage(4*count, zero=True) for i,v in enumerate([1.0,2.0,3.0,4.0]*(count/4)): raw_storage_setitem(va, i*8, rffi.cast(rffi.DOUBLE,v)) for i,v in enumerate([-1,-2,-3,-4]*(count/4)): raw_storage_setitem(vb, i*4, rffi.cast(rffi.INT,v)) vc = alloc_raw_storage(8*count, zero=True) self.meta_interp(f, [8*count, va, vb, vc], vec=True) for i in range(count): assert raw_storage_getitem(rffi.DOUBLE,vc,i*8) == 0.0 free_raw_storage(va) free_raw_storage(vb) free_raw_storage(vc)
def f(d): bc = d * rffi.sizeof(rffi.SIGNED) va = alloc_raw_storage(bc, zero=True) vb = alloc_raw_storage(bc, zero=True) vc = alloc_raw_storage(bc, zero=True) x = 1 for i in range(d): j = i * rffi.sizeof(rffi.SIGNED) raw_storage_setitem(va, j, rffi.cast(rffi.SIGNED, i)) raw_storage_setitem(vb, j, rffi.cast(rffi.SIGNED, i)) i = 0 while i < bc: myjitdriver.jit_merge_point() a = raw_storage_getitem(rffi.SIGNED, va, i) b = raw_storage_getitem(rffi.SIGNED, vb, i) c = a + b raw_storage_setitem(vc, i, rffi.cast(rffi.SIGNED, c)) i += 1 * rffi.sizeof(rffi.SIGNED) res = 0 for i in range(d): res += raw_storage_getitem(rffi.SIGNED, vc, i * rffi.sizeof(rffi.SIGNED)) free_raw_storage(va) free_raw_storage(vb) free_raw_storage(vc) return res
def test_float_int32_casts(self): myjitdriver = JitDriver(greens=[], reds='auto', vectorize=True) def f(bytecount, va, vb, vc): i = 0 j = 0 while j < bytecount: myjitdriver.jit_merge_point() a = raw_storage_getitem(rffi.DOUBLE, va, j) b = raw_storage_getitem(rffi.INT, vb, i) c = a + rffi.cast(rffi.DOUBLE, b) raw_storage_setitem(vc, j, c) i += 4 j += 8 count = 32 va = alloc_raw_storage(8 * count, zero=True) vb = alloc_raw_storage(4 * count, zero=True) for i, v in enumerate([1.0, 2.0, 3.0, 4.0] * (count / 4)): raw_storage_setitem(va, i * 8, rffi.cast(rffi.DOUBLE, v)) for i, v in enumerate([-1, -2, -3, -4] * (count / 4)): raw_storage_setitem(vb, i * 4, rffi.cast(rffi.INT, v)) vc = alloc_raw_storage(8 * count, zero=True) self.meta_interp(f, [8 * count, va, vb, vc], vec=True) for i in range(count): assert raw_storage_getitem(rffi.DOUBLE, vc, i * 8) == 0.0 free_raw_storage(va) free_raw_storage(vb) free_raw_storage(vc)
def f(): p = alloc_raw_storage(15) raw_storage_setitem(p, 3, 24) res = raw_storage_getitem(lltype.Signed, p, 3) free_raw_storage(p) return res
def __del__(self): free_raw_storage(self.storage)
def __del__(self): if self.gcstruct: self.gcstruct.length = 0 free_raw_storage(self.storage, track_allocation=False)
def __del__(self): free_raw_storage(self.indexes, track_allocation=False) free_raw_storage(self.values, track_allocation=False)
def f(v): r = alloc_raw_storage(24) raw_storage_setitem_unaligned(r, 3, v) res = raw_storage_getitem_unaligned(lltype.Float, r, 3) free_raw_storage(r) return res != v
def freeBuf(buf): free_raw_storage(buf.c_base) free(buf)
def f(): p = alloc_raw_storage(15, track_allocation=False, zero=True) raw_storage_setitem(p, 3, 24) res = raw_storage_getitem(lltype.Signed, p, 3) free_raw_storage(p, track_allocation=False) return res
def __del__(self): free_raw_storage(self.storage, track_allocation=False)
def clear(self): while self.arrays: array = self.arrays.pop() free_raw_storage(array)
def f(): p = alloc_raw_storage(15) raw_storage_setitem(p, 4, 2.4e15) res = raw_storage_getitem(lltype.Float, p, 4) free_raw_storage(p) return res
def f(): p = alloc_raw_storage(15) raw_storage_setitem(p, 5, rffi.cast(rffi.UCHAR, 254)) res = raw_storage_getitem(rffi.UCHAR, p, 5) free_raw_storage(p) return rffi.cast(lltype.Signed, res)
def test_untranslated_storage(): r = alloc_raw_storage(15) raw_storage_setitem(r, 3, 1<<30) res = raw_storage_getitem(lltype.Signed, r, 3) free_raw_storage(r) assert res == 1<<30
def f(i): r = alloc_raw_storage(24) raw_storage_setitem(r, 8, i) res = raw_storage_getitem(lltype.Signed, r, 8) free_raw_storage(r) return res != i