Function cupy_wrapper

dali/test/python/test_dltensor_operator.py:209–213

(fun, synchronize)

def cupy_wrapper(fun, synchronize):
    if synchronize:
        return lambda in1, in2: cupy_adapter_sync(fun, in1, in2)
    else:
        return lambda in1, in2: cupy_adapter(fun, in1, in2)


def cupy_compare(fun, synchronize, pre1, pre2, post1, post2):