未验证 提交 23d96cf2 编写于 作者: J Jacek Czaja 提交者: GitHub

[oneDNN] bumpup onednn 2.2 fixup version (#31473)

* - introduced fix onednn 2.2 version

* - compilation fix
上级 390cebee
...@@ -20,7 +20,7 @@ SET(MKLDNN_SOURCE_DIR ${THIRD_PARTY_PATH}/mkldnn/src/extern_mkldnn) ...@@ -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_INSTALL_DIR ${THIRD_PARTY_PATH}/install/mkldnn)
SET(MKLDNN_INC_DIR "${MKLDNN_INSTALL_DIR}/include" CACHE PATH "mkldnn include directory." FORCE) SET(MKLDNN_INC_DIR "${MKLDNN_INSTALL_DIR}/include" CACHE PATH "mkldnn include directory." FORCE)
SET(MKLDNN_REPOSITORY ${GIT_URL}/oneapi-src/oneDNN.git) SET(MKLDNN_REPOSITORY ${GIT_URL}/oneapi-src/oneDNN.git)
SET(MKLDNN_TAG 3d53cd3f17ce7ca365c980f0e1e50359751ca038) SET(MKLDNN_TAG 72efa005effb49595933e033cc732f215ef0445a)
# Introduce variables: # Introduce variables:
# * CMAKE_INSTALL_LIBDIR # * CMAKE_INSTALL_LIBDIR
......
...@@ -981,12 +981,7 @@ class ConvMKLDNNGradOpKernel : public paddle::framework::OpKernel<T> { ...@@ -981,12 +981,7 @@ class ConvMKLDNNGradOpKernel : public paddle::framework::OpKernel<T> {
* ('any') which lets a primitive (conv backward in this case) choose * ('any') which lets a primitive (conv backward in this case) choose
* the memory format preferred for best performance * the memory format preferred for best performance
*/ */
// TODO: NHWC is preferred starting from oneDNN 2.1 . Any may crash auto chosen_memory_format = MKLDNNMemoryFormat::any;
auto chosen_memory_format =
platform::MayIUse(platform::cpu_isa_t::avx512_core) &&
is_conv3d == false
? MKLDNNMemoryFormat::nhwc
: MKLDNNMemoryFormat::any;
weights_format = MKLDNNMemoryFormat::any; weights_format = MKLDNNMemoryFormat::any;
auto src_md = platform::MKLDNNMemDesc( auto src_md = platform::MKLDNNMemDesc(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册