[onert] Fix DepthwiseConvOp kernel condition

This commit fixes DepthwiseConvOp kernel usage condition. Eigen kernel is not supproting different width-height stride yet. ONE-DCO-1.0-Signed-off-by: Hyeongseok Oh <[email protected]>
Samsung · Aug 19, 2024 · 14261bc · 14261bc
1 parent f3d7383
commit 14261bc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/runtime/onert/backend/cpu/ops/DepthwiseConvolutionLayer.cc b/runtime/onert/backend/cpu/ops/DepthwiseConvolutionLayer.cc
@@ -81,9 +81,9 @@ void DepthwiseConvolutionLayer::convFloat32()
   op_params.float_activation_min = output_activation_min;
   op_params.float_activation_max = output_activation_max;
 
-  // Since DepthwiseConvOp does not support dilation yet, it uses the existing
-  // kernel in this case.
-  if (_dilationWidth == 1 && _dilationHeight == 1)
+  // Since DepthwiseConvOp does not support dilation and different W/H stride yet,
+  // it uses the existing kernel in this case.
+  if (_dilationWidth == 1 && _dilationHeight == 1 && _strideWidth == _strideHeight)
   {
     nnfw::cker::DepthwiseConvOp(op_params, getShape(_input), getBuffer<float>(_input),
                                 getShape(_kernel), getBuffer<float>(_kernel), getShape(_bias),