Arm backend: Test temp memory allocation return code in backend (pytorch#11814)

### Summary
Test return code of memory allocation and fail gracefully.

Signed-off-by: Zingo Andersen <[email protected]>

Author: zingo

backends/arm/runtime/EthosUBackend.cpp

@@ -113,7 +113,7 @@ class EthosUBackend final : public ::executorch::runtime::BackendInterface {
       BackendInitContext& context,
       FreeableBuffer* processed,
       ArrayRef<CompileSpec> compile_specs) const override {
-    ET_LOG(Info, "EthosUBackend::init %p", processed->data());
+    ET_LOG(Info, "data:%p", processed->data());
 
     const char* data = static_cast<const char*>(processed->data());
     size_t size = processed->size();
@@ -173,7 +173,7 @@ class EthosUBackend final : public ::executorch::runtime::BackendInterface {
         static_cast<const char*>(execution_handle->processed->data());
     EXECUTORCH_PROF_END(event_tracer, event_tracer_local_scope);
 
-    ET_LOG(Debug, "EthosUBackend::execute %p", data);
+    ET_LOG(Debug, "data:%p", data);
 
     EXECUTORCH_PROF_START(
         event_tracer,
@@ -182,7 +182,7 @@ class EthosUBackend final : public ::executorch::runtime::BackendInterface {
     // Read key sections from the vela_bin_stream
     if (vela_bin_read(data, &handles, execution_handle->processed->size()) ==
         false) {
-      ET_LOG(Error, "EthosUBackend::vela_read: error, invalid binary layout");
+      ET_LOG(Error, "vela_read: error, invalid binary layout");
       return Error::InvalidProgram;
     }
     EXECUTORCH_PROF_END(event_tracer, event_tracer_local_scope);
@@ -193,9 +193,16 @@ class EthosUBackend final : public ::executorch::runtime::BackendInterface {
     // the end of the execution of the Ethos-U custom delegate
     char* ethosu_scratch =
         static_cast<char*>(temp_allocator->allocate(handles.scratch_data_size));
+    if (ethosu_scratch == nullptr) {
+      ET_LOG(
+          Error,
+          "Failed to allocate scratch buffer of %zu bytes from temp_allocator",
+          handles.scratch_data_size);
+      return Error::MemoryAllocationFailed;
+    }
     ET_LOG(
         Debug,
-        "EthosUBackend::execute: Running program data:\n  cmd %p %zu\n  weight %p %zu\n  scratch %p %zu\n  fast scratch %p %zu\n",
+        "Running program data:\n  cmd %p %zu\n  weight %p %zu\n  scratch %p %zu\n  fast scratch %p %zu\n",
         handles.cmd_data,
         handles.cmd_data_size,
         handles.weight_data,
@@ -301,7 +308,7 @@ class EthosUBackend final : public ::executorch::runtime::BackendInterface {
         std::unique_ptr<ethosu_driver, decltype(&ethosu_release_driver)>(
             ethosu_reserve_driver(), ethosu_release_driver);
     if (driver == NULL) {
-      ET_LOG(Error, "EthosUBackend::execute: ethosu_reserve_driver failed");
+      ET_LOG(Error, "ethosu_reserve_driver failed");
       return Error::InvalidState;
     }
 
@@ -333,10 +340,7 @@ class EthosUBackend final : public ::executorch::runtime::BackendInterface {
     EXECUTORCH_PROF_END(event_tracer, event_tracer_local_scope);
 
     if (result != 0) {
-      ET_LOG(
-          Error,
-          "EthosUBackend::execute: Ethos-U invocation failed error (%d)",
-          result);
+      ET_LOG(Error, "Ethos-U invocation failed error (%d)", result);
       return Error::InvalidProgram;
     }
     int tensor_dim = 0, io_dim = 0;

examples/arm/executor_runner/arm_executor_runner.cpp

@@ -201,7 +201,13 @@ void et_pal_emit_log_message(
     const char* message,
     ET_UNUSED size_t length) {
   fprintf(
-      stderr, "%c [executorch:%s:%zu] %s\n", level, filename, line, message);
+      stderr,
+      "%c [executorch:%s:%zu %s()] %s\n",
+      level,
+      filename,
+      line,
+      function,
+      message);
 }
 
 /**