|
|
|
@ -146,22 +146,5 @@ void NaiveExecutor::CleanFeedFetchOps() {
|
|
|
|
|
ops_.swap(ops);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void NaiveExecutor::EnableMKLDNN(const ProgramDesc &program) {
|
|
|
|
|
#ifdef PADDLE_WITH_MKLDNN
|
|
|
|
|
VLOG(3) << "use_mkldnn=True";
|
|
|
|
|
for (size_t block_id = 0; block_id < program.Size(); ++block_id) {
|
|
|
|
|
auto *block = const_cast<ProgramDesc &>(program).MutableBlock(block_id);
|
|
|
|
|
for (auto *op : block->AllOps()) {
|
|
|
|
|
if (op->HasAttr("use_mkldnn")) {
|
|
|
|
|
op->SetAttr("use_mkldnn", true);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
#else
|
|
|
|
|
LOG(WARNING)
|
|
|
|
|
<< "'MKLDNN' is not supported, Please re-compile with WITH_MKLDNN option";
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
} // namespace framework
|
|
|
|
|
} // namespace paddle
|
|
|
|
|