From 1fed1d5f4c42486dd7429fe8e7acfaf4e85e3772 Mon Sep 17 00:00:00 2001
From: Kristofer Jonsson <kristofer.jonsson@arm.com>
Date: Mon, 21 Nov 2022 13:39:45 +0100
Subject: Use Tensorflow MicroPrintf

Change-Id: I01bbafc45ca2a56759d7d5afbbbacefb40fd9a26
---
 lib/arm_profiler/include/arm_profiler.hpp                       | 1 +
 lib/arm_profiler/src/arm_profiler.cpp                           | 9 ++++-----
 lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp | 2 +-
 lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp     | 4 ++--
 4 files changed, 8 insertions(+), 8 deletions(-)

(limited to 'lib')

diff --git a/lib/arm_profiler/include/arm_profiler.hpp b/lib/arm_profiler/include/arm_profiler.hpp
index 3a487bd..9c7b5f3 100644
--- a/lib/arm_profiler/include/arm_profiler.hpp
+++ b/lib/arm_profiler/include/arm_profiler.hpp
@@ -19,6 +19,7 @@
 #ifndef ARM_PROFILER_H
 #define ARM_PROFILER_H
 
+#include "tensorflow/lite/micro/compatibility.h"
 #include "tensorflow/lite/micro/micro_profiler_interface.h"
 #include <memory>
 
diff --git a/lib/arm_profiler/src/arm_profiler.cpp b/lib/arm_profiler/src/arm_profiler.cpp
index 7648584..2977474 100644
--- a/lib/arm_profiler/src/arm_profiler.cpp
+++ b/lib/arm_profiler/src/arm_profiler.cpp
@@ -17,7 +17,7 @@
  */
 
 #include "tensorflow/lite/kernels/internal/compatibility.h"
-#include "tensorflow/lite/micro/micro_error_reporter.h"
+#include "tensorflow/lite/micro/micro_log.h"
 #include "tensorflow/lite/micro/micro_time.h"
 
 #include <string.h>
@@ -36,7 +36,7 @@ ArmProfiler::ArmProfiler(size_t max_events) : max_events_(max_events), num_event
 
 uint32_t ArmProfiler::BeginEvent(const char *tag) {
     if (num_events_ == max_events_) {
-        tflite::GetMicroErrorReporter()->Report("Profiling event overflow, max: %u events", max_events_);
+        MicroPrintf("Profiling event overflow, max: %u events", max_events_);
         num_events_ = 0;
     }
 
@@ -63,10 +63,9 @@ uint64_t ArmProfiler::GetTotalTicks() const {
 }
 
 void ArmProfiler::ReportResults() const {
-    tflite::GetMicroErrorReporter()->Report("Profiler report, CPU cycles per operator:");
+    MicroPrintf("Profiler report, CPU cycles per operator:");
     for (size_t i = 0; i < num_events_; ++i) {
-        tflite::GetMicroErrorReporter()->Report(
-            "%s : cycle_cnt : %u cycles", tags_[i], end_ticks_[i] - start_ticks_[i]);
+        MicroPrintf("%s : cycle_cnt : %u cycles", tags_[i], end_ticks_[i] - start_ticks_[i]);
     }
 }
 
diff --git a/lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp b/lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp
index 649cfb5..a76ac41 100644
--- a/lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp
+++ b/lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp
@@ -20,7 +20,7 @@
 #define LAYER_BY_LAYER_PROFILER_H
 
 #include "EventRecorder.h"
-#include "tensorflow/lite/kernels/internal/compatibility.h"
+#include "tensorflow/lite/micro/compatibility.h"
 #include "tensorflow/lite/micro/micro_profiler_interface.h"
 #include <memory>
 #include <vector>
diff --git a/lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp b/lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp
index 4f525ee..927587f 100644
--- a/lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp
+++ b/lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp
@@ -17,7 +17,7 @@
  */
 
 #include "tensorflow/lite/kernels/internal/compatibility.h"
-#include "tensorflow/lite/micro/micro_error_reporter.h"
+#include "tensorflow/lite/micro/micro_log.h"
 #include "tensorflow/lite/micro/micro_profiler.h"
 #include "tensorflow/lite/micro/micro_time.h"
 
@@ -41,7 +41,7 @@ LayerByLayerProfiler::LayerByLayerProfiler(size_t max_events, Backend _backend,
 // NOTE: THIS PROFILER ONLY WORKS ON SYSTEMS WITH 1 NPU
 uint32_t LayerByLayerProfiler::BeginEvent(const char *tag) {
     if (num_events_ == max_events_) {
-        tflite::GetMicroErrorReporter()->Report("Profiling event overflow, max: %u events", max_events_);
+        MicroPrintf("Profiling event overflow, max: %u events", max_events_);
         num_events_ = 0;
     }
 
-- 
cgit v1.2.1