Update MKLDNN to v1.2 (#22521)

revert-22710-feature/integrated_ps_api
Adam 6 years ago committed by GitHub
parent f5db451d3b
commit 608447bfd5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -20,7 +20,7 @@ SET(MKLDNN_SOURCE_DIR ${THIRD_PARTY_PATH}/mkldnn/src/extern_mkldnn)
SET(MKLDNN_INSTALL_DIR ${THIRD_PARTY_PATH}/install/mkldnn)
SET(MKLDNN_INC_DIR "${MKLDNN_INSTALL_DIR}/include" CACHE PATH "mkldnn include directory." FORCE)
SET(MKLDNN_REPOSITORY https://github.com/intel/mkl-dnn.git)
SET(MKLDNN_TAG 52c3052df8ec1d5b8b45cb6c350a952840eabd42)
SET(MKLDNN_TAG 75d0b1a7f3586c212e37acebbb8acd221cee7216)
# Introduce variables:
# * CMAKE_INSTALL_LIBDIR

@ -288,7 +288,7 @@ class FCPrimitiveFactory {
// becuase we perform per-output-channel quantization
int mask = CreateMask(0, scale_data.size() > 1);
attributes.set_output_scales(mask, scale_data);
auto reorder = mkldnn::reorder({src_mem, dst_mem, attributes});
auto reorder = mkldnn::reorder(src_mem, dst_mem, attributes);
mkldnn::stream astream(engine_);
reorder.execute(astream,

Loading…
Cancel
Save