NvInferConsistency.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 1993-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef NV_INFER_CONSISTENCY_CHECKER_H
#define NV_INFER_CONSISTENCY_CHECKER_H
#include "NvInferForwardDecl.h"
#include "NvInferPluginBase.h"
#include "NvInferSafeRuntime.h"
#include <vector>
 
 
namespace nvinfer2
{
namespace safe
{
// Forward decal
class ISafeRecorder;
 
namespace consistency
{
 
class IConsistencyChecker
{
public:
    virtual bool validate() noexcept = 0;
    virtual ~IConsistencyChecker() = default;
 
protected:
    IConsistencyChecker() = default;
    IConsistencyChecker(IConsistencyChecker const& other) = delete;
    IConsistencyChecker& operator=(IConsistencyChecker const& other) & = delete;
    IConsistencyChecker(IConsistencyChecker&& other) = delete;
    IConsistencyChecker& operator=(IConsistencyChecker&& other) & = delete;
};
 
class IPluginChecker
{
public:
    virtual bool validate(std::vector<TensorDescriptor> const& Inputs, std::vector<TensorDescriptor> const& Outputs,
        PluginFieldCollection* fc) noexcept = 0;
 
    virtual ~IPluginChecker() = default;
 
protected:
    IPluginChecker() = default;
    IPluginChecker(IPluginChecker const& other) = delete;
    IPluginChecker& operator=(IPluginChecker const& other) & = delete;
    IPluginChecker(IPluginChecker&& other) = delete;
    IPluginChecker& operator=(IPluginChecker&& other) & = delete;
};
 
extern "C" nvinfer2::safe::ErrorCode createConsistencyChecker(IConsistencyChecker*& icc,
    nvinfer2::safe::ISafeRecorder& recorder, void const* blob, size_t size,
    std::vector<std::string> const& pluginBuildLibs = {}) noexcept;
 
} // namespace consistency
 
} // namespace safe
} // namespace nvinfer2
 
#endif // NV_INFER_CONSISTENCY_CHECKER_H