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"); } /**