Function wrapper

imperative/python/megengine/logger.py:249–253

(*args, **kwargs)

    @wraps(func)
    def wrapper(*args, **kwargs):
        if _all_log_counter[hash_key] <= count:
            _call_count_logger.warning(msg)
            _all_log_counter[hash_key] += 1
        return func(*args, **kwargs)

    return wrapper