codecov

Author: Semyon1104

src/layers_oneDNN/EWLayer.cpp

@@ -94,8 +94,6 @@ void EwLayerOneDnn::run(const std::vector<Tensor>& input,
       stream_->wait();
 
       output[0] = make_tensor(output_data, input_tensor.get_shape());
-    } else {
-      throw std::runtime_error("EwLayerOneDnn: Unsupported data type");
     }
 
   } catch (const std::exception& e) {

test/single_layer/test_ewlayer_onednn.cpp

@@ -246,7 +246,7 @@ TEST(ewlayer_onednn, multiple_input_tensors) {
   EXPECT_THROW({ layer.run(in, out); }, std::runtime_error);
 }
 
-TEST(ewlayer_onednn, unsupported_tensor_dimensionality) {
+TEST(ewlayer_onednn, unsupported_tensor_dimensionality1) {
   EwLayerOneDnn layer("relu");
 
   Shape shape_6d({2, 3, 4, 5, 6, 7});
@@ -536,3 +536,46 @@ TEST(ewlayer_onednn, all_functions_int) {
     }
   }
 }
+
+TEST(ewlayer_onednn, unsupported_tensor_dimensionality) {
+  EwLayerOneDnn layer("relu");
+  Shape shape6d({2, 2, 2, 2, 2, 2});
+  std::vector<float> data6d(64, 1.0f);
+  Tensor input = make_tensor(data6d, shape6d);
+  Tensor output;
+  std::vector<Tensor> in{input};
+  std::vector<Tensor> out{output};
+
+  EXPECT_THROW(layer.run(in, out), std::invalid_argument);
+
+  try {
+    layer.run(in, out);
+    FAIL() << "Expected std::invalid_argument exception";
+  } catch (const std::invalid_argument& e) {
+    EXPECT_NE(std::string(e.what()).find("Unsupported tensor dimensionality"),
+              std::string::npos);
+    EXPECT_NE(std::string(e.what()).find("6"), std::string::npos);
+  }
+}
+
+TEST(ewlayer_onednn, unsupported_function_algorithm) {
+  EwLayerOneDnn layer("relu");
+  EwLayerOneDnn invalid_layer("invalid_func_123", 1.0f, 0.0f);
+
+  Tensor input = make_tensor<float>({1.0F});
+  Tensor output;
+  std::vector<Tensor> in{input};
+  std::vector<Tensor> out{output};
+
+  EXPECT_THROW(invalid_layer.run(in, out), std::invalid_argument);
+
+  try {
+    invalid_layer.run(in, out);
+    FAIL() << "Expected std::invalid_argument";
+  } catch (const std::invalid_argument& e) {
+    EXPECT_NE(std::string(e.what()).find("Unsupported function for oneDNN"),
+              std::string::npos);
+    EXPECT_NE(std::string(e.what()).find("invalid_func_123"),
+              std::string::npos);
+  }
+}