Fix asan error due to additional error logging

PiperOrigin-RevId: 631870090
tensorflow · May 8, 2024 · eb93fb5 · eb93fb5
1 parent 7a49df9
commit eb93fb5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tensorflow/core/tfrt/mlrt/interpreter/execute.cc b/tensorflow/core/tfrt/mlrt/interpreter/execute.cc
@@ -247,10 +247,13 @@ void UnwindOnError(ExecutionContext& context, int64_t pc) {
     context.function_stack_.pop_back();
   }
 
+  context.LogError(absl::InternalError(
+      absl::StrCat("Finish UnwindOnError for function ", function_name)));
+
+  // Context may no longer be valid after exit_handler_ is called.
   if (context.exit_handler_) {
     std::move(context.exit_handler_)();
   }
-  context.LogError(absl::InternalError("Finish UnwindOnError"));
 }
 
 }  // namespace execute_internal