Ikva model name support (#325)

larq · Mar 10, 2021 · 016c3bd · 016c3bd
1 parent 7a50204
commit 016c3bd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/larq_compute_engine/ikva/runtime/main.cc b/larq_compute_engine/ikva/runtime/main.cc
@@ -17,6 +17,7 @@
 #define IKVA_MAIN_NUM_TEST_RUNS 2
 #endif
 
+extern const char model_name[];
 extern const unsigned char model_data[];
 
 const int tensor_arena_size = 8 * 1024 * 1024;  // 8 MB
@@ -142,7 +143,7 @@ int main(int argc, char* argv[]) {
 
     error_reporter->Report("Finished invoke.");
 
-    profiler.PrintResults("ikva_model", true);
+    profiler.PrintResults(model_name, true);
   }
 
   return 0;

diff --git a/larq_compute_engine/ikva/runtime/model_data_example.cc b/larq_compute_engine/ikva/runtime/model_data_example.cc
@@ -1,7 +1,9 @@
 #include <cstdint>
 
 extern std::uint8_t model_data[];
+extern const char model_name[];
 extern const int model_data_size;
+const char model_name[] = "example_model";
 __attribute__((aligned(8))) std::uint8_t model_data[] = {
     0x20, 0x00, 0x00, 0x00, 0x54, 0x46, 0x4c, 0x33, 0x00, 0x00, 0x00, 0x00,
     0x14, 0x00, 0x20, 0x00, 0x1c, 0x00, 0x18, 0x00, 0x14, 0x00, 0x10, 0x00,