From 88944de156b8ea5534481eeaa5379cf7fbd383e5 Mon Sep 17 00:00:00 2001 From: Alejandro Saucedo Date: Sun, 23 Aug 2020 06:50:04 +0100 Subject: [PATCH] Fully working end to end flow --- src/OpMult.cpp | 1 + src/OpMult.hpp | 8 +++++++- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/src/OpMult.cpp b/src/OpMult.cpp index fb80e39d6..d4c7a4828 100644 --- a/src/OpMult.cpp +++ b/src/OpMult.cpp @@ -36,6 +36,7 @@ template OpMult::~OpMult() { SPDLOG_DEBUG("Kompute OpMult destructor started"); + } template diff --git a/src/OpMult.hpp b/src/OpMult.hpp index 25a7bf2df..b7cceb725 100644 --- a/src/OpMult.hpp +++ b/src/OpMult.hpp @@ -38,11 +38,17 @@ class OpMult : public OpBase void postSubmit() override; private: + // Always owned resources + std::shared_ptr mTensorOutputStaging; + + // Optionally owned resources std::shared_ptr mAlgorithm; + bool mFreeAlgorithm = false; + + // Never owned resources std::shared_ptr mTensorLHS; std::shared_ptr mTensorRHS; std::shared_ptr mTensorOutput; - std::shared_ptr mTensorOutputStaging; uint32_t mX; uint32_t mY;