From 9253af7d91a6b1b9db3a36ef5c4b53f38a4bd40b Mon Sep 17 00:00:00 2001 From: Egor Duplensky Date: Tue, 11 May 2021 16:50:07 +0300 Subject: [PATCH] Correct store emitter usage --- .../src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp index 7c8dae85c448aa..1b698943536212 100644 --- a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp +++ b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp @@ -88,7 +88,8 @@ struct jit_uni_roi_pooling_kernel_f32 : public jit_uni_roi_pooling_kernel, publi this->postamble(); load_emitter->emit_data(); - store_emitter->emit_data(); + if (!mayiuse(avx512_core_bf16) && mayiuse(avx512_core) && store_emitter != nullptr && store_emitter->get_emu_vcvtneps2bf16() != nullptr) + store_emitter->get_emu_vcvtneps2bf16()->emit_data(); } private: