diff options
-rw-r--r-- | applications/message_handler/message_handler.cpp | 4 | ||||
-rw-r--r-- | applications/message_handler/message_handler.hpp | 4 |
2 files changed, 8 insertions, 0 deletions
diff --git a/applications/message_handler/message_handler.cpp b/applications/message_handler/message_handler.cpp index 4b77389..6630c0a 100644 --- a/applications/message_handler/message_handler.cpp +++ b/applications/message_handler/message_handler.cpp @@ -513,6 +513,7 @@ void InferenceHandler::runInference(ethosu_core_inference_req &req, ethosu_core_ job.clean(); } +#if defined(ETHOSU) /* * Print PMU counters */ @@ -528,6 +529,7 @@ void InferenceHandler::runInference(ethosu_core_inference_req &req, ethosu_core_ printf("ethosu_pmu_cycle_cntr : %" PRIu64 " cycles\n", rsp.pmu_cycle_counter_count); } } +#endif /* * Send inference response @@ -699,6 +701,7 @@ void OutgoingMessageHandler::readCapabilties(ethosu_core_msg_capabilities_rsp &r } // namespace MessageHandler +#if defined(ETHOSU) extern "C" void ethosu_inference_begin(struct ethosu_driver *drv, void *userArg) { MessageHandler::InferenceHandler *self = static_cast<MessageHandler::InferenceHandler *>(userArg); @@ -754,3 +757,4 @@ extern "C" void ethosu_inference_end(struct ethosu_driver *drv, void *userArg) { // Disable PMU ETHOSU_PMU_Disable(drv); } +#endif diff --git a/applications/message_handler/message_handler.hpp b/applications/message_handler/message_handler.hpp index 90b1cd2..13a3c60 100644 --- a/applications/message_handler/message_handler.hpp +++ b/applications/message_handler/message_handler.hpp @@ -24,7 +24,9 @@ #include "semphr.h" #include "message_queue.hpp" +#if defined(ETHOSU) #include <ethosu_driver.h> +#endif #include <inference_process.hpp> #include <mailbox.hpp> @@ -65,8 +67,10 @@ private: bool getInferenceJob(const EthosU::ethosu_core_inference_req &req, InferenceProcess::InferenceJob &job); +#if defined(ETHOSU) friend void ::ethosu_inference_begin(struct ethosu_driver *drv, void *userArg); friend void ::ethosu_inference_end(struct ethosu_driver *drv, void *userArg); +#endif QueueHandle_t inferenceQueue; QueueHandle_t outputQueue; |