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

resolve memory leak in cudnn8.0 (#31029)

上级 c4ddc3ab
...@@ -96,9 +96,13 @@ const std::vector<std::string> kTRTSubgraphPasses({ ...@@ -96,9 +96,13 @@ const std::vector<std::string> kTRTSubgraphPasses({
"conv_bn_fuse_pass", // "conv_bn_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
// cudnn8.0 has memory leak problem in conv + eltwise + act, so we
// disable the pass.
#if !(CUDNN_VERSION >= 8000 && CUDNN_VERSION < 8100)
"conv_elementwise_add_act_fuse_pass", // "conv_elementwise_add_act_fuse_pass", //
"conv_elementwise_add2_act_fuse_pass", // "conv_elementwise_add2_act_fuse_pass", //
#endif // #endif
#endif
"transpose_flatten_concat_fuse_pass", "transpose_flatten_concat_fuse_pass",
}); });
...@@ -228,7 +232,7 @@ void CpuPassStrategy::EnableMKLDNN() { ...@@ -228,7 +232,7 @@ void CpuPassStrategy::EnableMKLDNN() {
"batch_norm_act_fuse_pass", "batch_norm_act_fuse_pass",
// TODO(intel): Please fix the bug on windows. // TODO(intel): Please fix the bug on windows.
// https://github.com/PaddlePaddle/Paddle/issues/29710 // https://github.com/PaddlePaddle/Paddle/issues/29710
//"mkldnn_inplace_pass", // This pass should be activated after // "mkldnn_inplace_pass", // This pass should be activated after
// fuses. Disabled by default due to // fuses. Disabled by default due to
// little gain and lots of problems // little gain and lots of problems
})) { })) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册