From 45c82eefb41cffc976c34fb2a758c8bdccfc8de9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Erick=20Mu=C3=B1oz?= Date: Sun, 23 Apr 2023 08:20:26 -0700 Subject: [PATCH] [OneDNN] Fix poolgrad bug (#15557) * Fixed default dilatation value for poolgrad ops ### Description Changed default dilatation value to 0 in poolgrad ops ### Motivation and Context Fixes error on unit tests when --enable_training --use_dnnl flags are active and --- onnxruntime/core/providers/dnnl/subgraph/dnnl_poolgrad.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/dnnl/subgraph/dnnl_poolgrad.cc b/onnxruntime/core/providers/dnnl/subgraph/dnnl_poolgrad.cc index aec81d16d4..226d3b5053 100644 --- a/onnxruntime/core/providers/dnnl/subgraph/dnnl_poolgrad.cc +++ b/onnxruntime/core/providers/dnnl/subgraph/dnnl_poolgrad.cc @@ -92,8 +92,8 @@ void DnnlPoolGrad::CreatePrimitive(DnnlSubgraphPrimitive& sp, DnnlNode& node) { } } - // Dilatation of 1 - auto dilatation = dnnl::memory::dims(kernel_shape.size(), 1); + // Default dilatation to 0 + auto dilatation = dnnl::memory::dims(kernel_shape.size(), 0); dnnl::pooling_forward::primitive_desc pool_forward_pd(dnnl_engine, dnnl::prop_kind::forward, algo, fwd_dx_md, dy_md, strides, kernel_shape, dilatation, padding_left, padding_right);