[MERGE #1604 @leirocks] fix pinning number chunks for in-proc JIT

Merge pull request #1604 from leirocks:inprocnumber

Author: leirocks

lib/Backend/JITOutput.cpp

@@ -179,6 +179,10 @@ JITOutput::FinalizeNativeCode(Func *func, EmitBufferAllocation * alloc)
     {
         func->GetInProcJITEntryPointInfo()->SetInProcJITNativeCodeData(func->GetNativeCodeDataAllocator()->Finalize());
         func->GetInProcJITEntryPointInfo()->GetJitTransferData()->SetRawData(func->GetTransferDataAllocator()->Finalize());
+#if !FLOATVAR
+        CodeGenNumberChunk * numberChunks = func->GetNumberAllocator()->Finalize();
+        func->GetInProcJITEntryPointInfo()->SetNumberChunks(numberChunks);
+#endif
     }
 }
 

lib/Runtime/Base/FunctionBody.cpp

@@ -7790,9 +7790,9 @@ namespace Js
 
         if (this->numberPageSegments)
         {
-            auto numberChunks = this->GetScriptContext()->GetThreadContext()
+            auto numberArray = this->GetScriptContext()->GetThreadContext()
                 ->GetXProcNumberPageSegmentManager()->RegisterSegments(this->numberPageSegments);
-            this->SetNumberChunks(numberChunks);
+            this->SetNumberArray(numberArray);
             this->numberPageSegments = nullptr;
         }
     }

lib/Runtime/Base/FunctionBody.h

@@ -10,6 +10,7 @@
 struct CodeGenWorkItem;
 class SourceContextInfo;
 struct DeferredFunctionStub;
+struct CodeGenNumberChunk;
 #ifdef DYNAMIC_PROFILE_MUTATOR
 class DynamicProfileMutator;
 class DynamicProfileMutatorImpl;
@@ -512,7 +513,11 @@ namespace Js
 #if ENABLE_NATIVE_CODEGEN
         NativeCodeData * inProcJITNaticeCodedata;
         char* nativeDataBuffer;
-        Js::JavascriptNumber** numberChunks;
+        union 
+        {
+            Js::JavascriptNumber** numberArray;
+            CodeGenNumberChunk* numberChunks;
+        };
         XProcNumberPageSegment* numberPageSegments;
 #endif
 
@@ -577,11 +582,16 @@ namespace Js
         char** GetNativeDataBufferRef() { return &nativeDataBuffer; }
         char* GetNativeDataBuffer() { return nativeDataBuffer; }
         void SetInProcJITNativeCodeData(NativeCodeData* nativeCodeData) { inProcJITNaticeCodedata = nativeCodeData; }
-        void SetNumberChunks(Js::JavascriptNumber** chunks)
+        void SetNumberChunks(CodeGenNumberChunk* chunks)
         {
-            Assert(numberPageSegments != nullptr);
+            Assert(numberPageSegments == nullptr);
             numberChunks = chunks; 
         }
+        void SetNumberArray(Js::JavascriptNumber** array)
+        {
+            Assert(numberPageSegments != nullptr);
+            numberArray = array;
+        }
         void SetNumberPageSegment(XProcNumberPageSegment * segments)
         {
             Assert(numberPageSegments == nullptr);