22 spirv::TargetEnv::TargetEnv(spirv::TargetEnvAttr targetAttr)  in TargetEnv()
24   for (spirv::Extension ext : targetAttr.getExtensions())  in TargetEnv()
28   for (spirv::Extension ext :  in TargetEnv()
29        spirv::getImpliedExtensions(targetAttr.getVersion()))  in TargetEnv()
32   for (spirv::Capability cap : targetAttr.getCapabilities()) {  in TargetEnv()
36     for (spirv::Capability c : spirv::getRecursiveImpliedCapabilities(cap))  in TargetEnv()
41 spirv::Version spirv::TargetEnv::getVersion() const {  in getVersion()
45 bool spirv::TargetEnv::allows(spirv::Capability capability) const {  in allows()
49 Optional<spirv::Capability>
50 spirv::TargetEnv::allows(ArrayRef<spirv::Capability> caps) const {  in allows()
51   const auto *chosen = llvm::find_if(caps, [this](spirv::Capability cap) {  in allows()
59 bool spirv::TargetEnv::allows(spirv::Extension extension) const {  in allows()
63 Optional<spirv::Extension>
64 spirv::TargetEnv::allows(ArrayRef<spirv::Extension> exts) const {  in allows()
65   const auto *chosen = llvm::find_if(exts, [this](spirv::Extension ext) {  in allows()
73 spirv::Vendor spirv::TargetEnv::getVendorID() const {  in getVendorID()
77 spirv::DeviceType spirv::TargetEnv::getDeviceType() const {  in getDeviceType()
81 uint32_t spirv::TargetEnv::getDeviceID() const {  in getDeviceID()
85 spirv::ResourceLimitsAttr spirv::TargetEnv::getResourceLimits() const {  in getResourceLimits()
89 MLIRContext *spirv::TargetEnv::getContext() const {  in getContext()
97 StringRef spirv::getInterfaceVarABIAttrName() {  in getInterfaceVarABIAttrName()
101 spirv::InterfaceVarABIAttr
102 spirv::getInterfaceVarABIAttr(unsigned descriptorSet, unsigned binding,  in getInterfaceVarABIAttr()
103                               Optional<spirv::StorageClass> storageClass,  in getInterfaceVarABIAttr()
105   return spirv::InterfaceVarABIAttr::get(descriptorSet, binding, storageClass,  in getInterfaceVarABIAttr()
109 bool spirv::needsInterfaceVarABIAttrs(spirv::TargetEnvAttr targetAttr) {  in needsInterfaceVarABIAttrs()
110   for (spirv::Capability cap : targetAttr.getCapabilities()) {  in needsInterfaceVarABIAttrs()
111     if (cap == spirv::Capability::Kernel)  in needsInterfaceVarABIAttrs()
113     if (cap == spirv::Capability::Shader)  in needsInterfaceVarABIAttrs()
119 StringRef spirv::getEntryPointABIAttrName() { return "spv.entry_point_abi"; }  in getEntryPointABIAttrName()
121 spirv::EntryPointABIAttr
122 spirv::getEntryPointABIAttr(ArrayRef<int32_t> localSize, MLIRContext *context) {  in getEntryPointABIAttr()
124   return spirv::EntryPointABIAttr::get(  in getEntryPointABIAttr()
131 spirv::EntryPointABIAttr spirv::lookupEntryPointABI(Operation *op) {  in lookupEntryPointABI()
137   if (auto attr = op->getAttrOfType<spirv::EntryPointABIAttr>(  in lookupEntryPointABI()
138           spirv::getEntryPointABIAttrName()))  in lookupEntryPointABI()
144 DenseIntElementsAttr spirv::lookupLocalWorkGroupSize(Operation *op) {  in lookupLocalWorkGroupSize()
145   if (auto entryPoint = spirv::lookupEntryPointABI(op))  in lookupLocalWorkGroupSize()
151 spirv::ResourceLimitsAttr
152 spirv::getDefaultResourceLimits(MLIRContext *context) {  in getDefaultResourceLimits()
155   return spirv::ResourceLimitsAttr ::get(  in getDefaultResourceLimits()
163 StringRef spirv::getTargetEnvAttrName() { return "spv.target_env"; }  in getTargetEnvAttrName()
165 spirv::TargetEnvAttr spirv::getDefaultTargetEnv(MLIRContext *context) {  in getDefaultTargetEnv()
166   auto triple = spirv::VerCapExtAttr::get(spirv::Version::V_1_0,  in getDefaultTargetEnv()
167                                           {spirv::Capability::Shader},  in getDefaultTargetEnv()
169   return spirv::TargetEnvAttr::get(triple, spirv::Vendor::Unknown,  in getDefaultTargetEnv()
170                                    spirv::DeviceType::Unknown,  in getDefaultTargetEnv()
171                                    spirv::TargetEnvAttr::kUnknownDeviceID,  in getDefaultTargetEnv()
172                                    spirv::getDefaultResourceLimits(context));  in getDefaultTargetEnv()
175 spirv::TargetEnvAttr spirv::lookupTargetEnv(Operation *op) {  in lookupTargetEnv()
181     if (auto attr = op->getAttrOfType<spirv::TargetEnvAttr>(  in lookupTargetEnv()
182             spirv::getTargetEnvAttrName()))  in lookupTargetEnv()
191 spirv::TargetEnvAttr spirv::lookupTargetEnvOrDefault(Operation *op) {  in lookupTargetEnvOrDefault()
192   if (spirv::TargetEnvAttr attr = spirv::lookupTargetEnv(op))  in lookupTargetEnvOrDefault()
198 spirv::AddressingModel
199 spirv::getAddressingModel(spirv::TargetEnvAttr targetAttr) {  in getAddressingModel()
200   for (spirv::Capability cap : targetAttr.getCapabilities()) {  in getAddressingModel()
204       return spirv::AddressingModel::Physical64;  in getAddressingModel()
207   return spirv::AddressingModel::Logical;  in getAddressingModel()
210 FailureOr<spirv::ExecutionModel>
211 spirv::getExecutionModel(spirv::TargetEnvAttr targetAttr) {  in getExecutionModel()
212   for (spirv::Capability cap : targetAttr.getCapabilities()) {  in getExecutionModel()
213     if (cap == spirv::Capability::Kernel)  in getExecutionModel()
214       return spirv::ExecutionModel::Kernel;  in getExecutionModel()
215     if (cap == spirv::Capability::Shader)  in getExecutionModel()
216       return spirv::ExecutionModel::GLCompute;  in getExecutionModel()
221 FailureOr<spirv::MemoryModel>
222 spirv::getMemoryModel(spirv::TargetEnvAttr targetAttr) {  in getMemoryModel()
223   for (spirv::Capability cap : targetAttr.getCapabilities()) {  in getMemoryModel()
224     if (cap == spirv::Capability::Addresses)  in getMemoryModel()
225       return spirv::MemoryModel::OpenCL;  in getMemoryModel()
226     if (cap == spirv::Capability::Shader)  in getMemoryModel()
227       return spirv::MemoryModel::GLSL450;  in getMemoryModel()