Allow comma or no flag.

gfx-rs · Feb 14, 2025 · 78155a5 · 78155a5
1 parent dd46ed4
commit 78155a5
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/naga/src/front/wgsl/parse/lexer.rs b/naga/src/front/wgsl/parse/lexer.rs
@@ -465,16 +465,20 @@ impl<'a> Lexer<'a> {
     pub(in crate::front::wgsl) fn next_acceleration_structure_flags(
         &mut self,
     ) -> Result<bool, Error<'a>> {
-        Ok(if self.peek().0 == Token::Paren('<') {
-            self.expect(Token::Paren('<'))?;
-            let (name, span) = self.next_ident_with_span()?;
-            let ret = if name == "vertex_return" {
-                true
+        Ok(if self.skip(Token::Paren('<')) {
+            if !self.skip(Token::Paren('>')) {
+                let (name, span) = self.next_ident_with_span()?;
+                let ret = if name == "vertex_return" {
+                    true
+                } else {
+                    return Err(Error::UnknownAttribute(span));
+                };
+                self.skip(Token::Separator(','));
+                self.expect(Token::Paren('>'))?;
+                ret
             } else {
-                return Err(Error::UnknownAttribute(span));
-            };
-            self.expect(Token::Paren('>'))?;
-            ret
+                false
+            }
         } else {
             false
         })