Class EnqueueSafe

samples/common/sampleInference.cpp:637–663

\class EnqueueSafe \brief Functor to enqueue safe execution context

//! \brief Functor to enqueue safe execution context
//!
class EnqueueSafe
{
public:
    explicit EnqueueSafe(nvinfer1::safe::IExecutionContext& context, Bindings const& bindings)
        : mContext(context)
        , mBindings(bindings)
    {
        ASSERT(mBindings.setSafeTensorAddresses(mContext));
    }

    bool operator()(TrtCudaStream& stream) const
    {
        try
        {
            return mContext.enqueueV3(stream.get());
        }
        catch (const std::exception&)
        {
            return false;
        }
        return false;
    }

    nvinfer1::safe::IExecutionContext& mContext;
private:
    Bindings const& mBindings;
};

using EnqueueFunction = std::function<bool(TrtCudaStream&)>;