Function wrapper

tensorrt_llm/_torch/utils.py:107–109

(self, *args, **kwargs)

    def decorator(func):

        def wrapper(self, *args, **kwargs):
            with model_extra_attrs(get_attrs(self)):
                return func(self, *args, **kwargs)

        return wrapper