Python external_func_call Examples

Example #1

File: demo_external_func.py Project: samuela/taichi

def call_ext() -> ti.i32:
    a = 2.0
    b = 3.0
    c = 0.0
    d = 0.0
    e = 3
    ti.external_func_call(func=so.add_and_mul, args=(a, b), outputs=(c, d, e))
    p = 0
    ti.external_func_call(func=so.pow_int, args=(int(c + d), e), outputs=(p, ))
    return p

Example #2

File: demo_external_func.py Project: samuela/taichi

def pow_int_wrapper(a, b):
    p = 0
    ti.external_func_call(func=so.pow_int,
                          args=(int(a), int(b)),
                          outputs=(p, ))
    return p