Method ThreadMain

dali/pipeline/util/worker_thread.cc:138–201

}

void WorkerThreadImpl::ThreadMain(int device_id, bool set_affinity, const std::string &name) {
  SetThreadName(name.c_str());
  DeviceGuard g(device_id);
  try {
    if (set_affinity) {
#if NVML_ENABLED
      nvml::SetCPUAffinity();
#endif
    }
  } catch (std::exception &e) {
    std::lock_guard<std::mutex> lock(mutex_);
    errors_.push(e.what());
    running_ = false;
  } catch (...) {
    std::lock_guard<std::mutex> lock(mutex_);
    errors_.push("Unknown exception");
    running_ = false;
  }

  barrier_.Wait();

  while (running_) {
    // Check the queue for work
    std::unique_lock<std::mutex> lock(mutex_);
    while (work_queue_.empty() && running_) {
      cv_.wait(lock);
    }

    if (!running_) {
      break;
    }

    Work work = std::move(work_queue_.front());
    work_queue_.pop();
    lock.unlock();

    try {
      work();
    } catch (std::exception &e) {
      std::cerr << std::this_thread::get_id() << " Exception in thread: " << e.what() << endl;
      lock.lock();
      errors_.push(e.what());
      running_ = false;
      completed_.notify_one();
      lock.unlock();
      break;
    } catch (...) {
      std::cerr << std::this_thread::get_id() << " Exception in thread" << endl;
      lock.lock();
      errors_.push("Caught unknown exception in thread.");
      running_ = false;
      completed_.notify_one();
      lock.unlock();
      break;
    }

    lock.lock();