Function cupy_adapter_sync

dali/test/python/test_dltensor_operator.py:192–199

(fun, in1, in2)

def cupy_adapter_sync(fun, in1, in2):
    with cupy_stream:
        tin1 = [cupy.from_dlpack(dltensor) for dltensor in in1]
        tin2 = [cupy.from_dlpack(dltensor) for dltensor in in2]
        tout1, tout2 = fun(tin1, tin2)
        out1, out2 = [tout.__dlpack__() for tout in tout1], [tout.__dlpack__() for tout in tout2]
    cupy_stream.synchronize()
    return out1, out2


def cupy_adapter(fun, in1, in2):