|
|
@ -116,6 +116,7 @@ void GPUSession::GraphKernelOptimize(const std::shared_ptr<KernelGraph> &kernel_
|
|
|
|
}
|
|
|
|
}
|
|
|
|
auto optimizer = std::make_shared<opt::GraphOptimizer>();
|
|
|
|
auto optimizer = std::make_shared<opt::GraphOptimizer>();
|
|
|
|
auto pm = std::make_shared<opt::PassManager>("graph_kernel_pm");
|
|
|
|
auto pm = std::make_shared<opt::PassManager>("graph_kernel_pm");
|
|
|
|
|
|
|
|
pm->AddPass(std::make_shared<opt::GraphKernelExpander>());
|
|
|
|
pm->AddPass(std::make_shared<opt::BasicOpsFusion>());
|
|
|
|
pm->AddPass(std::make_shared<opt::BasicOpsFusion>());
|
|
|
|
pm->AddPass(std::make_shared<opt::CompositeOpsFusion>());
|
|
|
|
pm->AddPass(std::make_shared<opt::CompositeOpsFusion>());
|
|
|
|
pm->AddPass(std::make_shared<opt::GraphKernelSplitter>());
|
|
|
|
pm->AddPass(std::make_shared<opt::GraphKernelSplitter>());
|
|
|
|