MKL-DNN upgrade to 0.20 (#18370)

test=develop
DDDivano-patch-1
Jacek Czaja 6 years ago committed by Tao Luo
parent d6b6a337a9
commit 0d8e6c9b8b

@ -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;
}
}
}

Loading…
Cancel
Save