Safety Plugin Loader#
-
class SafetyPluginRegistration#
Public Functions
-
virtual ~SafetyPluginRegistration() noexcept = default#
- virtual bool initializeAttentionModules(
- bool remoteAutoTuning
Bind attention creators before parsing ONNX; remote tuning performs no local CUDA setup.
-
virtual bool completeGpuWork() noexcept = 0#
Complete profiling while TRT objects and their borrowed buffers are still alive.
-
virtual bool finishBuild() noexcept = 0#
Destroy all TRT objects first. Returns false if GPU completion or teardown fails.
-
virtual ~SafetyPluginRegistration() noexcept = default#
-
struct SafetyPluginDescriptor#
- std::unique_ptr<SafetyPluginRegistration> trt_edgellm::safety::builder::loadSafetyPlugins(
- std::string const &configuredPath,
- std::initializer_list<SafetyPluginDescriptor> requiredPlugins
Load the Safety plugin library and register the required creators. The returned object keeps the library loaded and deregisters creators before unloading it.