blob: e3c95d93126cc833950bd70df0631d9eb1874fb9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
//
// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "NetworkExecutionUtils.hpp"
#include <armnnUtils/Filesystem.hpp>
#include <iterator>
std::vector<std::string> ParseStringList(const std::string& inputString, const char* delimiter)
{
std::stringstream stream(inputString);
return ParseArrayImpl<std::string>(stream, [](const std::string& s) {
return armnn::stringUtils::StringTrimCopy(s); }, delimiter);
}
bool CheckInferenceTimeThreshold(const std::chrono::duration<double, std::milli>& duration,
const double& thresholdTime)
{
ARMNN_LOG(info) << "Inference time: " << std::setprecision(2)
<< std::fixed << duration.count() << " ms\n";
// If thresholdTime == 0.0 (default), then it hasn't been supplied at command line
if (thresholdTime != 0.0)
{
ARMNN_LOG(info) << "Threshold time: " << std::setprecision(2)
<< std::fixed << thresholdTime << " ms";
auto thresholdMinusInference = thresholdTime - duration.count();
ARMNN_LOG(info) << "Threshold time - Inference time: " << std::setprecision(2)
<< std::fixed << thresholdMinusInference << " ms" << "\n";
if (thresholdMinusInference < 0)
{
std::string errorMessage = "Elapsed inference time is greater than provided threshold time.";
ARMNN_LOG(fatal) << errorMessage;
return false;
}
}
return true;
}
bool ValidatePath(const std::string& file, const bool expectFile)
{
if (!fs::exists(file))
{
std::cerr << "Given file path '" << file << "' does not exist" << std::endl;
return false;
}
if (!fs::is_regular_file(file) && expectFile)
{
std::cerr << "Given file path '" << file << "' is not a regular file" << std::endl;
return false;
}
return true;
}
std::vector<unsigned int> ParseArray(std::istream& stream)
{
return ParseArrayImpl<unsigned int>(
stream,
[](const std::string& s) { return armnn::numeric_cast<unsigned int>(std::stoi(s)); });
}
bool ValidatePaths(const std::vector<std::string>& fileVec, const bool expectFile)
{
bool allPathsValid = true;
for (auto const& file : fileVec)
{
if(!ValidatePath(file, expectFile))
{
allPathsValid = false;
}
}
return allPathsValid;
}
void LogAndThrow(std::string eMsg)
{
ARMNN_LOG(error) << eMsg;
throw armnn::Exception(eMsg);
}
|