Method create_plugin

tensorrt_llm/python_plugin.py:492–502

(self, name, fc, phase)

        super().__init__()

    def create_plugin(self, name, fc, phase):
        if len(fc) == 1 and fc[0].name == "__plugin_pickle_obj__":
            data = fc[0].data
            plugin_dict = pickle.loads(data)  # nosec B301
            plugin = self.plugin_cls.__new__(self.plugin_cls)
            super(self.plugin_cls, plugin).__init__()
            plugin.__dict__.update(plugin_dict)
        else:
            raise RuntimeError("Expect to be called by TRT")
        plugin.plugin_phase = phase
        return plugin


def trtllm_plugin(