diff options
author | Juergen Ributzka <juergen@apple.com> | 2014-04-28 18:19:25 +0000 |
---|---|---|
committer | Juergen Ributzka <juergen@apple.com> | 2014-04-28 18:19:25 +0000 |
commit | 4e0cc51d7974029aa2bbffce08e2c878f529f98a (patch) | |
tree | e4a77f03d3c3cce1cf840ca93491a4d4244d2222 /unittests | |
parent | c430d3ffcde596bad10e1d089cfd15ba8d915204 (diff) | |
download | llvm-4e0cc51d7974029aa2bbffce08e2c878f529f98a.tar.gz llvm-4e0cc51d7974029aa2bbffce08e2c878f529f98a.tar.bz2 llvm-4e0cc51d7974029aa2bbffce08e2c878f529f98a.tar.xz |
[PM] Add pass run listeners to the pass manager.
This commit provides the necessary C/C++ APIs and infastructure to enable fine-
grain progress report and safe suspension points after each pass in the pass
manager.
Clients can provide a callback function to the pass manager to call after each
pass. This can be used in a variety of ways (progress report, dumping of IR
between passes, safe suspension of threads, etc).
The run listener list is maintained in the LLVMContext, which allows a multi-
threaded client to be only informed for it's own thread. This of course assumes
that the client created a LLVMContext for each thread.
This fixes <rdar://problem/16728690>
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@207430 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'unittests')
-rw-r--r-- | unittests/ExecutionEngine/MCJIT/MCJITCAPITest.cpp | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/unittests/ExecutionEngine/MCJIT/MCJITCAPITest.cpp b/unittests/ExecutionEngine/MCJIT/MCJITCAPITest.cpp index 3813d59dbd..c8e0c88229 100644 --- a/unittests/ExecutionEngine/MCJIT/MCJITCAPITest.cpp +++ b/unittests/ExecutionEngine/MCJIT/MCJITCAPITest.cpp @@ -28,6 +28,7 @@ using namespace llvm; static bool didCallAllocateCodeSection; static bool didAllocateCompactUnwindSection; +static bool didCallPassRunListener; static uint8_t *roundTripAllocateCodeSection(void *object, uintptr_t size, unsigned alignment, @@ -64,6 +65,12 @@ static void roundTripDestroy(void *object) { delete static_cast<SectionMemoryManager*>(object); } +static void passRunListenerCallback(LLVMContextRef C, LLVMPassRef P, + LLVMModuleRef M, LLVMValueRef F, + LLVMBasicBlockRef BB) { + didCallPassRunListener = true; +} + namespace { // memory manager to test reserve allocation space callback @@ -142,6 +149,7 @@ protected: virtual void SetUp() { didCallAllocateCodeSection = false; didAllocateCompactUnwindSection = false; + didCallPassRunListener = false; Module = 0; Function = 0; Engine = 0; @@ -429,3 +437,23 @@ TEST_F(MCJITCAPITest, reserve_allocation_space) { EXPECT_TRUE(MM->UsedCodeSize > 0); EXPECT_TRUE(MM->UsedDataSizeRW > 0); } + +TEST_F(MCJITCAPITest, pass_run_listener) { + SKIP_UNSUPPORTED_PLATFORM; + + buildSimpleFunction(); + buildMCJITOptions(); + buildMCJITEngine(); + LLVMContextRef C = LLVMGetGlobalContext(); + LLVMAddPassRunListener(C, passRunListenerCallback); + buildAndRunPasses(); + + union { + void *raw; + int (*usable)(); + } functionPointer; + functionPointer.raw = LLVMGetPointerToGlobal(Engine, Function); + + EXPECT_EQ(42, functionPointer.usable()); + EXPECT_TRUE(didCallPassRunListener); +} |