Validate oneDNN scalar reduction input

allnes · allnes · commit dce11263a3dc · 2026-05-01T20:51:57.000+02:00
diff --git a/src/layers_oneDNN/ReduceLayer.cpp b/src/layers_oneDNN/ReduceLayer.cpp
@@ -164,7 +164,7 @@ void ReduceLayerOneDnn::validate_input(const std::vector<Tensor>& input) {
   }
 
   const auto& shape = input[0].get_shape();
-  if (shape.dims() == 0) {
+  if (shape.dims() == 0 || (shape.dims() == 1 && shape.count() == 1)) {
     throw std::runtime_error("ReduceLayerOneDnn: Scalar input not supported");
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -164,7 +164,7 @@ void ReduceLayerOneDnn::validate_input(const std::vector<Tensor>& input) {`
`164`	`164`	`}`
`165`	`165`
`166`	`166`	`const auto& shape = input[0].get_shape();`
`167`		`- if (shape.dims() == 0) {`
	`167`	`+ if (shape.dims() == 0 \|\| (shape.dims() == 1 && shape.count() == 1)) {`
`168`	`168`	`throw std::runtime_error("ReduceLayerOneDnn: Scalar input not supported");`
`169`	`169`	`}`
`170`	`170`	`}`