提交 0d8e6c9b 编写于 作者: J Jacek Czaja 提交者: Tao Luo

MKL-DNN upgrade to 0.20 (#18370)

test=develop
上级 d6b6a337
......@@ -66,7 +66,7 @@ ExternalProject_Add(
${EXTERNAL_PROJECT_LOG_ARGS}
DEPENDS ${MKLDNN_DEPENDS}
GIT_REPOSITORY "https://github.com/intel/mkl-dnn.git"
GIT_TAG "863ff6e7042cec7d2e29897fe9f0872e0888b0fc"
GIT_TAG "aef88b7c233f48f8b945da310f1b973da31ad033"
PREFIX ${MKLDNN_SOURCES_DIR}
UPDATE_COMMAND ""
CMAKE_ARGS -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}
......
......@@ -640,7 +640,7 @@ class PoolingMKLDNNHandler : public MKLDNNHandler {
int desired_size = ComputeCeiledOutput(src_tz[i + 2], kernel_size[i],
paddings[i], strides[i]);
if (desired_size != dst_tz[i + 2]) {
right_bot_padding[i] += strides[i];
right_bot_padding[i] += strides[i] - 1;
}
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册