flang/unittests/Optimizer/FIRContextTest.cpp

   1 //===- FIRContextTest.cpp -------------------------------------------------===//
   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 "flang/Optimizer/Dialect/Support/FIRContext.h"
  10 #include "gtest/gtest.h"
  11 #include "mlir/Dialect/LLVMIR/LLVMAttrs.h"
  12 #include "mlir/Dialect/LLVMIR/LLVMDialect.h"
  13 #include "mlir/IR/BuiltinAttributes.h"
  14 #include "mlir/IR/BuiltinOps.h"
  15 #include "flang/Optimizer/Dialect/Support/KindMapping.h"
  16 #include "llvm/TargetParser/Host.h"
  17 #include <string>
  18
  19 using namespace fir;
  20
  21 struct StringAttributesTests : public testing::Test {
  22 public:
  23   void SetUp() {
  24     context.loadDialect<mlir::LLVM::LLVMDialect>();
  25     kindMap = new KindMapping(&context, kindMapInit, "r42a10c14d28i40l41");
  26     mod = mlir::ModuleOp::create(mlir::UnknownLoc::get(&context));
  27   }
  28
  29   void TearDown() { delete kindMap; }
  30
  31   mlir::MLIRContext context;
  32   KindMapping *kindMap{};
  33   std::string kindMapInit =
  34       "i10:80,l3:24,a1:8,r54:Double,r62:X86_FP80,r11:PPC_FP128";
  35   std::string target = "powerpc64le-unknown-linux-gnu";
  36   std::string targetCPU = "gfx90a";
  37   std::string tuneCPU = "generic";
  38   std::string targetFeatures = "+gfx9-insts,+wavefrontsize64";
  39   mlir::ModuleOp mod;
  40 };
  41
  42 TEST_F(StringAttributesTests, moduleStringAttrTest) {
  43   setTargetTriple(mod, target);
  44   setKindMapping(mod, *kindMap);
  45   setTargetCPU(mod, targetCPU);
  46   setTuneCPU(mod, tuneCPU);
  47   setTargetFeatures(mod, targetFeatures);
  48
  49   auto triple = getTargetTriple(mod);
  50   EXPECT_EQ(triple.getArch(), llvm::Triple::ArchType::ppc64le);
  51   EXPECT_EQ(triple.getOS(), llvm::Triple::OSType::Linux);
  52
  53   auto map = getKindMapping(mod);
  54   EXPECT_EQ(map.defaultsToString(), "a10c14d28i40l41r42");
  55
  56   auto mapStr = map.mapToString();
  57   EXPECT_EQ(mapStr.size(), kindMapInit.size());
  58   EXPECT_TRUE(mapStr.find("a1:8") != std::string::npos);
  59   EXPECT_TRUE(mapStr.find("l3:24") != std::string::npos);
  60   EXPECT_TRUE(mapStr.find("i10:80") != std::string::npos);
  61   EXPECT_TRUE(mapStr.find("r11:PPC_FP128") != std::string::npos);
  62   EXPECT_TRUE(mapStr.find("r54:Double") != std::string::npos);
  63   EXPECT_TRUE(mapStr.find("r62:X86_FP80") != std::string::npos);
  64
  65   EXPECT_EQ(getTargetCPU(mod), targetCPU);
  66   EXPECT_EQ(getTuneCPU(mod), tuneCPU);
  67
  68   auto features = getTargetFeatures(mod);
  69   auto featuresList = features.getFeatures();
  70   EXPECT_EQ(features.getFeaturesString(), targetFeatures);
  71   EXPECT_EQ(featuresList.size(), 2u);
  72   EXPECT_EQ(featuresList[0].str(), "+gfx9-insts");
  73   EXPECT_EQ(featuresList[1].str(), "+wavefrontsize64");
  74 }
  75
  76 // main() from gtest_main