1a9ac8606Spatrick //===--- OpenCLOptions.cpp---------------------------------------*- C++ -*-===//
2a9ac8606Spatrick //
3a9ac8606Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4a9ac8606Spatrick // See https://llvm.org/LICENSE.txt for license information.
5a9ac8606Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6a9ac8606Spatrick //
7a9ac8606Spatrick //===----------------------------------------------------------------------===//
8a9ac8606Spatrick
9a9ac8606Spatrick #include "clang/Basic/OpenCLOptions.h"
10a9ac8606Spatrick #include "clang/Basic/Diagnostic.h"
11a9ac8606Spatrick #include "clang/Basic/TargetInfo.h"
12a9ac8606Spatrick
13a9ac8606Spatrick namespace clang {
14a9ac8606Spatrick
15*12c85518Srobert // First feature in a pair requires the second one to be supported.
16*12c85518Srobert static const std::pair<StringRef, StringRef> DependentFeaturesList[] = {
17*12c85518Srobert {"__opencl_c_read_write_images", "__opencl_c_images"},
18*12c85518Srobert {"__opencl_c_3d_image_writes", "__opencl_c_images"},
19*12c85518Srobert {"__opencl_c_pipes", "__opencl_c_generic_address_space"},
20*12c85518Srobert {"__opencl_c_device_enqueue", "__opencl_c_generic_address_space"},
21*12c85518Srobert {"__opencl_c_device_enqueue", "__opencl_c_program_scope_global_variables"}};
22*12c85518Srobert
23*12c85518Srobert // Extensions and equivalent feature pairs.
24*12c85518Srobert static const std::pair<StringRef, StringRef> FeatureExtensionMap[] = {
25*12c85518Srobert {"cl_khr_fp64", "__opencl_c_fp64"},
26*12c85518Srobert {"cl_khr_3d_image_writes", "__opencl_c_3d_image_writes"}};
27*12c85518Srobert
isKnown(llvm::StringRef Ext) const28a9ac8606Spatrick bool OpenCLOptions::isKnown(llvm::StringRef Ext) const {
29a9ac8606Spatrick return OptMap.find(Ext) != OptMap.end();
30a9ac8606Spatrick }
31a9ac8606Spatrick
isAvailableOption(llvm::StringRef Ext,const LangOptions & LO) const32a9ac8606Spatrick bool OpenCLOptions::isAvailableOption(llvm::StringRef Ext,
33a9ac8606Spatrick const LangOptions &LO) const {
34a9ac8606Spatrick if (!isKnown(Ext))
35a9ac8606Spatrick return false;
36a9ac8606Spatrick
37a9ac8606Spatrick auto &OptInfo = OptMap.find(Ext)->getValue();
38a9ac8606Spatrick if (OptInfo.isCoreIn(LO) || OptInfo.isOptionalCoreIn(LO))
39a9ac8606Spatrick return isSupported(Ext, LO);
40a9ac8606Spatrick
41a9ac8606Spatrick return isEnabled(Ext);
42a9ac8606Spatrick }
43a9ac8606Spatrick
isEnabled(llvm::StringRef Ext) const44a9ac8606Spatrick bool OpenCLOptions::isEnabled(llvm::StringRef Ext) const {
45a9ac8606Spatrick auto I = OptMap.find(Ext);
46a9ac8606Spatrick return I != OptMap.end() && I->getValue().Enabled;
47a9ac8606Spatrick }
48a9ac8606Spatrick
isWithPragma(llvm::StringRef Ext) const49a9ac8606Spatrick bool OpenCLOptions::isWithPragma(llvm::StringRef Ext) const {
50a9ac8606Spatrick auto E = OptMap.find(Ext);
51a9ac8606Spatrick return E != OptMap.end() && E->second.WithPragma;
52a9ac8606Spatrick }
53a9ac8606Spatrick
isSupported(llvm::StringRef Ext,const LangOptions & LO) const54a9ac8606Spatrick bool OpenCLOptions::isSupported(llvm::StringRef Ext,
55a9ac8606Spatrick const LangOptions &LO) const {
56a9ac8606Spatrick auto I = OptMap.find(Ext);
57a9ac8606Spatrick return I != OptMap.end() && I->getValue().Supported &&
58a9ac8606Spatrick I->getValue().isAvailableIn(LO);
59a9ac8606Spatrick }
60a9ac8606Spatrick
isSupportedCore(llvm::StringRef Ext,const LangOptions & LO) const61a9ac8606Spatrick bool OpenCLOptions::isSupportedCore(llvm::StringRef Ext,
62a9ac8606Spatrick const LangOptions &LO) const {
63a9ac8606Spatrick auto I = OptMap.find(Ext);
64a9ac8606Spatrick return I != OptMap.end() && I->getValue().Supported &&
65a9ac8606Spatrick I->getValue().isCoreIn(LO);
66a9ac8606Spatrick }
67a9ac8606Spatrick
isSupportedOptionalCore(llvm::StringRef Ext,const LangOptions & LO) const68a9ac8606Spatrick bool OpenCLOptions::isSupportedOptionalCore(llvm::StringRef Ext,
69a9ac8606Spatrick const LangOptions &LO) const {
70a9ac8606Spatrick auto I = OptMap.find(Ext);
71a9ac8606Spatrick return I != OptMap.end() && I->getValue().Supported &&
72a9ac8606Spatrick I->getValue().isOptionalCoreIn(LO);
73a9ac8606Spatrick }
74a9ac8606Spatrick
isSupportedCoreOrOptionalCore(llvm::StringRef Ext,const LangOptions & LO) const75a9ac8606Spatrick bool OpenCLOptions::isSupportedCoreOrOptionalCore(llvm::StringRef Ext,
76a9ac8606Spatrick const LangOptions &LO) const {
77a9ac8606Spatrick return isSupportedCore(Ext, LO) || isSupportedOptionalCore(Ext, LO);
78a9ac8606Spatrick }
79a9ac8606Spatrick
isSupportedExtension(llvm::StringRef Ext,const LangOptions & LO) const80a9ac8606Spatrick bool OpenCLOptions::isSupportedExtension(llvm::StringRef Ext,
81a9ac8606Spatrick const LangOptions &LO) const {
82a9ac8606Spatrick auto I = OptMap.find(Ext);
83a9ac8606Spatrick return I != OptMap.end() && I->getValue().Supported &&
84a9ac8606Spatrick I->getValue().isAvailableIn(LO) &&
85a9ac8606Spatrick !isSupportedCoreOrOptionalCore(Ext, LO);
86a9ac8606Spatrick }
87a9ac8606Spatrick
enable(llvm::StringRef Ext,bool V)88a9ac8606Spatrick void OpenCLOptions::enable(llvm::StringRef Ext, bool V) {
89a9ac8606Spatrick OptMap[Ext].Enabled = V;
90a9ac8606Spatrick }
91a9ac8606Spatrick
acceptsPragma(llvm::StringRef Ext,bool V)92a9ac8606Spatrick void OpenCLOptions::acceptsPragma(llvm::StringRef Ext, bool V) {
93a9ac8606Spatrick OptMap[Ext].WithPragma = V;
94a9ac8606Spatrick }
95a9ac8606Spatrick
support(llvm::StringRef Ext,bool V)96a9ac8606Spatrick void OpenCLOptions::support(llvm::StringRef Ext, bool V) {
97a9ac8606Spatrick assert(!Ext.empty() && "Extension is empty.");
98a9ac8606Spatrick assert(Ext[0] != '+' && Ext[0] != '-');
99a9ac8606Spatrick OptMap[Ext].Supported = V;
100a9ac8606Spatrick }
101a9ac8606Spatrick
OpenCLOptions()102a9ac8606Spatrick OpenCLOptions::OpenCLOptions() {
103a9ac8606Spatrick #define OPENCL_GENERIC_EXTENSION(Ext, ...) \
104a9ac8606Spatrick OptMap.insert_or_assign(#Ext, OpenCLOptionInfo{__VA_ARGS__});
105a9ac8606Spatrick #include "clang/Basic/OpenCLExtensions.def"
106a9ac8606Spatrick }
107a9ac8606Spatrick
addSupport(const llvm::StringMap<bool> & FeaturesMap,const LangOptions & Opts)108a9ac8606Spatrick void OpenCLOptions::addSupport(const llvm::StringMap<bool> &FeaturesMap,
109a9ac8606Spatrick const LangOptions &Opts) {
110a9ac8606Spatrick for (const auto &F : FeaturesMap) {
111a9ac8606Spatrick const auto &Name = F.getKey();
112a9ac8606Spatrick if (F.getValue() && isKnown(Name) && OptMap[Name].isAvailableIn(Opts))
113a9ac8606Spatrick support(Name);
114a9ac8606Spatrick }
115a9ac8606Spatrick }
116a9ac8606Spatrick
disableAll()117a9ac8606Spatrick void OpenCLOptions::disableAll() {
118a9ac8606Spatrick for (auto &Opt : OptMap)
119a9ac8606Spatrick Opt.getValue().Enabled = false;
120a9ac8606Spatrick }
121a9ac8606Spatrick
diagnoseUnsupportedFeatureDependencies(const TargetInfo & TI,DiagnosticsEngine & Diags)122a9ac8606Spatrick bool OpenCLOptions::diagnoseUnsupportedFeatureDependencies(
123a9ac8606Spatrick const TargetInfo &TI, DiagnosticsEngine &Diags) {
124a9ac8606Spatrick auto OpenCLFeaturesMap = TI.getSupportedOpenCLOpts();
125a9ac8606Spatrick
126a9ac8606Spatrick bool IsValid = true;
127*12c85518Srobert for (auto &FeaturePair : DependentFeaturesList) {
128*12c85518Srobert auto Feature = FeaturePair.first;
129*12c85518Srobert auto Dep = FeaturePair.second;
130*12c85518Srobert if (TI.hasFeatureEnabled(OpenCLFeaturesMap, Feature) &&
131*12c85518Srobert !TI.hasFeatureEnabled(OpenCLFeaturesMap, Dep)) {
132a9ac8606Spatrick IsValid = false;
133*12c85518Srobert Diags.Report(diag::err_opencl_feature_requires) << Feature << Dep;
134*12c85518Srobert }
135a9ac8606Spatrick }
136a9ac8606Spatrick return IsValid;
137a9ac8606Spatrick }
138a9ac8606Spatrick
diagnoseFeatureExtensionDifferences(const TargetInfo & TI,DiagnosticsEngine & Diags)139a9ac8606Spatrick bool OpenCLOptions::diagnoseFeatureExtensionDifferences(
140a9ac8606Spatrick const TargetInfo &TI, DiagnosticsEngine &Diags) {
141a9ac8606Spatrick auto OpenCLFeaturesMap = TI.getSupportedOpenCLOpts();
142a9ac8606Spatrick
143a9ac8606Spatrick bool IsValid = true;
144a9ac8606Spatrick for (auto &ExtAndFeat : FeatureExtensionMap)
145*12c85518Srobert if (TI.hasFeatureEnabled(OpenCLFeaturesMap, ExtAndFeat.first) !=
146*12c85518Srobert TI.hasFeatureEnabled(OpenCLFeaturesMap, ExtAndFeat.second)) {
147a9ac8606Spatrick IsValid = false;
148a9ac8606Spatrick Diags.Report(diag::err_opencl_extension_and_feature_differs)
149*12c85518Srobert << ExtAndFeat.first << ExtAndFeat.second;
150a9ac8606Spatrick }
151a9ac8606Spatrick return IsValid;
152a9ac8606Spatrick }
153a9ac8606Spatrick
154a9ac8606Spatrick } // end namespace clang
155