Update main.cpp (#10740)

This commit is contained in:
Irina Efode
2022-03-03 14:09:55 +03:00
committed by GitHub
parent 8121de731c
commit fdf12c9537

View File

@@ -93,15 +93,15 @@ int main(int argc, char *argv[]) {
std::vector<std::string> local_cache_dirs = CommonTestUtils::splitStringByDelimiter(FLAGS_local_cache);
std::vector<std::string> dirs = CommonTestUtils::splitStringByDelimiter(FLAGS_input_folders);
auto cachedOps = findModelsInDirs(local_cache_dirs);
auto models = findModelsInDirs(dirs);
auto cache = SubgraphsDumper::OPCache::make_cache();
if (!FLAGS_local_cache.empty()) {
auto cachedOps = findModelsInDirs(local_cache_dirs);
cacheModels(cache, ret_code, cachedOps, FLAGS_extract_body);
}
cacheModels(cache, ret_code, models, FLAGS_extract_body);
cache->serialize_cached_ops(FLAGS_output_folder);
return ret_code;
}
}