Skip to content

Commit

Permalink
[LPT] Rollback device specific INT8
Browse files Browse the repository at this point in the history
  • Loading branch information
vzinovie committed Sep 28, 2020
1 parent 8a157c5 commit f4cf3a5
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions inference-engine/src/cldnn_engine/cldnn_engine.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -284,7 +284,7 @@ ExecutableNetworkInternal::Ptr clDNNEngine::LoadExeNetworkImpl(const InferenceEn

CLDNNPlugin::Config conf = _impl->m_config;
auto device_info = GetDeviceInfo(config);
conf.enableInt8 = true; // device_info.supports_imad || device_info.supports_immad;
conf.enableInt8 = device_info.supports_imad || device_info.supports_immad;
conf.UpdateFromMap(config);

if (conf.enableDynamicBatch) {
Expand Down Expand Up @@ -338,7 +338,7 @@ ExecutableNetworkInternal::Ptr clDNNEngine::LoadExeNetworkImpl(const InferenceEn

CLDNNPlugin::Config conf = getContextImpl(casted)->GetConfig();
auto device_info = GetDeviceInfo(config);
conf.enableInt8 = true; // device_info.supports_imad || device_info.supports_immad;
conf.enableInt8 = device_info.supports_imad || device_info.supports_immad;
conf.UpdateFromMap(config);

if (conf.enableDynamicBatch) {
Expand Down Expand Up @@ -383,16 +383,13 @@ void clDNNEngine::QueryNetwork(const ICNNNetwork& network,
std::string>& config,
QueryNetworkResult& res) const {
GetDeviceInfo(config); // Verify device id
CLDNNPlugin::Config conf = _impl->m_config;
conf.enableInt8 = true; // device_info.supports_imad || device_info.supports_immad;
conf.UpdateFromMap(config);
auto function = network.getFunction();
if (function != nullptr) {
std::unordered_set<std::string> originalOps;
for (auto&& node : function->get_ops()) {
originalOps.emplace(node->get_friendly_name());
}
auto clonedNetwork = CloneAndTransformNetwork(network, conf);
auto clonedNetwork = CloneAndTransformNetwork(network, _impl->m_config);
std::unordered_set<std::string> supported;
std::unordered_set<std::string> unsupported;

Expand Down

0 comments on commit f4cf3a5

Please sign in to comment.