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 ir {
std::unique_ptr<ir::Graph> ExpectedKernelCachePass::ApplyImpl(
std::unique_ptr<ir::Graph> graph) const {
void ExpectedKernelCachePass::ApplyImpl(ir::Graph* graph) const {
VLOG(3) << "Applies Expected Kernel Cache strategy.";
for (const Node* n : graph->Nodes()) {
if (n->IsOp()) {
n->Op()->SetAttr(kEnableCacheExpectedKernel, true);
}
}
return graph;
}
} // namespace ir

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

Loading…
Cancel
Save