Python insert_pure_function Examples

Example #1

def atan2_f32_impl(context, builder, sig, args):
    assert len(args) == 2
    mod = builder.module
    fnty = Type.function(Type.float(), [Type.float(), Type.float()])
    fn = cgutils.insert_pure_function(builder.module, fnty, name="atan2f")
    res = builder.call(fn, args)
    return impl_ret_untracked(context, builder, sig.return_type, res)

Example #2

File: mathimpl.py Project: MPOWER4RU/numba

def atan2_f32_impl(context, builder, sig, args):
    assert len(args) == 2
    mod = builder.module
    fnty = Type.function(Type.float(), [Type.float(), Type.float()])
    fn = cgutils.insert_pure_function(builder.module, fnty, name="atan2f")
    res = builder.call(fn, args)
    return impl_ret_untracked(context, builder, sig.return_type, res)

Example #3

File: mathimpl.py Project: kalatestimine/numba

def ldexp_impl(context, builder, sig, args):
    val, exp = args
    fltty, intty = map(context.get_data_type, sig.args)
    fnty = Type.function(fltty, (fltty, intty))
    fname = {"float": "numba_ldexpf", "double": "numba_ldexp"}[str(fltty)]
    fn = cgutils.insert_pure_function(builder.module, fnty, name=fname)
    res = builder.call(fn, (val, exp))
    return impl_ret_untracked(context, builder, sig.return_type, res)

Example #4

File: mathimpl.py Project: chas-mellish/numba

def atan2_f64_impl(context, builder, sig, args):
    assert len(args) == 2
    mod = builder.module
    fnty = Type.function(Type.double(), [Type.double(), Type.double()])
    # Workaround atan2() issues under Windows
    fname = "atan2_fixed" if sys.platform == "win32" else "atan2"
    fn = cgutils.insert_pure_function(builder.module, fnty, name=fname)
    return builder.call(fn, args)

Example #5

File: mathimpl.py Project: CaptainAL/Spyder

def atan2_f64_impl(context, builder, sig, args):
    assert len(args) == 2
    mod = builder.module
    fnty = Type.function(Type.double(), [Type.double(), Type.double()])
    # Workaround atan2() issues under Windows
    fname = "atan2_fixed" if sys.platform == "win32" else "atan2"
    fn = cgutils.insert_pure_function(builder.module, fnty, name=fname)
    return builder.call(fn, args)

Example #6

File: mathimpl.py Project: MPOWER4RU/numba

 def implementer(context, builder, sig, args):
     [val] = args
     mod = builder.module
     lty = context.get_value_type(input_type)
     fnty = Type.function(lty, [lty])
     fn = cgutils.insert_pure_function(builder.module, fnty, name=extern_func)
     res = builder.call(fn, (val,))
     if restype is not None:
         res = context.cast(builder, res, input_type, restype)
     return impl_ret_untracked(context, builder, sig.return_type, res)

Example #7

File: mathimpl.py Project: chas-mellish/numba

def ldexp_impl(context, builder, sig, args):
    val, exp = args
    fltty, intty = map(context.get_data_type, sig.args)
    fnty = Type.function(fltty, (fltty, intty))
    fname = {
        "float": "numba_ldexpf",
        "double": "numba_ldexp",
    }[str(fltty)]
    fn = cgutils.insert_pure_function(builder.module, fnty, name=fname)
    return builder.call(fn, (val, exp))

Example #8

File: mathimpl.py Project: menghaozhu/hat

 def implementer(context, builder, sig, args):
     [val] = args
     mod = builder.module
     lty = context.get_value_type(input_type)
     fnty = Type.function(lty, [lty])
     fn = cgutils.insert_pure_function(builder.module, fnty, name=extern_func)
     res = builder.call(fn, (val,))
     if restype is not None:
         res = context.cast(builder, res, input_type, restype)
     return impl_ret_untracked(context, builder, sig.return_type, res)

Example #9

File: mathimpl.py Project: toddrme2178/numba

def atan2_float_impl(context, builder, sig, args):
    assert len(args) == 2
    mod = builder.module
    ty = sig.args[0]
    lty = context.get_value_type(ty)
    func_name = {
        types.float32: "atan2f",
        types.float64: "atan2"
        }[ty]
    fnty = Type.function(lty, (lty, lty))
    fn = cgutils.insert_pure_function(builder.module, fnty, name=func_name)
    res = builder.call(fn, args)
    return impl_ret_untracked(context, builder, sig.return_type, res)

Example #10

File: mathimpl.py Project: ChiLi90/LifetimeFit

def atan2_float_impl(context, builder, sig, args):
    assert len(args) == 2
    mod = builder.module
    ty = sig.args[0]
    lty = context.get_value_type(ty)
    func_name = {
        types.float32: "atan2f",
        # Workaround atan2() issues under Windows
        types.float64: "atan2_fixed" if sys.platform == "win32" else "atan2"
        }[ty]
    fnty = Type.function(lty, (lty, lty))
    fn = cgutils.insert_pure_function(builder.module, fnty, name=func_name)
    res = builder.call(fn, args)
    return impl_ret_untracked(context, builder, sig.return_type, res)

Example #11

File: mathimpl.py Project: esc/numba

def atan2_float_impl(context, builder, sig, args):
    assert len(args) == 2
    mod = builder.module
    ty = sig.args[0]
    lty = context.get_value_type(ty)
    func_name = {
        types.float32: "atan2f",
        # Workaround atan2() issues under Windows
        types.float64: "atan2_fixed" if sys.platform == "win32" else "atan2"
        }[ty]
    fnty = Type.function(lty, (lty, lty))
    fn = cgutils.insert_pure_function(builder.module, fnty, name=func_name)
    res = builder.call(fn, args)
    return impl_ret_untracked(context, builder, sig.return_type, res)

Example #12

File: mathimpl.py Project: ChiLi90/LifetimeFit

 def float_impl(context, builder, sig, args):
     """
     Implement *fn* for a types.Float input.
     """
     [val] = args
     mod = builder.module
     input_type = sig.args[0]
     lty = context.get_value_type(input_type)
     func_name = {
         types.float32: f32extern,
         types.float64: f64extern,
         }[input_type]
     fnty = Type.function(lty, [lty])
     fn = cgutils.insert_pure_function(builder.module, fnty, name=func_name)
     res = builder.call(fn, (val,))
     res = context.cast(builder, res, input_type, sig.return_type)
     return impl_ret_untracked(context, builder, sig.return_type, res)

Example #13

File: mathimpl.py Project: esc/numba

 def float_impl(context, builder, sig, args):
     """
     Implement *fn* for a types.Float input.
     """
     [val] = args
     mod = builder.module
     input_type = sig.args[0]
     lty = context.get_value_type(input_type)
     func_name = {
         types.float32: f32extern,
         types.float64: f64extern,
         }[input_type]
     fnty = Type.function(lty, [lty])
     fn = cgutils.insert_pure_function(builder.module, fnty, name=func_name)
     res = builder.call(fn, (val,))
     res = context.cast(builder, res, input_type, sig.return_type)
     return impl_ret_untracked(context, builder, sig.return_type, res)