This fixes the remaining problem with the recently introduced global jit memory manager. This change again uses a memory manager that is local to gallivm_state. This implementation still frees the majority of the memory immediately after compilation. Only the generated code is deferred until this code is no longer used.
This change and the previous one pooling the LLVMContext instances I can now safely run several independent OpenGL contexts driven by llvmpipe from different threads. Signed-off-by: Mathias Froehlich <mathias.froehl...@web.de> --- src/gallium/auxiliary/gallivm/lp_bld_init.c | 7 +++++ src/gallium/auxiliary/gallivm/lp_bld_init.h | 1 + src/gallium/auxiliary/gallivm/lp_bld_misc.cpp | 38 +++++++++++++-------------- src/gallium/auxiliary/gallivm/lp_bld_misc.h | 3 +++ 4 files changed, 30 insertions(+), 19 deletions(-) diff --git a/src/gallium/auxiliary/gallivm/lp_bld_init.c b/src/gallium/auxiliary/gallivm/lp_bld_init.c index 3a9d10f..ec6af2f 100644 --- a/src/gallium/auxiliary/gallivm/lp_bld_init.c +++ b/src/gallium/auxiliary/gallivm/lp_bld_init.c @@ -279,6 +279,8 @@ gallivm_free_code(struct gallivm_state *gallivm) assert(!gallivm->engine); lp_free_generated_code(gallivm->code); gallivm->code = NULL; + LLVMDisposeMCJITMemoryManager(gallivm->memorymgr); + gallivm->memorymgr = NULL; } @@ -300,6 +302,7 @@ init_gallivm_engine(struct gallivm_state *gallivm) ret = lp_build_create_jit_compiler_for_module(&gallivm->engine, &gallivm->code, gallivm->module, + gallivm->memorymgr, (unsigned) optlevel, USE_MCJIT, &error); @@ -369,6 +372,10 @@ init_gallivm_state(struct gallivm_state *gallivm, const char *name) if (!gallivm->builder) goto fail; + gallivm->memorymgr = lp_get_default_memory_manager(); + if (!gallivm->memorymgr) + goto fail; + /* FIXME: MC-JIT only allows compiling one module at a time, and it must be * complete when MC-JIT is created. So defer the MC-JIT engine creation for * now. diff --git a/src/gallium/auxiliary/gallivm/lp_bld_init.h b/src/gallium/auxiliary/gallivm/lp_bld_init.h index 2e32cf8..0a0e0e5 100644 --- a/src/gallium/auxiliary/gallivm/lp_bld_init.h +++ b/src/gallium/auxiliary/gallivm/lp_bld_init.h @@ -44,6 +44,7 @@ struct gallivm_state LLVMPassManagerRef passmgr; LLVMContextRef context; LLVMBuilderRef builder; + LLVMMCJITMemoryManagerRef memorymgr; struct lp_generated_code *code; unsigned compiled; }; diff --git a/src/gallium/auxiliary/gallivm/lp_bld_misc.cpp b/src/gallium/auxiliary/gallivm/lp_bld_misc.cpp index 6bea964..fe515c1 100644 --- a/src/gallium/auxiliary/gallivm/lp_bld_misc.cpp +++ b/src/gallium/auxiliary/gallivm/lp_bld_misc.cpp @@ -319,15 +319,15 @@ class DelegatingJITMemoryManager : public llvm::JITMemoryManager { */ class ShaderMemoryManager : public DelegatingJITMemoryManager { - static llvm::JITMemoryManager *TheMM; - static unsigned NumUsers; + llvm::JITMemoryManager *TheMM; struct GeneratedCode { typedef std::vector<void *> Vec; Vec FunctionBody, ExceptionTable; + llvm::JITMemoryManager *TheMM; - GeneratedCode() { - ++NumUsers; + GeneratedCode(llvm::JITMemoryManager *MM) { + TheMM = MM; } ~GeneratedCode() { @@ -344,27 +344,20 @@ class ShaderMemoryManager : public DelegatingJITMemoryManager { for ( i = ExceptionTable.begin(); i != ExceptionTable.end(); ++i ) TheMM->deallocateExceptionTable(*i); #endif - --NumUsers; - if (NumUsers == 0) { - delete TheMM; - TheMM = 0; - } } }; GeneratedCode *code; llvm::JITMemoryManager *mgr() const { - if (!TheMM) { - TheMM = CreateDefaultMemManager(); - } return TheMM; } public: - ShaderMemoryManager() { - code = new GeneratedCode; + ShaderMemoryManager(llvm::JITMemoryManager* MM) { + TheMM = MM; + code = new GeneratedCode(MM); } virtual ~ShaderMemoryManager() { @@ -395,10 +388,6 @@ class ShaderMemoryManager : public DelegatingJITMemoryManager { } }; -llvm::JITMemoryManager *ShaderMemoryManager::TheMM = 0; -unsigned ShaderMemoryManager::NumUsers = 0; - - /** * Same as LLVMCreateJITCompilerForModule, but: * - allows using MCJIT and enabling AVX feature where available. @@ -414,6 +403,7 @@ LLVMBool lp_build_create_jit_compiler_for_module(LLVMExecutionEngineRef *OutJIT, lp_generated_code **OutCode, LLVMModuleRef M, + LLVMMCJITMemoryManagerRef CMM, unsigned OptLevel, int useMCJIT, char **OutError) @@ -495,7 +485,8 @@ lp_build_create_jit_compiler_for_module(LLVMExecutionEngineRef *OutJIT, builder.setMCPU(MCPU); #endif - ShaderMemoryManager *MM = new ShaderMemoryManager(); + llvm::JITMemoryManager* JMM = reinterpret_cast<llvm::JITMemoryManager*>(CMM); + ShaderMemoryManager *MM = new ShaderMemoryManager(JMM); *OutCode = MM->getGeneratedCode(); builder.setJITMemoryManager(MM); @@ -531,3 +522,12 @@ lp_free_generated_code(struct lp_generated_code *code) { ShaderMemoryManager::freeGeneratedCode(code); } + +extern "C" +LLVMMCJITMemoryManagerRef +lp_get_default_memory_manager() +{ + llvm::JITMemoryManager *mm; + mm = llvm::JITMemoryManager::CreateDefaultMemManager(); + return reinterpret_cast<LLVMMCJITMemoryManagerRef>(mm); +} diff --git a/src/gallium/auxiliary/gallivm/lp_bld_misc.h b/src/gallium/auxiliary/gallivm/lp_bld_misc.h index 64d2a04..40d3e79 100644 --- a/src/gallium/auxiliary/gallivm/lp_bld_misc.h +++ b/src/gallium/auxiliary/gallivm/lp_bld_misc.h @@ -54,6 +54,7 @@ extern int lp_build_create_jit_compiler_for_module(LLVMExecutionEngineRef *OutJIT, struct lp_generated_code **OutCode, LLVMModuleRef M, + LLVMMCJITMemoryManagerRef MM, unsigned OptLevel, int useMCJIT, char **OutError); @@ -61,6 +62,8 @@ lp_build_create_jit_compiler_for_module(LLVMExecutionEngineRef *OutJIT, extern void lp_free_generated_code(struct lp_generated_code *code); +extern LLVMMCJITMemoryManagerRef +lp_get_default_memory_manager(); #ifdef __cplusplus } -- 1.9.3 _______________________________________________ mesa-dev mailing list mesa-dev@lists.freedesktop.org http://lists.freedesktop.org/mailman/listinfo/mesa-dev