MCPcopy Create free account
hub / github.com/NVIDIA/DALI / ConsumerWait

Method ConsumerWait

dali/operators/reader/reader_op.h:389–395  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

387 }
388
389 void ConsumerWait() {
390 DomainTimeRange tr("[DALI][DataReader] ConsumerWait #" + to_string(curr_batch_consumer_),
391 DomainTimeRange::kMagenta);
392 std::unique_lock<std::mutex> prefetch_lock(prefetch_access_mutex_);
393 consumer_.wait(prefetch_lock, [this]() { return finished_ || !IsPrefetchQueueEmpty(); });
394 if (prefetch_error_) std::rethrow_exception(prefetch_error_);
395 }
396
397 void ConsumerAdvanceQueue() {
398 {

Callers

nothing calls this directly

Calls 3

rethrow_exceptionFunction · 0.85
to_stringFunction · 0.50
waitMethod · 0.45

Tested by

no test coverage detected