diff --git a/cmake/external/mkldnn.cmake b/cmake/external/mkldnn.cmake index b1e437a9007072c82ab375bf5ed79fc7d6c80c47..0ca37f506f61568bcf9eb76e86e5a6074b8043d1 100644 --- a/cmake/external/mkldnn.cmake +++ b/cmake/external/mkldnn.cmake @@ -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} diff --git a/paddle/fluid/platform/mkldnn_reuse.h b/paddle/fluid/platform/mkldnn_reuse.h index 70f8d9dbd8d623aac53b37d0dd5dc980cee8bfb1..4571a7ea13ea31af3405005c45bf9861c19bd921 100644 --- a/paddle/fluid/platform/mkldnn_reuse.h +++ b/paddle/fluid/platform/mkldnn_reuse.h @@ -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; } } }