Validate missing precisions

openvinotoolkit · Apr 29, 2021 · 12e5706 · 12e5706
1 parent 793f4fa
commit 12e5706
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roll_node.cpp b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roll_node.cpp
@@ -35,7 +35,7 @@ MKLDNNRollNode::MKLDNNRollNode(const InferenceEngine::CNNLayerPtr& layer, const
     const auto &dataShape = dataTensor.getDims();
     const auto &dataPrecision = dataTensor.getPrecision();
 
-    if (!MKLDNNPlugin::one_of(dataPrecision, Precision::I8, Precision::U8, Precision::I16, Precision::I32, Precision::FP32, Precision::I64)) {
+    if (!MKLDNNPlugin::one_of(dataPrecision, Precision::I8, Precision::U8, Precision::I16, Precision::I32, Precision::FP32, Precision::I64, Precision::BF16)) {
         IE_THROW() << layerErrorPrefix << " has unsupported 'data' input precision: " << dataPrecision.name();
     }
     if (dataShape.size() < 1) {
@@ -95,7 +95,6 @@ void MKLDNNRollNode::initSupportedPrimitiveDescriptors() {
 
     InferenceEngine::Precision precision = inputData->getPrecision();
 
-
     auto dataType = MKLDNNExtensionUtils::IEPrecisionToDataType(precision);
 
     auto srcDims = getParentEdgeAt(0)->getDims();

diff --git a/...nce-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/roll.cpp b/...nce-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/roll.cpp
@@ -17,6 +17,8 @@ const std::vector<InferenceEngine::Precision> inputPrecision = {
     InferenceEngine::Precision::I16,
     InferenceEngine::Precision::I32,
     InferenceEngine::Precision::FP32,
+    InferenceEngine::Precision::I64,
+    InferenceEngine::Precision::BF16
 };
 
 const auto testCase2DZeroShifts = ::testing::Combine(