1 files changed, 11 insertions, 0 deletions
diff --git a/compiler/optimizing/code_generator.h b/compiler/optimizing/code_generator.h
index 99de61ddce..cf59be86bf 100644
--- a/compiler/optimizing/code_generator.h
+++ b/compiler/optimizing/code_generator.h
@@ -403,6 +403,14 @@ class CodeGenerator : public DeletableArenaObject<kArenaAllocCodeGenerator> {
     requires_current_method_ = true;
   }
 
+  bool NeedsSuspendCheckEntry() const {
+    return needs_suspend_check_entry_;
+  }
+
+  void MarkNeedsSuspendCheckEntry() {
+    needs_suspend_check_entry_ = true;
+  }
+
   void SetRequiresCurrentMethod() {
     requires_current_method_ = true;
   }
@@ -855,6 +863,9 @@ class CodeGenerator : public DeletableArenaObject<kArenaAllocCodeGenerator> {
   // Whether the method is a leaf method.
   bool is_leaf_;
 
+  // Whether the method has to emit a SuspendCheck at entry.
+  bool needs_suspend_check_entry_;
+
   // Whether an instruction in the graph accesses the current method.
   // TODO: Rename: this actually indicates that some instruction in the method
   // needs the environment including a valid stack frame.