IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[INT8+BF16] x8s8s32x_deconv_fwd bf16 output case elimination (#6237)

Browse Source
This commit is contained in:
Alexey Varyzgin 2021-06-25 16:47:47 +03:00 committed by GitHub
parent 9868bde421
commit e8301fb545
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
inference-engine/src/mkldnn_plugin/nodes/mkldnn_deconv_node.cpp
View File

@ -202,7 +202,13 @@ void MKLDNNDeconvolutionNode::getSupportedDescriptors() {
InferenceEngine::Precision inPrecision = getOriginalInputPrecisionAtPort(0); InferenceEngine::Precision inPrecision = getOriginalInputPrecisionAtPort(0);
InferenceEngine::Precision outPrecision = getOriginalOutputPrecisionAtPort(0); InferenceEngine::Precision outPrecision = getOriginalOutputPrecisionAtPort(0);
if (!isInt8) { if (isInt8) {
// TODO: We have to extend jit_avx512_core_x8s8s32x_deconv_fwd_kernel from oneDNN to support BF16 output data type
if (InferenceEngine::Precision::BF16 == inPrecision)
inPrecision = InferenceEngine::Precision::FP32;
if (InferenceEngine::Precision::BF16 == outPrecision)
outPrecision = InferenceEngine::Precision::FP32;
} else {
if (!one_of(inPrecision, InferenceEngine::Precision::FP32, InferenceEngine::Precision::BF16)) if (!one_of(inPrecision, InferenceEngine::Precision::FP32, InferenceEngine::Precision::BF16))
inPrecision = InferenceEngine::Precision::FP32; inPrecision = InferenceEngine::Precision::FP32;
if (!one_of(outPrecision, InferenceEngine::Precision::FP32, InferenceEngine::Precision::BF16)) if (!one_of(outPrecision, InferenceEngine::Precision::FP32, InferenceEngine::Precision::BF16))