diff options
-rw-r--r-- | shim/sl/canonical/ArmnnDriver.hpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/shim/sl/canonical/ArmnnDriver.hpp b/shim/sl/canonical/ArmnnDriver.hpp index c42ffa20b5..877faa667e 100644 --- a/shim/sl/canonical/ArmnnDriver.hpp +++ b/shim/sl/canonical/ArmnnDriver.hpp @@ -92,12 +92,12 @@ public: { VLOG(DRIVER) << "ArmnnDriver::getNumberOfCacheFilesNeeded()"; unsigned int numberOfCachedModelFiles = 0; - //TODO: -// for (auto& backend : m_Options.GetBackends()) -// { -// numberOfCachedModelFiles += GetNumberOfCacheFiles(backend); -// } - return std::make_pair(numberOfCachedModelFiles, 0); + for (auto& backend : m_Options.GetBackends()) + { + numberOfCachedModelFiles += GetNumberOfCacheFiles(backend); + VLOG(DRIVER) << "ArmnnDriver::getNumberOfCacheFilesNeeded() = " << std::to_string(numberOfCachedModelFiles); + } + return std::make_pair(numberOfCachedModelFiles, 1ul); } GeneralResult<void> wait() const override |