Function wrapper

tensorrt_llm/_torch/device_mesh.py:20–23

(self, *args, **kwargs)

    @wraps(func)
    def wrapper(self, *args, **kwargs):
        if DeviceMeshTopologyImpl.device_mesh is None:
            self.build_mesh()
        return func(self, *args, **kwargs)

    return wrapper