m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

resolve memory leak in cudnn8.0 (#31029)

Browse Source
revert-31068-fix_conv3d_windows
Wilber 4 years ago committed by GitHub
parent c4ddc3ab0d
commit cf8b8f9c5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File

8
paddle/fluid/inference/api/paddle_pass_builder.cc
Unescape View File

@ -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() {
"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
})) { })) {

Write Preview
Loading…
Cancel
Save