def METHOD_NAME(*args, **kwargs):
    assert decorated_func.__name__[:6] == "track_"
    # Run the decorated benchmark within the added memory context
    # manager.
    with TrackAddedMemoryAllocation() as mb:
        decorated_func(*args, **kwargs)
    return mb.addedmem_mb()