| 645 | } |
| 646 | |
| 647 | bool operator()(TrtCudaStream& stream) const |
| 648 | { |
| 649 | try |
| 650 | { |
| 651 | return mContext.enqueueV3(stream.get()); |
| 652 | } |
| 653 | catch (const std::exception&) |
| 654 | { |
| 655 | return false; |
| 656 | } |
| 657 | return false; |
| 658 | } |
| 659 | |
| 660 | nvinfer1::safe::IExecutionContext& mContext; |
| 661 | private: |