未验证 提交 d1ae7b98 编写于 作者: W Wilber 提交者: GitHub

disable memory leak pass in cudnn8 (#30838)

上级 71dc03cd
......@@ -126,11 +126,15 @@ GpuPassStrategy::GpuPassStrategy() : PassStrategy({}) {
"fc_elementwise_layernorm_fuse_pass", //
#if CUDNN_VERSION >= 7100 // To run conv_fusion, the version of cudnn must be
// guaranteed at least v7
// TODO(wilber): cudnn8 has memory leak problem in conv + eltwise + act, so we
// disable the pass temporarily.
#if CUDNN_VERSION < 8000
"conv_elementwise_add_act_fuse_pass", //
"conv_elementwise_add2_act_fuse_pass", //
"conv_elementwise_add_fuse_pass", //
#endif //
"transpose_flatten_concat_fuse_pass", //
#endif
"conv_elementwise_add_fuse_pass", //
#endif //
"transpose_flatten_concat_fuse_pass", //
// following pass should be located in the last, since it will
// work on all fused ops.
"runtime_context_cache_pass"
......
......@@ -240,7 +240,7 @@ function cmake_base() {
-DWITH_GLOO=${gloo_flag}
-DWITH_LITE=${WITH_LITE:-OFF}
-DWITH_XPU=${WITH_XPU:-OFF}
-DLITE_GIT_TAG=develop
-DLITE_GIT_TAG=release/v2.8
========================================
EOF
# Disable UNITTEST_USE_VIRTUALENV in docker because
......@@ -272,7 +272,7 @@ EOF
-DWITH_GRPC=${grpc_flag} \
-DWITH_PSCORE=${distibuted_flag} \
-DWITH_GLOO=${gloo_flag} \
-DLITE_GIT_TAG=develop \
-DLITE_GIT_TAG=release/v2.8 \
-DWITH_XPU=${WITH_XPU:-OFF} \
-DWITH_LITE=${WITH_LITE:-OFF};build_error=$?
if [ "$build_error" != 0 ];then
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册