Commit e3cabeb4 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon

CPU Direct Execution: Add implemented builders to the build script

and track callees within the ExternalFunction
parent 41939833
...@@ -28,9 +28,24 @@ set(SRC ...@@ -28,9 +28,24 @@ set(SRC
cpu_tensor_view.cpp cpu_tensor_view.cpp
cpu_tracing.cpp cpu_tracing.cpp
builder/avg_pool.cpp builder/avg_pool.cpp
builder/batch_norm.cpp
builder/broadcast.cpp
builder/concat.cpp
builder/convert.cpp
builder/convert_layout.cpp builder/convert_layout.cpp
builder/convolution.cpp builder/convolution.cpp
builder/dot.cpp
builder/function_call.cpp
builder/matmul_bias.cpp
builder/max.cpp
builder/max_pool.cpp
builder/min.cpp
builder/reshape.cpp builder/reshape.cpp
builder/reverse.cpp
builder/reverse_sequence.cpp
builder/select.cpp
builder/select_and_scatter.cpp
builder/sum.cpp
kernel/eigen_thread_pool.cpp kernel/eigen_thread_pool.cpp
kernel/pad.cpp kernel/pad.cpp
kernel/reduce_max.cpp kernel/reduce_max.cpp
......
...@@ -105,6 +105,11 @@ namespace ngraph ...@@ -105,6 +105,11 @@ namespace ngraph
{ {
return executor; return executor;
} }
std::unordered_map<std::string, std::shared_ptr<CPU_ExternalFunction>>&
get_callees()
{
return callees;
}
bool is_direct_execution() const { return m_direct_execution; } bool is_direct_execution() const { return m_direct_execution; }
protected: protected:
void build(); void build();
...@@ -167,6 +172,7 @@ namespace ngraph ...@@ -167,6 +172,7 @@ namespace ngraph
std::unordered_map<std::string, bool> tensor_stale; std::unordered_map<std::string, bool> tensor_stale;
std::unordered_map<std::string, size_t> intermediates_offsets; std::unordered_map<std::string, size_t> intermediates_offsets;
std::unordered_map<std::string, size_t> function_input_index, function_output_index; std::unordered_map<std::string, size_t> function_input_index, function_output_index;
std::unordered_map<std::string, std::shared_ptr<CPU_ExternalFunction>> callees;
bool m_is_built; bool m_is_built;
bool m_direct_execution; bool m_direct_execution;
}; };
......
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