Skip to content

Commit 016c3bd

Browse files
authored
Ikva model name support (#325)
1 parent 7a50204 commit 016c3bd

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

larq_compute_engine/ikva/runtime/main.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
#define IKVA_MAIN_NUM_TEST_RUNS 2
1818
#endif
1919

20+
extern const char model_name[];
2021
extern const unsigned char model_data[];
2122

2223
const int tensor_arena_size = 8 * 1024 * 1024; // 8 MB
@@ -142,7 +143,7 @@ int main(int argc, char* argv[]) {
142143

143144
error_reporter->Report("Finished invoke.");
144145

145-
profiler.PrintResults("ikva_model", true);
146+
profiler.PrintResults(model_name, true);
146147
}
147148

148149
return 0;

larq_compute_engine/ikva/runtime/model_data_example.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,9 @@
11
#include <cstdint>
22

33
extern std::uint8_t model_data[];
4+
extern const char model_name[];
45
extern const int model_data_size;
6+
const char model_name[] = "example_model";
57
__attribute__((aligned(8))) std::uint8_t model_data[] = {
68
0x20, 0x00, 0x00, 0x00, 0x54, 0x46, 0x4c, 0x33, 0x00, 0x00, 0x00, 0x00,
79
0x14, 0x00, 0x20, 0x00, 0x1c, 0x00, 0x18, 0x00, 0x14, 0x00, 0x10, 0x00,

0 commit comments

Comments
 (0)