104 lines · cpp
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-exception6//7//===----------------------------------------------------------------------===//8//9// This file defines the C/C++ enums from SPIR-V spec.10//11//===----------------------------------------------------------------------===//12 13#include "mlir/Dialect/SPIRV/IR/SPIRVEnums.h"14 15#include "llvm/ADT/SetVector.h"16#include "llvm/ADT/StringExtras.h"17 18using namespace mlir;19 20// Pull in all enum utility function definitions21#include "mlir/Dialect/SPIRV/IR/SPIRVEnums.cpp.inc"22 23// Pull in all enum type availability query function definitions24#include "mlir/Dialect/SPIRV/IR/SPIRVEnumAvailability.cpp.inc"25 26//===----------------------------------------------------------------------===//27// Availability relationship28//===----------------------------------------------------------------------===//29 30ArrayRef<spirv::Extension> spirv::getImpliedExtensions(spirv::Version version) {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_pointers38 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_controls44 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_model51 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_invocation57 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,72 V_1_5_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,77 V_1_5_IMPLIED_EXTS, V_1_6_IMPLIED_EXTS};78 return exts;79 }80 }81 82#undef V_1_6_IMPLIED_EXTS83#undef V_1_5_IMPLIED_EXTS84#undef V_1_4_IMPLIED_EXTS85#undef V_1_3_IMPLIED_EXTS86}87 88// Pull in utility function definition for implied capabilities89#include "mlir/Dialect/SPIRV/IR/SPIRVCapabilityImplication.inc"90 91SmallVector<spirv::Capability, 0>92spirv::getRecursiveImpliedCapabilities(spirv::Capability cap) {93 ArrayRef<spirv::Capability> directCaps = getDirectImpliedCapabilities(cap);94 SetVector<spirv::Capability, SmallVector<spirv::Capability, 0>> allCaps(95 llvm::from_range, directCaps);96 97 // TODO: This is insufficient; find a better way to handle this98 // (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}104