NvInferRuntimeCommon.h

前往此文件的文档。

/*
 * SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. 保留所有权利。
 * SPDX-License-Identifier: Apache-2.0
 *
 * 根据 Apache 许可证 2.0 版本（“许可证”）获得许可；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可以在以下位置获得许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则
 * 根据许可证分发的软件按“现状”基础分发，
 * 不附带任何种类的明示或暗示的保证或条件。
 * 有关管理权限和限制的具体语言，请参阅许可证。
 * 许可证下的限制。
 */
 
#ifndef NV_INFER_RUNTIME_COMMON_H
#define NV_INFER_RUNTIME_COMMON_H
 
#define NV_INFER_INTERNAL_INCLUDE 1
#include "NvInferPluginBase.h"
#undef NV_INFER_INTERNAL_INCLUDE
#include "NvInferRuntimePlugin.h"
 
namespace nvinfer1
{
class IPluginRegistry
{
public
 using PluginLibraryHandle = void*;
 
 TRT_DEPRECATED virtual bool registerCreator(
 IPluginCreator& creator, AsciiChar const* const pluginNamespace) noexcept = 0;
 
 TRT_DEPRECATED virtual IPluginCreator* const* getPluginCreatorList(int32_t* const numCreators) const noexcept = 0;
 
 TRT_DEPRECATED virtual IPluginCreator* getPluginCreator(AsciiChar const* const pluginName,
 AsciiChar const* const pluginVersion, AsciiChar const* const pluginNamespace = "") noexcept = 0;
 
 // @cond SuppressDoxyWarnings
 IPluginRegistry() = default;
 IPluginRegistry(IPluginRegistry const&) = delete;
 IPluginRegistry(IPluginRegistry&&) = delete;
 IPluginRegistry& operator=(IPluginRegistry const&) & = delete;
 IPluginRegistry& operator=(IPluginRegistry&&) & = delete;
 // @endcond
 
protected
 virtual ~IPluginRegistry() noexcept = default;
 
public
 virtual void setErrorRecorder(IErrorRecorder* const recorder) noexcept = 0;
 
 virtual IErrorRecorder* getErrorRecorder() const noexcept = 0;
 
 TRT_DEPRECATED virtual bool deregisterCreator(IPluginCreator const& creator) noexcept = 0;
 
 virtual bool isParentSearchEnabled() const = 0;
 
 virtual void setParentSearchEnabled(bool const enabled) = 0;
 
 virtual PluginLibraryHandle loadLibrary(AsciiChar const* pluginPath) noexcept = 0;
 
 virtual void deregisterLibrary(PluginLibraryHandle handle) noexcept = 0;
 
 virtual bool registerCreator(IPluginCreatorInterface& creator, AsciiChar const* const pluginNamespace) noexcept = 0;
 
 virtual IPluginCreatorInterface* const* getAllCreators(int32_t* const numCreators) const noexcept = 0;
 
 virtual IPluginCreatorInterface* getCreator(AsciiChar const* const pluginName, AsciiChar const* const pluginVersion,
 AsciiChar const* const pluginNamespace = "") noexcept = 0;
 
 virtual bool deregisterCreator(IPluginCreatorInterface const& creator) noexcept = 0;
 
 virtual IPluginResource* acquirePluginResource(AsciiChar const* key, IPluginResource* resource) noexcept = 0;
 
 virtual int32_t releasePluginResource(AsciiChar const* key) noexcept = 0;
};
 
} // namespace nvinfer1
 
#endif /* NV_INFER_RUNTIME_COMMON_H */