aboutsummaryrefslogtreecommitdiff
path: root/llvm/unittests/Frontend/OpenMPDirectiveNameTest.cpp
blob: da648157ee9b60fce9c5857433e100ce028414bd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
//===- llvm/unittests/Frontend/OpenMPDirectiveNameTest.cpp ----------------===//
//
// 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/ADT/DenseMap.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Frontend/OpenMP/OMP.h"
#include "gtest/gtest.h"

using namespace llvm;
using namespace llvm::omp;

const DenseMap<Directive, StringRef> &Expected52() {
  static const DenseMap<Directive, StringRef> Names{
      {OMPD_begin_declare_target, "begin declare target"},
      {OMPD_begin_declare_variant, "begin declare variant"},
      {OMPD_cancellation_point, "cancellation point"},
      {OMPD_declare_mapper, "declare mapper"},
      {OMPD_declare_reduction, "declare reduction"},
      {OMPD_declare_simd, "declare simd"},
      {OMPD_declare_target, "declare target"},
      {OMPD_declare_variant, "declare variant"},
      {OMPD_end_declare_target, "end declare target"},
      {OMPD_end_declare_variant, "end declare variant"},
      {OMPD_target_data, "target data"},
      {OMPD_target_enter_data, "target enter data"},
      {OMPD_target_exit_data, "target exit data"},
      {OMPD_target_update, "target update"},
  };
  return Names;
}

const DenseMap<Directive, StringRef> &Expected60() {
  static const DenseMap<Directive, StringRef> Names{
      {OMPD_begin_declare_target, "begin declare_target"},
      {OMPD_begin_declare_variant, "begin declare_variant"},
      {OMPD_cancellation_point, "cancellation_point"},
      {OMPD_declare_mapper, "declare_mapper"},
      {OMPD_declare_reduction, "declare_reduction"},
      {OMPD_declare_simd, "declare_simd"},
      {OMPD_declare_target, "declare_target"},
      {OMPD_declare_variant, "declare_variant"},
      {OMPD_end_declare_target, "end declare_target"},
      {OMPD_end_declare_variant, "end declare_variant"},
      {OMPD_target_data, "target_data"},
      {OMPD_target_enter_data, "target_enter_data"},
      {OMPD_target_exit_data, "target_exit_data"},
      {OMPD_target_update, "target_update"},
  };
  return Names;
}

class VersionTest : public testing::TestWithParam<unsigned> {
public:
  void SetUp() override {
    Version = GetParam();

    if (Version < 60)
      KindToName = &Expected52();
    else
      KindToName = &Expected60();
  }

  const DenseMap<Directive, StringRef> *KindToName;
  unsigned Version;
};

INSTANTIATE_TEST_SUITE_P(OpenMPDirectiveNames, VersionTest,
                         testing::ValuesIn(getOpenMPVersions()));

TEST_P(VersionTest, DirectiveName) {
  for (auto [Kind, Name] : *KindToName)
    ASSERT_EQ(Name, getOpenMPDirectiveName(Kind, Version));
}

TEST(OpenMPDirectiveNames, DirectiveKind52) {
  for (auto [Kind, Name] : Expected52()) {
    auto [K, R] = getOpenMPDirectiveKindAndVersions(Name);
    ASSERT_EQ(K, Kind);
    // Expect the name to be valid in 5.2, but not in 6.0.
    EXPECT_TRUE(52 <= R.Max && R.Max < 60);
  }
}

TEST(OpenMPDirectiveNames, DirectiveKind60) {
  for (auto [Kind, Name] : Expected60()) {
    auto [K, R] = getOpenMPDirectiveKindAndVersions(Name);
    ASSERT_EQ(K, Kind);
    // Expect the name to be valid in 6.0 and later.
    EXPECT_TRUE(60 <= R.Min);
  }
}