Update jit_compiler_a64.cpp

This commit is contained in:
SChernykh 2021-02-01 22:52:02 +01:00
parent db03573804
commit 346892e170

View file

@ -108,7 +108,6 @@ JitCompilerA64::JitCompilerA64(bool hugePagesEnable, bool) :
hugePages(hugePagesJIT && hugePagesEnable), hugePages(hugePagesJIT && hugePagesEnable),
literalPos(ImulRcpLiteralsEnd) literalPos(ImulRcpLiteralsEnd)
{ {
allocate(CodeSize);
} }
JitCompilerA64::~JitCompilerA64() JitCompilerA64::~JitCompilerA64()
@ -118,8 +117,13 @@ JitCompilerA64::~JitCompilerA64()
void JitCompilerA64::generateProgram(Program& program, ProgramConfiguration& config, uint32_t) void JitCompilerA64::generateProgram(Program& program, ProgramConfiguration& config, uint32_t)
{ {
if (!allocatedSize) {
allocate(CodeSize);
}
#ifdef XMRIG_SECURE_JIT #ifdef XMRIG_SECURE_JIT
enableWriting(); else {
enableWriting();
}
#endif #endif
uint32_t codePos = MainLoopBegin + 4; uint32_t codePos = MainLoopBegin + 4;
@ -173,8 +177,13 @@ void JitCompilerA64::generateProgram(Program& program, ProgramConfiguration& con
void JitCompilerA64::generateProgramLight(Program& program, ProgramConfiguration& config, uint32_t datasetOffset) void JitCompilerA64::generateProgramLight(Program& program, ProgramConfiguration& config, uint32_t datasetOffset)
{ {
if (!allocatedSize) {
allocate(CodeSize);
}
#ifdef XMRIG_SECURE_JIT #ifdef XMRIG_SECURE_JIT
enableWriting(); else {
enableWriting();
}
#endif #endif
uint32_t codePos = MainLoopBegin + 4; uint32_t codePos = MainLoopBegin + 4;
@ -235,12 +244,8 @@ void JitCompilerA64::generateProgramLight(Program& program, ProgramConfiguration
template<size_t N> template<size_t N>
void JitCompilerA64::generateSuperscalarHash(SuperscalarProgram(&programs)[N]) void JitCompilerA64::generateSuperscalarHash(SuperscalarProgram(&programs)[N])
{ {
const size_t requiredSize = CodeSize + CalcDatasetItemSize(); if (!allocatedSize) {
if (allocatedSize < requiredSize) { allocate(CodeSize + CalcDatasetItemSize());
if (allocatedSize) {
freePagedMemory(code, allocatedSize);
}
allocate(requiredSize);
} }
#ifdef XMRIG_SECURE_JIT #ifdef XMRIG_SECURE_JIT
else { else {