blob: 57c8dc9c89ff01264291fe28568974052eea9496 (
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
|
// Copyright (C) 2018 Intel Corporation
//
// SPDX-License-Identifier: Apache-2.0
//
#include "config.h"
#include "ie_plugin_config.hpp"
#include "ie_common.h"
#include <string>
#include <map>
#include <algorithm>
#include <cpp_interfaces/exception2status.hpp>
namespace MKLDNNPlugin {
using namespace InferenceEngine;
void Config::readProperties(const std::map<std::string, std::string> &prop) {
for (auto& kvp : prop) {
std::string key = kvp.first;
std::string val = kvp.second;
if (key == PluginConfigParams::KEY_CPU_BIND_THREAD) {
if (val == PluginConfigParams::YES) useThreadBinding = true;
else if (val == PluginConfigParams::NO) useThreadBinding = false;
else
THROW_IE_EXCEPTION << "Wrong value for property key " << PluginConfigParams::KEY_CPU_BIND_THREAD
<< ". Expected only YES/NO";
} else if (key == PluginConfigParams::KEY_DYN_BATCH_LIMIT) {
int val_i = std::stoi(val);
// zero and any negative value will be treated
// as default batch size
batchLimit = std::max(val_i, 0);
} else if (key == PluginConfigParams::KEY_PERF_COUNT) {
if (val == PluginConfigParams::YES) collectPerfCounters = true;
else if (val == PluginConfigParams::NO) collectPerfCounters = false;
else
THROW_IE_EXCEPTION << "Wrong value for property key " << PluginConfigParams::KEY_PERF_COUNT
<< ". Expected only YES/NO";
} else if (key == PluginConfigParams::KEY_EXCLUSIVE_ASYNC_REQUESTS) {
if (val == PluginConfigParams::YES) exclusiveAsyncRequests = true;
else if (val == PluginConfigParams::NO) exclusiveAsyncRequests = false;
else
THROW_IE_EXCEPTION << "Wrong value for property key " << PluginConfigParams::KEY_EXCLUSIVE_ASYNC_REQUESTS
<< ". Expected only YES/NO";
} else if (key.compare(PluginConfigParams::KEY_DYN_BATCH_ENABLED) == 0) {
if (val.compare(PluginConfigParams::YES) == 0)
enableDynamicBatch = true;
else if (val.compare(PluginConfigParams::NO) == 0)
enableDynamicBatch = false;
else
THROW_IE_EXCEPTION << "Wrong value for property key " << PluginConfigParams::KEY_DYN_BATCH_ENABLED
<< ". Expected only YES/NO";
} else {
THROW_IE_EXCEPTION << NOT_FOUND_str << "Unsupported property " << key << " by CPU plugin";
}
}
}
} // namespace MKLDNNPlugin
|
