未验证 提交 6ef8129d 编写于 作者: W Wojciech Uss 提交者: GitHub

upgrade oneDNN with GRU INT8 optimizations (#28420)

* upgrade oneDNN with GRU INT8 optimizations

* fix test
上级 dfffee8a
......@@ -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
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册