MLIR  22.0.0git
SPIRVEnums.cpp
Go to the documentation of this file.
1 //===- SPIRVEnums.cpp - MLIR SPIR-V Enums ---------------------------------===//
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 // This file defines the C/C++ enums from SPIR-V spec.
10 //
11 //===----------------------------------------------------------------------===//
12 
14 
15 #include "llvm/ADT/SetVector.h"
16 #include "llvm/ADT/StringExtras.h"
17 
18 using namespace mlir;
19 
20 // Pull in all enum utility function definitions
21 #include "mlir/Dialect/SPIRV/IR/SPIRVEnums.cpp.inc"
22 
23 // Pull in all enum type availability query function definitions
24 #include "mlir/Dialect/SPIRV/IR/SPIRVEnumAvailability.cpp.inc"
25 
26 //===----------------------------------------------------------------------===//
27 // Availability relationship
28 //===----------------------------------------------------------------------===//
29 
31  // Note: the following lists are from "Appendix A: Changes" of the spec.
32 
33 #define V_1_3_IMPLIED_EXTS \
34  Extension::SPV_KHR_shader_draw_parameters, Extension::SPV_KHR_16bit_storage, \
35  Extension::SPV_KHR_device_group, Extension::SPV_KHR_multiview, \
36  Extension::SPV_KHR_storage_buffer_storage_class, \
37  Extension::SPV_KHR_variable_pointers
38 
39 #define V_1_4_IMPLIED_EXTS \
40  Extension::SPV_KHR_no_integer_wrap_decoration, \
41  Extension::SPV_GOOGLE_decorate_string, \
42  Extension::SPV_GOOGLE_hlsl_functionality1, \
43  Extension::SPV_KHR_float_controls
44 
45 #define V_1_5_IMPLIED_EXTS \
46  Extension::SPV_KHR_8bit_storage, Extension::SPV_EXT_descriptor_indexing, \
47  Extension::SPV_EXT_shader_viewport_index_layer, \
48  Extension::SPV_EXT_physical_storage_buffer, \
49  Extension::SPV_KHR_physical_storage_buffer, \
50  Extension::SPV_KHR_vulkan_memory_model
51 
52 #define V_1_6_IMPLIED_EXTS \
53  Extension::SPV_KHR_non_semantic_info, \
54  Extension::SPV_KHR_integer_dot_product, \
55  Extension::SPV_KHR_terminate_invocation, \
56  Extension::SPV_EXT_demote_to_helper_invocation
57 
58  switch (version) {
59  default:
60  return {};
61  case Version::V_1_3: {
62  // The following manual ArrayRef constructor call is to satisfy GCC 5.
63  static const Extension exts[] = {V_1_3_IMPLIED_EXTS};
64  return exts;
65  }
66  case Version::V_1_4: {
67  static const Extension exts[] = {V_1_3_IMPLIED_EXTS, V_1_4_IMPLIED_EXTS};
68  return exts;
69  }
70  case Version::V_1_5: {
71  static const Extension exts[] = {V_1_3_IMPLIED_EXTS, V_1_4_IMPLIED_EXTS,
73  return exts;
74  }
75  case Version::V_1_6: {
76  static const Extension exts[] = {V_1_3_IMPLIED_EXTS, V_1_4_IMPLIED_EXTS,
78  return exts;
79  }
80  }
81 
82 #undef V_1_6_IMPLIED_EXTS
83 #undef V_1_5_IMPLIED_EXTS
84 #undef V_1_4_IMPLIED_EXTS
85 #undef V_1_3_IMPLIED_EXTS
86 }
87 
88 // Pull in utility function definition for implied capabilities
89 #include "mlir/Dialect/SPIRV/IR/SPIRVCapabilityImplication.inc"
90 
92 spirv::getRecursiveImpliedCapabilities(spirv::Capability cap) {
95  llvm::from_range, directCaps);
96 
97  // TODO: This is insufficient; find a better way to handle this
98  // (e.g., using static lists) if this turns out to be a bottleneck.
99  for (unsigned i = 0; i < allCaps.size(); ++i)
100  allCaps.insert_range(getDirectImpliedCapabilities(allCaps[i]));
101 
102  return allCaps.takeVector();
103 }
#define V_1_6_IMPLIED_EXTS
#define V_1_3_IMPLIED_EXTS
#define V_1_4_IMPLIED_EXTS
#define V_1_5_IMPLIED_EXTS
SmallVector< Capability, 0 > getRecursiveImpliedCapabilities(Capability cap)
Returns the recursively implied capabilities for the given capability.
ArrayRef< Extension > getImpliedExtensions(Version version)
Returns the implied extensions for the given version.
ArrayRef< Capability > getDirectImpliedCapabilities(Capability cap)
Returns the directly implied capabilities for the given capability.
Include the generated interface declarations.