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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
//===----RTLs/amdgpu/utils/UtilitiesRTL.h ------------------------- 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
//
//===----------------------------------------------------------------------===//
//
// RTL Utilities for AMDGPU plugins
//
//===----------------------------------------------------------------------===//
#include <cstdint>
#include "Debug.h"
#include "omptarget.h"
#include "llvm/ADT/StringMap.h"
#include "llvm/ADT/StringRef.h"
namespace llvm {
namespace omp {
namespace target {
namespace plugin {
namespace utils {
// The implicit arguments of AMDGPU kernels.
struct AMDGPUImplicitArgsTy {
uint64_t OffsetX;
uint64_t OffsetY;
uint64_t OffsetZ;
uint64_t HostcallPtr;
uint64_t Unused0;
uint64_t Unused1;
uint64_t Unused2;
};
static_assert(sizeof(AMDGPUImplicitArgsTy) == 56,
"Unexpected size of implicit arguments");
/// Parse a TargetID to get processor arch and feature map.
/// Returns processor subarch.
/// Returns TargetID features in \p FeatureMap argument.
/// If the \p TargetID contains feature+, FeatureMap it to true.
/// If the \p TargetID contains feature-, FeatureMap it to false.
/// If the \p TargetID does not contain a feature (default), do not map it.
StringRef parseTargetID(StringRef TargetID, StringMap<bool> &FeatureMap) {
if (TargetID.empty())
return llvm::StringRef();
auto ArchFeature = TargetID.split(":");
auto Arch = ArchFeature.first;
auto Features = ArchFeature.second;
if (Features.empty())
return Arch;
if (Features.contains("sramecc+")) {
FeatureMap.insert(std::pair<StringRef, bool>("sramecc", true));
} else if (Features.contains("sramecc-")) {
FeatureMap.insert(std::pair<StringRef, bool>("sramecc", false));
}
if (Features.contains("xnack+")) {
FeatureMap.insert(std::pair<StringRef, bool>("xnack", true));
} else if (Features.contains("xnack-")) {
FeatureMap.insert(std::pair<StringRef, bool>("xnack", false));
}
return Arch;
}
/// Check if an image is compatible with current system's environment.
bool isImageCompatibleWithEnv(const __tgt_image_info *Info,
StringRef EnvTargetID) {
llvm::StringRef ImageTargetID(Info->Arch);
// Compatible in case of exact match.
if (ImageTargetID == EnvTargetID) {
DP("Compatible: Exact match \t[Image: %s]\t:\t[Env: %s]\n",
ImageTargetID.data(), EnvTargetID.data());
return true;
}
// Incompatible if Archs mismatch.
StringMap<bool> ImgMap, EnvMap;
StringRef ImgArch = utils::parseTargetID(ImageTargetID, ImgMap);
StringRef EnvArch = utils::parseTargetID(EnvTargetID, EnvMap);
// Both EnvArch and ImgArch can't be empty here.
if (EnvArch.empty() || ImgArch.empty() || !ImgArch.contains(EnvArch)) {
DP("Incompatible: Processor mismatch \t[Image: %s]\t:\t[Env: %s]\n",
ImageTargetID.data(), EnvTargetID.data());
return false;
}
// Incompatible if image has more features than the environment,
// irrespective of type or sign of features.
if (ImgMap.size() > EnvMap.size()) {
DP("Incompatible: Image has more features than the Environment \t[Image: "
"%s]\t:\t[Env: %s]\n",
ImageTargetID.data(), EnvTargetID.data());
return false;
}
// Compatible if each target feature specified by the environment is
// compatible with target feature of the image. The target feature is
// compatible if the iamge does not specify it (meaning Any), or if it
// specifies it with the same value (meaning On or Off).
for (const auto &ImgFeature : ImgMap) {
auto EnvFeature = EnvMap.find(ImgFeature.first());
if (EnvFeature == EnvMap.end() ||
(EnvFeature->first() == ImgFeature.first() &&
EnvFeature->second != ImgFeature.second)) {
DP("Incompatible: Value of Image's non-ANY feature is not matching with "
"the Environment's non-ANY feature \t[Image: %s]\t:\t[Env: %s]\n",
ImageTargetID.data(), EnvTargetID.data());
return false;
}
}
// Image is compatible if all features of Environment are:
// - either, present in the Image's features map with the same sign,
// - or, the feature is missing from Image's features map i.e. it is
// set to ANY
DP("Compatible: Target IDs are compatible \t[Image: %s]\t:\t[Env: %s]\n",
ImageTargetID.data(), EnvTargetID.data());
return true;
}
} // namespace utils
} // namespace plugin
} // namespace target
} // namespace omp
} // namespace llvm
|
