[llvm] d122d80 - Reapply "[ORC] Add unit tests for parts of the ..." with fixes and improvements.

Evgenii Stepanov via llvm-commits llvm-commits at lists.llvm.org
Tue Apr 27 10:13:16 PDT 2021


Looks like it was fixed in
https://lab.llvm.org/buildbot/#/builders/5/builds/7086 -- thanks!


On Mon, Apr 26, 2021 at 10:13 PM Vitaly Buka <vitalybuka at google.com> wrote:

> +Evgenii Stepanov <eugenis at google.com>
> It still fails tests
> https://lab.llvm.org/buildbot/#/builders/5/builds/7082
>
> On Mon, 26 Apr 2021 at 20:56, Lang Hames via llvm-commits <
> llvm-commits at lists.llvm.org> wrote:
>
>>
>> Author: Lang Hames
>> Date: 2021-04-26T20:44:40-07:00
>> New Revision: d122d80b3d1c3ffdfbf6a2168f3d61fa4337facc
>>
>> URL:
>> https://github.com/llvm/llvm-project/commit/d122d80b3d1c3ffdfbf6a2168f3d61fa4337facc
>> DIFF:
>> https://github.com/llvm/llvm-project/commit/d122d80b3d1c3ffdfbf6a2168f3d61fa4337facc.diff
>>
>> LOG: Reapply "[ORC] Add unit tests for parts of the ..." with fixes and
>> improvements.
>>
>> This reapplies 8740360093b, which was reverted in bbddadd46e4 due to
>> buildbot
>> errors.
>>
>> This version checks that a JIT instance can be safely constructed,
>> skipping
>> tests if it can not be. To enable this it introduces new C API to
>> retrieve and
>> set the target triple for a JITTargetMachineBuilder.
>>
>> Added:
>>     llvm/unittests/ExecutionEngine/Orc/OrcCAPITest.cpp
>>
>> Modified:
>>     llvm/include/llvm-c/LLJIT.h
>>     llvm/include/llvm-c/Orc.h
>>     llvm/lib/ExecutionEngine/Orc/OrcV2CBindings.cpp
>>     llvm/unittests/ExecutionEngine/Orc/CMakeLists.txt
>>
>> Removed:
>>
>>
>>
>>
>> ################################################################################
>> diff  --git a/llvm/include/llvm-c/LLJIT.h b/llvm/include/llvm-c/LLJIT.h
>> index 976fda22bc48..b140a5d05881 100644
>> --- a/llvm/include/llvm-c/LLJIT.h
>> +++ b/llvm/include/llvm-c/LLJIT.h
>> @@ -78,6 +78,9 @@ void LLVMOrcDisposeLLJITBuilder(LLVMOrcLLJITBuilderRef
>> Builder);
>>   * instance. Calling this function is optional: if it is not called then
>> the
>>   * LLJITBuilder will use JITTargeTMachineBuilder::detectHost to
>> construct a
>>   * JITTargetMachineBuilder.
>> + *
>> + * This function takes ownership of the JTMB argument: clients should not
>> + * dispose of the JITTargetMachineBuilder after calling this function.
>>   */
>>  void LLVMOrcLLJITBuilderSetJITTargetMachineBuilder(
>>      LLVMOrcLLJITBuilderRef Builder, LLVMOrcJITTargetMachineBuilderRef
>> JTMB);
>>
>> diff  --git a/llvm/include/llvm-c/Orc.h b/llvm/include/llvm-c/Orc.h
>> index ac20986233f1..78f6f28a4657 100644
>> --- a/llvm/include/llvm-c/Orc.h
>> +++ b/llvm/include/llvm-c/Orc.h
>> @@ -623,8 +623,9 @@ void
>> LLVMOrcDisposeThreadSafeModule(LLVMOrcThreadSafeModuleRef TSM);
>>   * Create a JITTargetMachineBuilder by detecting the host.
>>   *
>>   * On success the client owns the resulting JITTargetMachineBuilder. It
>> must be
>> - * passed to a consuming operation (e.g. LLVMOrcCreateLLJITBuilder) or
>> disposed
>> - * of by calling LLVMOrcDisposeJITTargetMachineBuilder.
>> + * passed to a consuming operation (e.g.
>> + * LLVMOrcLLJITBuilderSetJITTargetMachineBuilder) or disposed of by
>> calling
>> + * LLVMOrcDisposeJITTargetMachineBuilder.
>>   */
>>  LLVMErrorRef LLVMOrcJITTargetMachineBuilderDetectHost(
>>      LLVMOrcJITTargetMachineBuilderRef *Result);
>> @@ -646,6 +647,29 @@
>> LLVMOrcJITTargetMachineBuilderCreateFromTargetMachine(LLVMTargetMachineRef
>> TM);
>>  void LLVMOrcDisposeJITTargetMachineBuilder(
>>      LLVMOrcJITTargetMachineBuilderRef JTMB);
>>
>> +/**
>> + * Returns the target triple for the given JITTargetMachineBuilder as a
>> string.
>> + *
>> + * The caller owns the resulting string as must dispose of it by calling
>> + * LLVMOrcJITTargetMachineBuilderDisposeTargetTriple.
>> + */
>> +char *LLVMOrcJITTargetMachineBuilderGetTargetTriple(
>> +    LLVMOrcJITTargetMachineBuilderRef JTMB);
>> +
>> +/**
>> + * Sets the target triple for the given JITTargetMachineBuilder to the
>> given
>> + * string.
>> + */
>> +void LLVMOrcJITTargetMachineBuilderSetTargetTriple(
>> +    LLVMOrcJITTargetMachineBuilderRef JTMB, const char *TargetTriple);
>> +
>> +/**
>> + * Destroy a triple string returned by
>> + * LLVMOrcJITTargetMachineBuilderGetTargetTriple.
>> + */
>> +void LLVMOrcJITTargetMachineBuilderDisposeTargetTriple(
>> +    LLVMOrcJITTargetMachineBuilderRef JTMB, char *TargetTriple);
>> +
>>  /**
>>   * Emit an object buffer to an ObjectLayer.
>>   *
>>
>> diff  --git a/llvm/lib/ExecutionEngine/Orc/OrcV2CBindings.cpp
>> b/llvm/lib/ExecutionEngine/Orc/OrcV2CBindings.cpp
>> index 796daf58d9bb..419f6e81e6e3 100644
>> --- a/llvm/lib/ExecutionEngine/Orc/OrcV2CBindings.cpp
>> +++ b/llvm/lib/ExecutionEngine/Orc/OrcV2CBindings.cpp
>> @@ -473,6 +473,24 @@ void LLVMOrcDisposeJITTargetMachineBuilder(
>>    delete unwrap(JTMB);
>>  }
>>
>> +char *LLVMOrcJITTargetMachineBuilderGetTargetTriple(
>> +    LLVMOrcJITTargetMachineBuilderRef JTMB) {
>> +  auto Tmp = unwrap(JTMB)->getTargetTriple().str();
>> +  char *TargetTriple = (char *)malloc(Tmp.size() + 1);
>> +  strcpy(TargetTriple, Tmp.c_str());
>> +  return TargetTriple;
>> +}
>> +
>> +void LLVMOrcJITTargetMachineBuilderSetTargetTriple(
>> +    LLVMOrcJITTargetMachineBuilderRef JTMB, const char *TargetTriple) {
>> +  unwrap(JTMB)->getTargetTriple() = Triple(TargetTriple);
>> +}
>> +
>> +void LLVMOrcJITTargetMachineBuilderDisposeTargetTriple(
>> +    LLVMOrcJITTargetMachineBuilderRef JTMB, char *TargetTriple) {
>> +  free(TargetTriple);
>> +}
>> +
>>  void LLVMOrcObjectLayerEmit(LLVMOrcObjectLayerRef ObjLayer,
>>                              LLVMOrcMaterializationResponsibilityRef R,
>>                              LLVMMemoryBufferRef ObjBuffer) {
>>
>> diff  --git a/llvm/unittests/ExecutionEngine/Orc/CMakeLists.txt
>> b/llvm/unittests/ExecutionEngine/Orc/CMakeLists.txt
>> index d205d53829aa..505692233ebe 100644
>> --- a/llvm/unittests/ExecutionEngine/Orc/CMakeLists.txt
>> +++ b/llvm/unittests/ExecutionEngine/Orc/CMakeLists.txt
>> @@ -17,6 +17,7 @@ add_llvm_unittest(OrcJITTests
>>    IndirectionUtilsTest.cpp
>>    JITTargetMachineBuilderTest.cpp
>>    LazyCallThroughAndReexportsTest.cpp
>> +  OrcCAPITest.cpp
>>    OrcTestCommon.cpp
>>    QueueChannel.cpp
>>    ResourceTrackerTest.cpp
>>
>> diff  --git a/llvm/unittests/ExecutionEngine/Orc/OrcCAPITest.cpp
>> b/llvm/unittests/ExecutionEngine/Orc/OrcCAPITest.cpp
>> new file mode 100644
>> index 000000000000..eaad4a206836
>> --- /dev/null
>> +++ b/llvm/unittests/ExecutionEngine/Orc/OrcCAPITest.cpp
>> @@ -0,0 +1,307 @@
>> +//===--- OrcCAPITest.cpp - Unit tests for the OrcJIT v2 C API ---*- C++
>> -*-===//
>> +//
>> +// Part of the LLVM Project, under the Apache License v2.0 with LLVM
>> Exceptions.
>> +// See https://llvm.org/LICENSE.txt for license information.
>> +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
>> +//
>>
>> +//===----------------------------------------------------------------------===//
>> +
>> +#include "llvm-c/Core.h"
>> +#include "llvm-c/LLJIT.h"
>> +#include "llvm-c/Orc.h"
>> +#include "gtest/gtest.h"
>> +
>> +#include "llvm/Support/Debug.h"
>> +#include "llvm/Support/raw_ostream.h"
>> +
>> +using namespace llvm;
>> +
>> +// OrcCAPITestBase contains several helper methods and pointers for unit
>> tests
>> +// written for the LLVM-C API. It provides the following helpers:
>> +//
>> +// 1. Jit: an LLVMOrcLLJIT instance which is freed upon test exit
>> +// 2. ExecutionSession: the LLVMOrcExecutionSession for the JIT
>> +// 3. MainDylib: the main JITDylib for the LLJIT instance
>> +// 4. materializationUnitFn: function pointer to an empty function, used
>> for
>> +//                           materialization unit testing
>> +// 5. definitionGeneratorFn: function pointer for a basic
>> +//
>>  LLVMOrcCAPIDefinitionGeneratorTryToGenerateFunction
>> +// 6. createTestModule: helper method for creating a basic
>> thread-safe-module
>> +class OrcCAPITestBase : public testing::Test {
>> +protected:
>> +  LLVMOrcLLJITRef Jit = nullptr;
>> +  LLVMOrcExecutionSessionRef ExecutionSession = nullptr;
>> +  LLVMOrcJITDylibRef MainDylib = nullptr;
>> +
>> +public:
>> +  static void SetUpTestCase() {
>> +    LLVMInitializeNativeTarget();
>> +
>> +    // Attempt to set up a JIT instance once to verify that we can.
>> +    LLVMOrcJITTargetMachineBuilderRef JTMB = nullptr;
>> +    if (LLVMErrorRef E =
>> LLVMOrcJITTargetMachineBuilderDetectHost(&JTMB)) {
>> +      // If setup fails then disable these tests.
>> +      LLVMConsumeError(E);
>> +      TargetSupported = false;
>> +      return;
>> +    }
>> +
>> +    char *Triple = LLVMOrcJITTargetMachineBuilderGetTargetTriple(JTMB);
>> +    if (!isSupported(Triple)) {
>> +      // If this triple isn't supported then bail out.
>> +      TargetSupported = false;
>> +      LLVMOrcJITTargetMachineBuilderDisposeTargetTriple(JTMB, Triple);
>> +      LLVMOrcDisposeJITTargetMachineBuilder(JTMB);
>> +      return;
>> +    }
>> +
>> +    LLVMOrcJITTargetMachineBuilderDisposeTargetTriple(JTMB, Triple);
>> +    LLVMOrcLLJITBuilderRef Builder = LLVMOrcCreateLLJITBuilder();
>> +    LLVMOrcLLJITBuilderSetJITTargetMachineBuilder(Builder, JTMB);
>> +    LLVMOrcLLJITRef J;
>> +    if (LLVMErrorRef E = LLVMOrcCreateLLJIT(&J, Builder)) {
>> +      // If setup fails then disable these tests.
>> +      TargetSupported = false;
>> +      LLVMConsumeError(E);
>> +      return;
>> +    }
>> +
>> +    LLVMOrcDisposeLLJIT(J);
>> +    TargetSupported = true;
>> +  }
>> +
>> +  void SetUp() override {
>> +    if (!TargetSupported)
>> +      return;
>> +
>> +    LLVMOrcJITTargetMachineBuilderRef JTMB = nullptr;
>> +    LLVMErrorRef E1 = LLVMOrcJITTargetMachineBuilderDetectHost(&JTMB);
>> +    assert(E1 == LLVMErrorSuccess && "Expected call to detect host to
>> succeed");
>> +
>> +    LLVMOrcLLJITBuilderRef Builder = LLVMOrcCreateLLJITBuilder();
>> +    LLVMOrcLLJITBuilderSetJITTargetMachineBuilder(Builder, JTMB);
>> +    LLVMErrorRef E2 = LLVMOrcCreateLLJIT(&Jit, Builder);
>> +    assert(E2 == LLVMErrorSuccess &&
>> +           "Expected call to create LLJIT to succeed");
>> +    ExecutionSession = LLVMOrcLLJITGetExecutionSession(Jit);
>> +    MainDylib = LLVMOrcLLJITGetMainJITDylib(Jit);
>> +  }
>> +  void TearDown() override {
>> +    LLVMOrcDisposeLLJIT(Jit);
>> +    Jit = nullptr;
>> +  }
>> +
>> +protected:
>> +  static bool isSupported(StringRef Triple) {
>> +    if (Triple.startswith("armv7"))
>> +      return false;
>> +    return true;
>> +  }
>> +
>> +  static void materializationUnitFn() {}
>> +  // Stub definition generator, where all Names are materialized from the
>> +  // materializationUnitFn() test function and defined into the JIT Dylib
>> +  static LLVMErrorRef
>> +  definitionGeneratorFn(LLVMOrcDefinitionGeneratorRef G, void *Ctx,
>> +                        LLVMOrcLookupStateRef *LS, LLVMOrcLookupKind K,
>> +                        LLVMOrcJITDylibRef JD,
>> LLVMOrcJITDylibLookupFlags F,
>> +                        LLVMOrcCLookupSet Names, size_t NamesCount) {
>> +    for (size_t I = 0; I < NamesCount; I++) {
>> +      LLVMOrcCLookupSetElement Element = Names[I];
>> +      LLVMOrcJITTargetAddress Addr =
>> +          (LLVMOrcJITTargetAddress)(&materializationUnitFn);
>> +      LLVMJITSymbolFlags Flags = {LLVMJITSymbolGenericFlagsWeak, 0};
>> +      LLVMJITEvaluatedSymbol Sym = {Addr, Flags};
>> +      LLVMOrcRetainSymbolStringPoolEntry(Element.Name);
>> +      LLVMJITCSymbolMapPair Pair = {Element.Name, Sym};
>> +      LLVMJITCSymbolMapPair Pairs[] = {Pair};
>> +      LLVMOrcMaterializationUnitRef MU = LLVMOrcAbsoluteSymbols(Pairs,
>> 1);
>> +      LLVMErrorRef Err = LLVMOrcJITDylibDefine(JD, MU);
>> +      if (Err)
>> +        return Err;
>> +    }
>> +    return LLVMErrorSuccess;
>> +  }
>> +  // create a test LLVM IR module containing a function named "sum"
>> which has
>> +  // returns the sum of its two parameters
>> +  static LLVMOrcThreadSafeModuleRef createTestModule() {
>> +    LLVMOrcThreadSafeContextRef TSC =
>> LLVMOrcCreateNewThreadSafeContext();
>> +    LLVMContextRef Ctx = LLVMOrcThreadSafeContextGetContext(TSC);
>> +    LLVMModuleRef Mod = LLVMModuleCreateWithNameInContext("test", Ctx);
>> +    {
>> +      LLVMTypeRef Int32Ty = LLVMInt32TypeInContext(Ctx);
>> +      LLVMTypeRef ParamTys[] = {Int32Ty, Int32Ty};
>> +      LLVMTypeRef TestFnTy = LLVMFunctionType(Int32Ty, ParamTys, 2, 0);
>> +      LLVMValueRef TestFn = LLVMAddFunction(Mod, "sum", TestFnTy);
>> +      LLVMBuilderRef IRBuilder = LLVMCreateBuilderInContext(Ctx);
>> +      LLVMBasicBlockRef EntryBB = LLVMAppendBasicBlock(TestFn, "entry");
>> +      LLVMPositionBuilderAtEnd(IRBuilder, EntryBB);
>> +      LLVMValueRef Arg1 = LLVMGetParam(TestFn, 0);
>> +      LLVMValueRef Arg2 = LLVMGetParam(TestFn, 1);
>> +      LLVMValueRef Sum = LLVMBuildAdd(IRBuilder, Arg1, Arg2, "");
>> +      LLVMBuildRet(IRBuilder, Sum);
>> +      LLVMDisposeBuilder(IRBuilder);
>> +    }
>> +    LLVMOrcThreadSafeModuleRef TSM =
>> LLVMOrcCreateNewThreadSafeModule(Mod, TSC);
>> +    LLVMOrcDisposeThreadSafeContext(TSC);
>> +    return TSM;
>> +  }
>> +
>> +  static bool TargetSupported;
>> +};
>> +
>> +bool OrcCAPITestBase::TargetSupported = false;
>> +
>> +TEST_F(OrcCAPITestBase, SymbolStringPoolUniquing) {
>> +  if (!Jit) {
>> +    // TODO: Use GTEST_SKIP() when GTest is updated to version 1.10.0
>> +    return;
>> +  }
>> +
>> +  LLVMOrcSymbolStringPoolEntryRef E1 =
>> +      LLVMOrcExecutionSessionIntern(ExecutionSession, "aaa");
>> +  LLVMOrcSymbolStringPoolEntryRef E2 =
>> +      LLVMOrcExecutionSessionIntern(ExecutionSession, "aaa");
>> +  LLVMOrcSymbolStringPoolEntryRef E3 =
>> +      LLVMOrcExecutionSessionIntern(ExecutionSession, "bbb");
>> +  const char *SymbolName = LLVMOrcSymbolStringPoolEntryStr(E1);
>> +  ASSERT_EQ(E1, E2) << "String pool entries are not unique";
>> +  ASSERT_NE(E1, E3) << "Unique symbol pool entries are equal";
>> +  ASSERT_STREQ("aaa", SymbolName) << "String value of symbol is not
>> equal";
>> +  LLVMOrcReleaseSymbolStringPoolEntry(E1);
>> +  LLVMOrcReleaseSymbolStringPoolEntry(E2);
>> +  LLVMOrcReleaseSymbolStringPoolEntry(E3);
>> +}
>> +
>> +TEST_F(OrcCAPITestBase, JITDylibLookup) {
>> +  if (!Jit) {
>> +    // TODO: Use GTEST_SKIP() when GTest is updated to version 1.10.0
>> +    return;
>> +  }
>> +  LLVMOrcJITDylibRef DoesNotExist =
>> +      LLVMOrcExecutionSessionGetJITDylibByName(ExecutionSession, "test");
>> +  ASSERT_FALSE(!!DoesNotExist);
>> +  LLVMOrcJITDylibRef L1 =
>> +      LLVMOrcExecutionSessionCreateBareJITDylib(ExecutionSession,
>> "test");
>> +  LLVMOrcJITDylibRef L2 =
>> +      LLVMOrcExecutionSessionGetJITDylibByName(ExecutionSession, "test");
>> +  ASSERT_EQ(L1, L2) << "Located JIT Dylib is not equal to original";
>> +}
>> +
>> +TEST_F(OrcCAPITestBase, MaterializationUnitCreation) {
>> +  if (!Jit) {
>> +    // TODO: Use GTEST_SKIP() when GTest is updated to version 1.10.0
>> +    return;
>> +  }
>> +
>> +  LLVMOrcSymbolStringPoolEntryRef Name =
>> +      LLVMOrcLLJITMangleAndIntern(Jit, "test");
>> +  LLVMJITSymbolFlags Flags = {LLVMJITSymbolGenericFlagsWeak, 0};
>> +  LLVMOrcJITTargetAddress Addr =
>> +      (LLVMOrcJITTargetAddress)(&materializationUnitFn);
>> +  LLVMJITEvaluatedSymbol Sym = {Addr, Flags};
>> +  LLVMJITCSymbolMapPair Pair = {Name, Sym};
>> +  LLVMJITCSymbolMapPair Pairs[] = {Pair};
>> +  LLVMOrcMaterializationUnitRef MU = LLVMOrcAbsoluteSymbols(Pairs, 1);
>> +  LLVMOrcJITDylibDefine(MainDylib, MU);
>> +  LLVMOrcJITTargetAddress OutAddr;
>> +  if (LLVMOrcLLJITLookup(Jit, &OutAddr, "test")) {
>> +    FAIL() << "Failed to look up \"test\" symbol";
>> +  }
>> +  ASSERT_EQ(Addr, OutAddr);
>> +}
>> +
>> +TEST_F(OrcCAPITestBase, DefinitionGenerators) {
>> +  if (!Jit) {
>> +    // TODO: Use GTEST_SKIP() when GTest is updated to version 1.10.0
>> +    return;
>> +  }
>> +
>> +  LLVMOrcDefinitionGeneratorRef Gen =
>> +      LLVMOrcCreateCustomCAPIDefinitionGenerator(&definitionGeneratorFn,
>> +                                                 nullptr);
>> +  LLVMOrcJITDylibAddGenerator(MainDylib, Gen);
>> +  LLVMOrcJITTargetAddress OutAddr;
>> +  if (LLVMOrcLLJITLookup(Jit, &OutAddr, "test")) {
>> +    FAIL() << "The DefinitionGenerator did not create symbol \"test\"";
>> +  }
>> +  LLVMOrcJITTargetAddress ExpectedAddr =
>> +      (LLVMOrcJITTargetAddress)(&materializationUnitFn);
>> +  ASSERT_EQ(ExpectedAddr, OutAddr);
>> +}
>> +
>> +TEST_F(OrcCAPITestBase, ResourceTrackerDefinitionLifetime) {
>> +  if (!Jit) {
>> +    // TODO: Use GTEST_SKIP() when GTest is updated to version 1.10.0
>> +    return;
>> +  }
>> +
>> +  // This test case ensures that all symbols loaded into a JITDylib with
>> a
>> +  // ResourceTracker attached are cleared from the JITDylib once the RT
>> is
>> +  // removed.
>> +  LLVMOrcResourceTrackerRef RT =
>> +      LLVMOrcJITDylibCreateResourceTracker(MainDylib);
>> +  LLVMOrcThreadSafeModuleRef TSM = createTestModule();
>> +  if (LLVMErrorRef E = LLVMOrcLLJITAddLLVMIRModuleWithRT(Jit, RT, TSM)) {
>> +    FAIL() << "Failed to add LLVM IR module to LLJIT";
>> +  }
>> +  LLVMOrcJITTargetAddress TestFnAddr;
>> +  if (LLVMOrcLLJITLookup(Jit, &TestFnAddr, "sum")) {
>> +    FAIL() << "Symbol \"sum\" was not added into JIT";
>> +  }
>> +  ASSERT_TRUE(!!TestFnAddr);
>> +  LLVMOrcResourceTrackerRemove(RT);
>> +  LLVMOrcJITTargetAddress OutAddr;
>> +  LLVMErrorRef Err = LLVMOrcLLJITLookup(Jit, &OutAddr, "sum");
>> +  ASSERT_TRUE(Err);
>> +  ASSERT_FALSE(OutAddr);
>> +  LLVMOrcReleaseResourceTracker(RT);
>> +  LLVMConsumeError(Err);
>> +}
>> +
>> +TEST_F(OrcCAPITestBase, ResourceTrackerTransfer) {
>> +  if (!Jit) {
>> +    // TODO: Use GTEST_SKIP() when GTest is updated to version 1.10.0
>> +    return;
>> +  }
>> +
>> +  LLVMOrcResourceTrackerRef DefaultRT =
>> +      LLVMOrcJITDylibGetDefaultResourceTracker(MainDylib);
>> +  LLVMOrcResourceTrackerRef RT2 =
>> +      LLVMOrcJITDylibCreateResourceTracker(MainDylib);
>> +  LLVMOrcThreadSafeModuleRef TSM = createTestModule();
>> +  if (LLVMErrorRef E = LLVMOrcLLJITAddLLVMIRModuleWithRT(Jit, DefaultRT,
>> TSM)) {
>> +    FAIL() << "Failed to add LLVM IR module to LLJIT";
>> +  }
>> +  LLVMOrcJITTargetAddress Addr;
>> +  if (LLVMOrcLLJITLookup(Jit, &Addr, "sum")) {
>> +    FAIL() << "Symbol \"sum\" was not added into JIT";
>> +  }
>> +  LLVMOrcResourceTrackerTransferTo(DefaultRT, RT2);
>> +  LLVMErrorRef Err = LLVMOrcLLJITLookup(Jit, &Addr, "sum");
>> +  ASSERT_FALSE(Err);
>> +  LLVMOrcReleaseResourceTracker(RT2);
>> +}
>> +
>> +TEST_F(OrcCAPITestBase, ExecutionTest) {
>> +  if (!Jit) {
>> +    // TODO: Use GTEST_SKIP() when GTest is updated to version 1.10.0
>> +    return;
>> +  }
>> +
>> +  using SumFunctionType = int32_t (*)(int32_t, int32_t);
>> +
>> +  // This test performs OrcJIT compilation of a simple sum module
>> +  LLVMInitializeNativeAsmPrinter();
>> +  LLVMOrcThreadSafeModuleRef TSM = createTestModule();
>> +  if (LLVMErrorRef E = LLVMOrcLLJITAddLLVMIRModule(Jit, MainDylib, TSM))
>> {
>> +    FAIL() << "Failed to add LLVM IR module to LLJIT";
>> +  }
>> +  LLVMOrcJITTargetAddress TestFnAddr;
>> +  if (LLVMOrcLLJITLookup(Jit, &TestFnAddr, "sum")) {
>> +    FAIL() << "Symbol \"sum\" was not added into JIT";
>> +  }
>> +  auto *SumFn = (SumFunctionType)(TestFnAddr);
>> +  int32_t Result = SumFn(1, 1);
>> +  ASSERT_EQ(2, Result);
>> +}
>>
>>
>>
>> _______________________________________________
>> llvm-commits mailing list
>> llvm-commits at lists.llvm.org
>> https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-commits
>>
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20210427/50f4b8e0/attachment.html>


More information about the llvm-commits mailing list