\class EnqueueGraphSafe \brief Functor to enqueue inference from CUDA Graph
| 614 | //! \brief Functor to enqueue inference from CUDA Graph |
| 615 | //! |
| 616 | class EnqueueGraphSafe |
| 617 | { |
| 618 | |
| 619 | public: |
| 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 |
no outgoing calls
no test coverage detected