1 //===- unittests/Interpreter/InterpreterTestBase.h ------------------ C++ -===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #ifndef LLVM_CLANG_UNITTESTS_INTERPRETER_INTERPRETERTESTBASE_H 10 #define LLVM_CLANG_UNITTESTS_INTERPRETER_INTERPRETERTESTBASE_H 11 12 #include "llvm/ExecutionEngine/Orc/LLJIT.h" 13 #include "llvm/Support/Error.h" 14 #include "llvm/Support/ManagedStatic.h" 15 #include "llvm/Support/TargetSelect.h" 16 17 #include "gtest/gtest.h" 18 19 #if defined(_AIX) || defined(__MVS__) 20 #define CLANG_INTERPRETER_PLATFORM_CANNOT_CREATE_LLJIT 21 #endif 22 23 namespace clang { 24 25 class InterpreterTestBase : public ::testing::Test { 26 protected: HostSupportsJIT()27 static bool HostSupportsJIT() { 28 #ifdef CLANG_INTERPRETER_PLATFORM_CANNOT_CREATE_LLJIT 29 return false; 30 #else 31 if (auto JIT = llvm::orc::LLJITBuilder().create()) { 32 return true; 33 } else { 34 llvm::consumeError(JIT.takeError()); 35 return false; 36 } 37 #endif 38 } 39 SetUp()40 void SetUp() override { 41 if (!HostSupportsJIT()) 42 GTEST_SKIP(); 43 } 44 TearDown()45 void TearDown() override {} 46 SetUpTestSuite()47 static void SetUpTestSuite() { 48 llvm::InitializeNativeTarget(); 49 llvm::InitializeNativeTargetAsmPrinter(); 50 } 51 TearDownTestSuite()52 static void TearDownTestSuite() { llvm::llvm_shutdown(); } 53 }; 54 55 } // namespace clang 56 57 #undef CLANG_INTERPRETER_PLATFORM_CANNOT_CREATE_LLJIT 58 59 #endif // LLVM_CLANG_UNITTESTS_INTERPRETER_INTERPRETERTESTBASE_H 60