Skip to content

N
ngraph
Commit 1ad6c7f0 authored by Fenglei's avatar Fenglei Committed by Robert Kimball
Browse files
Options

add GetOutputElementElimination (#2302)

parent 4680678d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/ngraph/runtime/gpu/gpu_compiled_function.cpp
View file @ 1ad6c7f0
......@@ -34,6 +34,7 @@
#include "ngraph/node.hpp"
#include "ngraph/pass/algebraic_simplification.hpp"
#include "ngraph/pass/any_all_replacement.hpp"
#include "ngraph/pass/get_output_element_elimination.hpp"
#include "ngraph/pass/like_replacement.hpp"
#include "ngraph/runtime/gpu/gpu_backend.hpp"
......@@ -173,6 +174,7 @@ void runtime::gpu::GPUCompiledFunction::compile()
pass_manager.register_pass<ngraph::pass::LikeReplacement>();
pass_manager.register_pass<runtime::gpu::pass::GPULayout>(this);
pass_manager.register_pass<ngraph::pass::AssignLayout<descriptor::layout::DenseTensorLayout>>();
pass_manager.register_pass<ngraph::pass::GetOutputElementElimination>();
pass_manager.register_pass<ngraph::pass::Liveness>();
pass_manager.register_pass<ngraph::pass::MemoryLayout>(get_memory_alignment());
pass_manager.register_pass<runtime::gpu::pass::TensorMemoryReservation>(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment