Add function entry point API. (#624)

* Add function entry point API. * Modify return type of get_current_entry_point
kendryte · Aug 22, 2022 · 8baf43a · 8baf43a
1 parent e3bd2eb
commit 8baf43a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/include/nncase/codegen/module_builder.h b/include/nncase/codegen/module_builder.h
@@ -93,6 +93,7 @@ class NNCASE_API module_builder
     section *find_section(std::string_view section_name);
     void merge_to_rdata_section(std::string_view from);
     function_call_id function_id(ir::graph *graph);
+    std::streampos get_current_entry_point();
     void set_current_entry_point(std::streampos pos);
     void set_current_function_text_end(std::streampos pos);
 

diff --git a/src/codegen/module_builder.cpp b/src/codegen/module_builder.cpp
@@ -189,6 +189,11 @@ function_call_id module_builder::function_id(ir::graph *graph)
     throw std::invalid_argument("Can't find graph " + graph->name() + " in modules");
 }
 
+std::streampos module_builder::get_current_entry_point()
+{
+    return entry_points_.at(current_function_);
+}
+
 void module_builder::set_current_entry_point(std::streampos pos)
 {
     entry_points_[current_function_] = pos;