MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / operator()

Method operator()

samples/common/sampleInference.cpp:518–535  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

516 }
517
518 bool operator()(TrtCudaStream& stream) const
519 {
520 try
521 {
522 bool const result = mContext.enqueue(mBatch, mBuffers, stream.get(), nullptr);
523 // Collecting layer timing info from current profile index of execution context
524 if (mContext.getProfiler() && !mContext.getEnqueueEmitsProfile() && !mContext.reportToProfiler())
525 {
526 gLogWarning << "Failed to collect layer timing info from previous enqueue()" << std::endl;
527 }
528 return result;
529 }
530 catch (const std::exception&)
531 {
532 return false;
533 }
534 return false;
535 }
536
537private:
538 void** mBuffers{};

Callers

nothing calls this directly

Calls 5

getProfilerMethod · 0.80
reportToProfilerMethod · 0.80
enqueueMethod · 0.45
getMethod · 0.45

Tested by

no test coverage detected