Skip to content

Commit

Permalink
make use_mkldnn_ = false
Browse files Browse the repository at this point in the history
  • Loading branch information
zhanglirong1999 committed Nov 23, 2023
1 parent e3c78ec commit 535ba48
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions paddle/fluid/inference/api/analysis_config.cc
Original file line number Diff line number Diff line change
Expand Up @@ -953,6 +953,7 @@ void AnalysisConfig::Update() {
(!phi::backends::cpu::MayIUse(phi::backends::cpu::cpu_isa_t::avx2) &&
use_mkldnn_)) {
// User manually disable mkldnn or disable when not support AVX2
use_mkldnn_ = false;
pass_builder()->DisableMKLDNN();
}
#endif
Expand Down

0 comments on commit 535ba48

Please sign in to comment.