Ejemplos de declare_instruction en Python

Ejemplo n.º 1

Archivo: math_utilities.py Proyecto: ASPP/numba

def py_modulo(restype, argtypes):
    if restype.is_float:
        def py_modulo(a, n):
            r = rem(a, n)

            if (r != 0) and (r < 0) ^ (n < 0):
                r += n

            return r

        instr = 'frem'
    else:
        assert restype.is_int

        def py_modulo(a, n):
            r = rem(a, n)

            if r != 0 and (r ^ n) < 0:
                r += n

            return r

        if restype.is_unsigned:
            instr = 'urem'
        else:
            instr = 'srem'

    rem = nb.declare_instruction(restype(restype, restype), instr)
    return nb.jit(restype(*argtypes))(py_modulo)

Ejemplo n.º 2

def py_modulo(restype, argtypes):
    if restype.is_float:

        def py_modulo(a, n):
            r = rem(a, n)

            if (r != 0) and (r < 0) ^ (n < 0):
                r += n

            return r

        instr = 'frem'
    else:
        assert restype.is_int

        def py_modulo(a, n):
            r = rem(a, n)

            if r != 0 and (r ^ n) < 0:
                r += n

            return r

        if restype.is_unsigned:
            instr = 'urem'
        else:
            instr = 'srem'

    rem = nb.declare_instruction(restype(restype, restype), instr)
    return nb.jit(restype(*argtypes))(py_modulo)

Ejemplo n.º 3

def test_intrinsics():
    intrin = declare_instruction(int32(int32, int32), 'srem')
    assert intrin(5, 3) == 2

Ejemplo n.º 4

Archivo: test_intrinsic.py Proyecto: ASPP/numba

def test_intrinsics():
    intrin = declare_instruction(int32(int32, int32), 'srem')
    assert intrin(5, 3) == 2