update expected_kernel_cache_pass

test=develop
revert-16734-refine/test_imperative_transformer
luotao1 6 years ago
parent 226596a296
commit 695f2db6a0

@ -20,15 +20,13 @@ namespace paddle {
namespace framework { namespace framework {
namespace ir { namespace ir {
std::unique_ptr<ir::Graph> ExpectedKernelCachePass::ApplyImpl( void ExpectedKernelCachePass::ApplyImpl(ir::Graph* graph) const {
std::unique_ptr<ir::Graph> graph) const {
VLOG(3) << "Applies Expected Kernel Cache strategy."; VLOG(3) << "Applies Expected Kernel Cache strategy.";
for (const Node* n : graph->Nodes()) { for (const Node* n : graph->Nodes()) {
if (n->IsOp()) { if (n->IsOp()) {
n->Op()->SetAttr(kEnableCacheExpectedKernel, true); n->Op()->SetAttr(kEnableCacheExpectedKernel, true);
} }
} }
return graph;
} }
} // namespace ir } // namespace ir

@ -23,8 +23,7 @@ namespace ir {
class ExpectedKernelCachePass : public Pass { class ExpectedKernelCachePass : public Pass {
protected: protected:
std::unique_ptr<ir::Graph> ApplyImpl( void ApplyImpl(ir::Graph* graph) const override;
std::unique_ptr<ir::Graph> graph) const override;
}; };
} // namespace ir } // namespace ir

Loading…
Cancel
Save