From 33df1dec4e8e157ce8d4860a78f85230af3f61e3 Mon Sep 17 00:00:00 2001 From: Alejandro Saucedo Date: Thu, 15 Oct 2020 07:16:01 +0100 Subject: [PATCH] Added evalOpAwait and evalOpAsync into the manager --- src/include/kompute/Manager.hpp | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/src/include/kompute/Manager.hpp b/src/include/kompute/Manager.hpp index 70eea2e73..7f95aa9ab 100644 --- a/src/include/kompute/Manager.hpp +++ b/src/include/kompute/Manager.hpp @@ -143,14 +143,16 @@ class Manager void evalOpAwait(std::string sequenceName, uint64_t waitFor = UINT64_MAX) { SPDLOG_DEBUG("Kompute Manager evalOpAwait triggered"); - std::weak_ptr sqWeakPtr = - this->getOrCreateManagedSequence(sequenceName); + std::unordered_map>::iterator found = + this->mManagedSequences.find(sequenceName); - if (std::shared_ptr sq = sqWeakPtr.lock()) { - SPDLOG_DEBUG("Kompute Manager evalOpAwait running sequence EVAL AWAIT"); - sq->evalAwait(waitFor); + if (found == this->mManagedSequences.end()) { + if (std::shared_ptr sq = found->second) { + SPDLOG_DEBUG("Kompute Manager evalOpAwait running sequence Sequence EVAL AWAIT"); + sq->evalAwait(waitFor); + } + SPDLOG_DEBUG("Kompute Manager evalOpAwait running sequence SUCCESS"); } - SPDLOG_DEBUG("Kompute Manager evalOpAwait running sequence SUCCESS"); } /**