llvm/unittests/Transforms/Utils/VFABIUtils.cpp

   1 //===------- VFABIUtils.cpp - VFABI Unittests  ----------------------------===//
   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 #include "llvm/AsmParser/Parser.h"
  10 #include "llvm/IR/InstIterator.h"
  11 #include "llvm/IR/Instructions.h"
  12 #include "llvm/IR/Module.h"
  13 #include "llvm/Support/SourceMgr.h"
  14 #include "llvm/Transforms/Utils/ModuleUtils.h"
  15 #include "gtest/gtest.h"
  16
  17 using namespace llvm;
  18
  19 class VFABIAttrTest : public testing::Test {
  20 protected:
  21   void SetUp() override {
  22     M = parseAssemblyString(IR, Err, Ctx);
  23     // Get the only call instruction in the block, which is the first
  24     // instruction.
  25     CI = dyn_cast<CallInst>(&*(instructions(M->getFunction("f")).begin()));
  26   }
  27   const char *IR = "define i32 @f(i32 %a) {\n"
  28                    " %1 = call i32 @g(i32 %a) #0\n"
  29                    "  ret i32 %1\n"
  30                    "}\n"
  31                    "declare i32 @g(i32)\n"
  32                    "declare <2 x i32> @custom_vg(<2 x i32>)"
  33                    "declare <4 x i32> @_ZGVnN4v_g(<4 x i32>)"
  34                    "declare <8 x i32> @_ZGVnN8v_g(<8 x i32>)"
  35                    "attributes #0 = { "
  36                    "\"vector-function-abi-variant\"=\""
  37                    "_ZGVnN2v_g(custom_vg),_ZGVnN4v_g\" }";
  38   LLVMContext Ctx;
  39   SMDiagnostic Err;
  40   std::unique_ptr<Module> M;
  41   CallInst *CI;
  42   SmallVector<std::string, 8> Mappings;
  43 };
  44
  45 TEST_F(VFABIAttrTest, Write) {
  46   Mappings.push_back("_ZGVnN8v_g");
  47   Mappings.push_back("_ZGVnN2v_g(custom_vg)");
  48   VFABI::setVectorVariantNames(CI, Mappings);
  49   const StringRef S =
  50       CI->getFnAttr("vector-function-abi-variant").getValueAsString();
  51   EXPECT_EQ(S, "_ZGVnN8v_g,_ZGVnN2v_g(custom_vg)");
  52 }