lld/unittests/AsLibELF/ROCm.cpp

   1 //===- ROCm.cpp -------------------------------------------------*- C++ -*-===//
   2 //
   3 // This file is licensed 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 // The purpose of this test is to showcase a more singular usage of LLD as a
   9 // library, where only one LLD driver is being used (and linked in the target
  10 // application). We also expect that linking twice the same object files
  11 // would yield a successfull result. When used as library, LLD always cleans its
  12 // internal memory context after each linker call.
  13 //===----------------------------------------------------------------------===//
  14
  15 // When this flag is on, we actually need the MinGW driver library, not the
  16 // ELF one. Here our test only covers the case where the ELF driver is linked
  17 // into the unit test binary.
  18 #ifndef LLD_DEFAULT_LD_LLD_IS_MINGW
  19
  20 #include "lld/Common/Driver.h"
  21 #include "llvm/ADT/SmallString.h"
  22 #include "llvm/Support/FileSystem.h"
  23 #include "llvm/Support/FileUtilities.h"
  24 #include "llvm/Support/Path.h"
  25 #include "gmock/gmock.h"
  26 #include <algorithm>
  27
  28 static std::string expand(const char *path) {
  29   if (!llvm::StringRef(path).contains("%"))
  30     return std::string(path);
  31
  32   llvm::SmallString<256> thisPath;
  33   thisPath.append(getenv("LLD_SRC_DIR"));
  34   llvm::sys::path::append(thisPath, "unittests", "AsLibELF");
  35
  36   std::string expanded(path);
  37   expanded.replace(expanded.find("%S"), 2, thisPath.data(), thisPath.size());
  38   return expanded;
  39 }
  40
  41 LLD_HAS_DRIVER(elf)
  42
  43 static bool lldInvoke(const char *inPath, const char *outPath) {
  44   std::vector<const char *> args{"ld.lld", "-shared", inPath, "-o", outPath};
  45   lld::Result s = lld::lldMain(args, llvm::outs(), llvm::errs(),
  46                                {{lld::Gnu, &lld::elf::link}});
  47   return !s.retCode && s.canRunAgain;
  48 }
  49
  50 static bool runLinker(const char *path) {
  51   // Create a temp file for HSA code object.
  52   int tempHsacoFD = -1;
  53   llvm::SmallString<128> tempHsacoFilename;
  54   if (llvm::sys::fs::createTemporaryFile("kernel", "hsaco", tempHsacoFD,
  55                                          tempHsacoFilename)) {
  56     return false;
  57   }
  58   llvm::FileRemover cleanupHsaco(tempHsacoFilename);
  59   // Invoke lld. Expect a true return value from lld.
  60   std::string expandedPath = expand(path);
  61   if (!lldInvoke(expandedPath.data(), tempHsacoFilename.c_str())) {
  62     llvm::errs() << "Failed to link: " << expandedPath << "\n";
  63     return false;
  64   }
  65   return true;
  66 }
  67
  68 TEST(AsLib, ROCm) {
  69   EXPECT_TRUE(runLinker("%S/Inputs/kernel1.o"));
  70   EXPECT_TRUE(runLinker("%S/Inputs/kernel2.o"));
  71   EXPECT_TRUE(runLinker("%S/Inputs/kernel1.o"));
  72   EXPECT_TRUE(runLinker("%S/Inputs/kernel2.o"));
  73 }
  74 #endif