[spirv] Define a few more extensions in SPIRVBase.td
PiperOrigin-RevId: 283798496
Change-Id: I24b5782eac3eb0660836cb0c0a41d9e86442499f
diff --git a/third_party/mlir/include/mlir/Dialect/SPIRV/SPIRVBase.td b/third_party/mlir/include/mlir/Dialect/SPIRV/SPIRVBase.td
index 2ee8f3b..dd15895 100644
--- a/third_party/mlir/include/mlir/Dialect/SPIRV/SPIRVBase.td
+++ b/third_party/mlir/include/mlir/Dialect/SPIRV/SPIRVBase.td
@@ -62,20 +62,60 @@
// https://github.com/KhronosGroup/SPIRV-Registry has the full list.
def SPV_KHR_16bit_storage : StrEnumAttrCase<"SPV_KHR_16bit_storage">;
def SPV_KHR_8bit_storage : StrEnumAttrCase<"SPV_KHR_8bit_storage">;
+def SPV_KHR_device_group : StrEnumAttrCase<"SPV_KHR_device_group">;
def SPV_KHR_float_controls : StrEnumAttrCase<"SPV_KHR_float_controls">;
+def SPV_KHR_physical_storage_buffer : StrEnumAttrCase<"SPV_KHR_physical_storage_buffer">;
+def SPV_KHR_multiview : StrEnumAttrCase<"SPV_KHR_multiview">;
+def SPV_KHR_no_integer_wrap_decoration : StrEnumAttrCase<"SPV_KHR_no_integer_wrap_decoration">;
+def SPV_KHR_post_depth_coverage : StrEnumAttrCase<"SPV_KHR_post_depth_coverage">;
def SPV_KHR_shader_atomic_counter_ops : StrEnumAttrCase<"SPV_KHR_shader_atomic_counter_ops">;
def SPV_KHR_shader_ballot : StrEnumAttrCase<"SPV_KHR_shader_ballot">;
+def SPV_KHR_shader_draw_parameters : StrEnumAttrCase<"SPV_KHR_shader_draw_parameters">;
def SPV_KHR_storage_buffer_storage_class : StrEnumAttrCase<"SPV_KHR_storage_buffer_storage_class">;
def SPV_KHR_subgroup_vote : StrEnumAttrCase<"SPV_KHR_subgroup_vote">;
def SPV_KHR_variable_pointers : StrEnumAttrCase<"SPV_KHR_variable_pointers">;
def SPV_KHR_vulkan_memory_model : StrEnumAttrCase<"SPV_KHR_vulkan_memory_model">;
+def SPV_EXT_fragment_fully_covered : StrEnumAttrCase<"SPV_EXT_fragment_fully_covered">;
+def SPV_EXT_fragment_invocation_density : StrEnumAttrCase<"SPV_EXT_fragment_invocation_density">;
+def SPV_EXT_fragment_shader_interlock : StrEnumAttrCase<"SPV_EXT_fragment_shader_interlock">;
+def SPV_EXT_physical_storage_buffer : StrEnumAttrCase<"SPV_EXT_physical_storage_buffer">;
+def SPV_EXT_shader_stencil_export : StrEnumAttrCase<"SPV_EXT_shader_stencil_export">;
+
+def SPV_AMD_shader_explicit_vertex_parameter : StrEnumAttrCase<"SPV_AMD_shader_explicit_vertex_parameter">;
+
+def SPV_GOOGLE_user_type : StrEnumAttrCase<"SPV_GOOGLE_user_type">;
+
+def SPV_NV_compute_shader_derivatives : StrEnumAttrCase<"SPV_NV_compute_shader_derivatives">;
+def SPV_NV_fragment_shader_barycentric : StrEnumAttrCase<"SPV_NV_fragment_shader_barycentric">;
+def SPV_NV_geometry_shader_passthrough : StrEnumAttrCase<"SPV_NV_geometry_shader_passthrough">;
+def SPV_NV_mesh_shader : StrEnumAttrCase<"SPV_NV_mesh_shader">;
+def SPV_NV_ray_tracing : StrEnumAttrCase<"SPV_NV_ray_tracing">;
+def SPV_NV_sample_mask_override_coverage : StrEnumAttrCase<"SPV_NV_sample_mask_override_coverage">;
+def SPV_NV_shader_sm_builtins : StrEnumAttrCase<"SPV_NV_shader_sm_builtins">;
+def SPV_NV_shading_rate : StrEnumAttrCase<"SPV_NV_shading_rate">;
+def SPV_NV_stereo_view_rendering : StrEnumAttrCase<"SPV_NV_stereo_view_rendering">;
+def SPV_NV_viewport_array2 : StrEnumAttrCase<"SPV_NV_viewport_array2">;
+
+def SPV_NVX_multiview_per_view_attributes : StrEnumAttrCase<"SPV_NVX_multiview_per_view_attributes">;
+
def SPV_ExtensionAttr :
StrEnumAttr<"Extension", "supported SPIR-V extensions", [
- SPV_KHR_16bit_storage, SPV_KHR_8bit_storage, SPV_KHR_float_controls,
+ SPV_KHR_16bit_storage, SPV_KHR_8bit_storage, SPV_KHR_device_group,
+ SPV_KHR_float_controls, SPV_KHR_physical_storage_buffer, SPV_KHR_multiview,
+ SPV_KHR_no_integer_wrap_decoration, SPV_KHR_post_depth_coverage,
SPV_KHR_shader_atomic_counter_ops, SPV_KHR_shader_ballot,
- SPV_KHR_storage_buffer_storage_class, SPV_KHR_subgroup_vote,
- SPV_KHR_variable_pointers, SPV_KHR_vulkan_memory_model
+ SPV_KHR_shader_draw_parameters, SPV_KHR_storage_buffer_storage_class,
+ SPV_KHR_subgroup_vote, SPV_KHR_variable_pointers,
+ SPV_KHR_vulkan_memory_model, SPV_EXT_fragment_fully_covered,
+ SPV_EXT_fragment_invocation_density, SPV_EXT_fragment_shader_interlock,
+ SPV_EXT_physical_storage_buffer, SPV_EXT_shader_stencil_export,
+ SPV_AMD_shader_explicit_vertex_parameter, SPV_GOOGLE_user_type,
+ SPV_NV_compute_shader_derivatives, SPV_NV_fragment_shader_barycentric,
+ SPV_NV_geometry_shader_passthrough, SPV_NV_mesh_shader, SPV_NV_ray_tracing,
+ SPV_NV_sample_mask_override_coverage, SPV_NV_shader_sm_builtins,
+ SPV_NV_shading_rate, SPV_NV_stereo_view_rendering,
+ SPV_NV_viewport_array2, SPV_NVX_multiview_per_view_attributes,
]> {
let cppNamespace = "::mlir::spirv";
}