[spv-in] allow non-strict check for capabilities

bin/naga.rs

@@ -105,6 +105,7 @@ fn main() {
         "spv" => {
             let options = naga::front::spv::Options {
                 adjust_coordinate_space: params.spv_adjust_coordinate_space,
+                strict_capabilities: false,
                 flow_graph_dump_prefix: params.spv_flow_dump_prefix.map(std::path::PathBuf::from),
             };
             let input = fs::read(input_path).unwrap();

src/front/glsl/parser.rs

@@ -1,4 +1,9 @@
-#![allow(clippy::panic, clippy::needless_lifetimes, clippy::upper_case_acronyms)]
+#![allow(
+    unused_braces,
+    clippy::panic,
+    clippy::needless_lifetimes,
+    clippy::upper_case_acronyms
+)]
 use pomelo::pomelo;
 pomelo! {
     //%verbose;

src/front/spv/mod.rs

@@ -68,7 +68,10 @@ pub const SUPPORTED_CAPABILITIES: &[spirv::Capability] = &[
     spirv::Capability::UniformBufferArrayDynamicIndexing,
     spirv::Capability::StorageBufferArrayDynamicIndexing,
 ];
-pub const SUPPORTED_EXTENSIONS: &[&str] = &["SPV_KHR_vulkan_memory_model"];
+pub const SUPPORTED_EXTENSIONS: &[&str] = &[
+    "SPV_KHR_storage_buffer_storage_class",
+    "SPV_KHR_vulkan_memory_model",
+];
 pub const SUPPORTED_EXT_SETS: &[&str] = &["GLSL.std.450"];
 
 #[derive(Copy, Clone)]
@@ -347,6 +350,8 @@ pub struct Options {
     /// so by default we flip the Y coordinate of the `BuiltIn::Position`.
     /// This flag can be used to avoid this.
     pub adjust_coordinate_space: bool,
+    /// Only allow shaders with the known set of capabilities.
+    pub strict_capabilities: bool,
     pub flow_graph_dump_prefix: Option<PathBuf>,
 }
 
@@ -354,6 +359,7 @@ impl Default for Options {
     fn default() -> Self {
         Options {
             adjust_coordinate_space: true,
+            strict_capabilities: false,
             flow_graph_dump_prefix: None,
         }
     }
@@ -2208,7 +2214,11 @@ impl<I: Iterator<Item = u32>> Parser<I> {
         let cap =
             spirv::Capability::from_u32(capability).ok_or(Error::UnknownCapability(capability))?;
         if !SUPPORTED_CAPABILITIES.contains(&cap) {
-            return Err(Error::UnsupportedCapability(cap));
+            if self.options.strict_capabilities {
+                return Err(Error::UnsupportedCapability(cap));
+            } else {
+                log::warn!("Unknown capability {:?}", cap);
+            }
         }
         Ok(())
     }

tests/snapshots.rs

@@ -263,6 +263,7 @@ fn convert_spv(name: &str, adjust_coordinate_space: bool, targets: Targets) {
         &fs::read(format!("{}/{}/{}.spv", root, DIR_IN, name)).expect("Couldn't find spv file"),
         &naga::front::spv::Options {
             adjust_coordinate_space,
+            strict_capabilities: false,
             flow_graph_dump_prefix: None,
         },
     )