diff --git a/cmake/external/mkldnn.cmake b/cmake/external/mkldnn.cmake index 9c9e1d18d90d6333b225e0b99e5d652c5c13d84a..30f9005fc81761afb48a5bdded39dd59f631d008 100644 --- a/cmake/external/mkldnn.cmake +++ b/cmake/external/mkldnn.cmake @@ -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 ${GIT_URL}/oneapi-src/oneDNN.git) -SET(MKLDNN_TAG 361725600224f41b7347a1c6bee9b04d1e6c14d7) +SET(MKLDNN_TAG b530ba24c7005ec0f72c06cb55cecd5dffdc5e37) # Introduce variables: # * CMAKE_INSTALL_LIBDIR diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_flags_mkldnn_ops_on_off.py b/python/paddle/fluid/tests/unittests/mkldnn/test_flags_mkldnn_ops_on_off.py index 21269fada45908d460a00defd7a985661740e70c..e935c279b4183c2d903557ddc322c8ff3e40139a 100644 --- a/python/paddle/fluid/tests/unittests/mkldnn/test_flags_mkldnn_ops_on_off.py +++ b/python/paddle/fluid/tests/unittests/mkldnn/test_flags_mkldnn_ops_on_off.py @@ -33,7 +33,7 @@ class TestFlagsUseMkldnn(unittest.TestCase): self.relu_regex = b"^dnnl_verbose,exec,cpu,eltwise,.+alg:eltwise_relu alpha:0 beta:0,10x20x20" self.ew_add_regex = b"^dnnl_verbose,exec,cpu,binary.+alg:binary_add,10x20x30:10x20x30 10x20x30" - self.matmul_regex = b"^dnnl_verbose,exec,cpu,matmul,.*b10m20n20k30" + self.matmul_regex = b"^dnnl_verbose,exec,cpu,matmul,.*10x20x30:10x30x20:10x20x20" def flags_use_mkl_dnn_common(self, e): cmd = self._python_interp