Function wrapped

lite/pylite/test/test_tensor.py:15–17

(*args, **kwargs)

    @functools.wraps(func)
    def wrapped(*args, **kwargs):
        if LiteGlobal.get_device_count(LiteDeviceType.LITE_CUDA):
            return func(*args, **kwargs)

    return wrapped