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

disable memory leak pass in cudnn8 (#30838)

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