19#define DEBUG_TYPE "spirv-commandline"
23static const std::map<std::string, SPIRV::Extension::Extension>
25 {
"SPV_EXT_shader_atomic_float_add",
26 SPIRV::Extension::Extension::SPV_EXT_shader_atomic_float_add},
27 {
"SPV_EXT_shader_atomic_float16_add",
28 SPIRV::Extension::Extension::SPV_EXT_shader_atomic_float16_add},
29 {
"SPV_EXT_shader_atomic_float_min_max",
30 SPIRV::Extension::Extension::SPV_EXT_shader_atomic_float_min_max},
31 {
"SPV_INTEL_arbitrary_precision_integers",
32 SPIRV::Extension::Extension::SPV_INTEL_arbitrary_precision_integers},
33 {
"SPV_INTEL_optnone", SPIRV::Extension::Extension::SPV_INTEL_optnone},
34 {
"SPV_INTEL_usm_storage_classes",
35 SPIRV::Extension::Extension::SPV_INTEL_usm_storage_classes},
36 {
"SPV_INTEL_subgroups",
37 SPIRV::Extension::Extension::SPV_INTEL_subgroups},
38 {
"SPV_KHR_uniform_group_instructions",
39 SPIRV::Extension::Extension::SPV_KHR_uniform_group_instructions},
40 {
"SPV_KHR_no_integer_wrap_decoration",
41 SPIRV::Extension::Extension::SPV_KHR_no_integer_wrap_decoration},
42 {
"SPV_KHR_float_controls",
43 SPIRV::Extension::Extension::SPV_KHR_float_controls},
44 {
"SPV_KHR_expect_assume",
45 SPIRV::Extension::Extension::SPV_KHR_expect_assume},
46 {
"SPV_KHR_bit_instructions",
47 SPIRV::Extension::Extension::SPV_KHR_bit_instructions},
48 {
"SPV_KHR_linkonce_odr",
49 SPIRV::Extension::Extension::SPV_KHR_linkonce_odr},
50 {
"SPV_INTEL_bfloat16_conversion",
51 SPIRV::Extension::Extension::SPV_INTEL_bfloat16_conversion},
52 {
"SPV_KHR_subgroup_rotate",
53 SPIRV::Extension::Extension::SPV_KHR_subgroup_rotate},
54 {
"SPV_INTEL_variable_length_array",
55 SPIRV::Extension::Extension::SPV_INTEL_variable_length_array},
56 {
"SPV_INTEL_function_pointers",
57 SPIRV::Extension::Extension::SPV_INTEL_function_pointers},
62 std::set<SPIRV::Extension::Extension> &Vals) {
64 ArgValue.
split(Tokens,
",", -1,
false);
65 std::sort(Tokens.
begin(), Tokens.
end());
67 std::set<SPIRV::Extension::Extension> EnabledExtensions;
69 for (
const auto &Token : Tokens) {
72 EnabledExtensions.insert(ExtensionEnum);
77 if (Token.empty() || (!Token.starts_with(
"+") && !Token.starts_with(
"-")))
78 return O.error(
"Invalid extension list format: " + Token.str());
84 return O.error(
"Unknown SPIR-V extension: " + Token.str());
86 if (Token.starts_with(
"+")) {
87 EnabledExtensions.insert(NameValuePair->second);
88 }
else if (EnabledExtensions.count(NameValuePair->second)) {
89 if (std::find(Tokens.
begin(), Tokens.
end(),
"+" + ExtensionName.
str()) !=
92 "Extension cannot be allowed and disallowed at the same time: " +
95 EnabledExtensions.erase(NameValuePair->second);
99 Vals = std::move(EnabledExtensions);
static const std::map< std::string, SPIRV::Extension::Extension > SPIRVExtensionMap
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
StringRef - Represent a constant reference to a string, i.e.
std::pair< StringRef, StringRef > split(char Separator) const
Split into two substrings around the first occurrence of a separator character.
std::string str() const
str - Get the contents as an std::string.
constexpr StringRef substr(size_t Start, size_t N=npos) const
Return a reference to the substring from [Start, Start + N).
This is an optimization pass for GlobalISel generic memory operations.
bool parse(cl::Option &O, StringRef ArgName, StringRef ArgValue, std::set< SPIRV::Extension::Extension > &Vals)
Parses SPIR-V extension name from CLI arguments.