MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / EnqueueGraphSafe

Class EnqueueGraphSafe

samples/common/sampleInference.cpp:616–631  ·  view source on GitHub ↗

\class EnqueueGraphSafe \brief Functor to enqueue inference from CUDA Graph

Source from the content-addressed store, hash-verified

614//! \brief Functor to enqueue inference from CUDA Graph
615//!
616class EnqueueGraphSafe
617{
618
619public:
620 explicit EnqueueGraphSafe(TrtCudaGraph& graph)
621 : mGraph(graph)
622 {
623 }
624
625 bool operator()(TrtCudaStream& stream) const
626 {
627 return mGraph.launch(stream);
628 }
629
630 TrtCudaGraph& mGraph;
631};
632
633//!
634//! \class EnqueueSafe

Callers 1

createEnqueueFunctionMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected