1 files changed, 3 insertions, 5 deletions
diff --git a/llvm/lib/Fuzzer/FuzzerInternal.h b/llvm/lib/Fuzzer/FuzzerInternal.h
index e6bef799f13..c26aa5f4b21 100644
--- a/llvm/lib/Fuzzer/FuzzerInternal.h
+++ b/llvm/lib/Fuzzer/FuzzerInternal.h
@@ -403,11 +403,7 @@ public:
   void ResetCoverage();
 
   bool InFuzzingThread() const { return IsMyThread; }
-  size_t GetCurrentUnitInFuzzingThead(const uint8_t **Data) const {
-    assert(InFuzzingThread());
-    *Data = CurrentUnitData;
-    return CurrentUnitSize;
-  }
+  size_t GetCurrentUnitInFuzzingThead(const uint8_t **Data) const;
 
 private:
   void AlarmCallback();
@@ -445,6 +441,8 @@ private:
   void DumpCurrentUnit(const char *Prefix);
   void DeathCallback();
 
+  void SetCurrentUnit(const uint8_t *Data, size_t Size);
+  size_t GetCurrentUnitNoThreadCheck(const uint8_t **Data) const;
   const uint8_t *CurrentUnitData = nullptr;
   size_t CurrentUnitSize = 0;
   bool InOOMState = false;