ptx_parser/parser/instruction/
alloca.rs

1//! Original PTX specification:
2//!
3//! alloca.type  ptr, size{, immAlign};
4//! .type = { .u32, .u64 };
5
6#![allow(unused)]
7
8use crate::lexer::PtxToken;
9use crate::parser::{PtxParseError, PtxParser, PtxTokenStream, Span};
10use crate::r#type::common::*;
11
12pub mod section_0 {
13    use super::*;
14    use crate::r#type::instruction::alloca::section_0::*;
15
16    // ============================================================================
17    // Generated enum parsers
18    // ============================================================================
19
20    impl PtxParser for Type {
21        fn parse(stream: &mut PtxTokenStream) -> Result<Self, PtxParseError> {
22            // Try U32
23            {
24                let saved_pos = stream.position();
25                if stream.expect_string(".u32").is_ok() {
26                    return Ok(Type::U32);
27                }
28                stream.set_position(saved_pos);
29            }
30            let saved_pos = stream.position();
31            // Try U64
32            {
33                let saved_pos = stream.position();
34                if stream.expect_string(".u64").is_ok() {
35                    return Ok(Type::U64);
36                }
37                stream.set_position(saved_pos);
38            }
39            stream.set_position(saved_pos);
40            let span = stream.peek().map(|(_, s)| s.clone()).unwrap_or(Span { start: 0, end: 0 });
41            let expected = &[".u32", ".u64"];
42            let found = stream.peek().map(|(t, _)| format!("{:?}", t)).unwrap_or_else(|_| "<end of input>".to_string());
43            Err(crate::parser::unexpected_value(span, expected, found))
44        }
45    }
46
47    impl PtxParser for AllocaType {
48        fn parse(stream: &mut PtxTokenStream) -> Result<Self, PtxParseError> {
49            stream.expect_string("alloca")?;
50            let type_ = Type::parse(stream)?;
51            stream.expect_complete()?;
52            let ptr = GeneralOperand::parse(stream)?;
53            stream.expect_complete()?;
54            stream.expect(&PtxToken::Comma)?;
55            let size = GeneralOperand::parse(stream)?;
56            stream.expect_complete()?;
57            let saved_pos = stream.position();
58            let has_comma = stream.expect(&PtxToken::Comma).is_ok();
59            if !has_comma {
60                stream.set_position(saved_pos);
61            }
62            let saved_pos = stream.position();
63            let immalign = match GeneralOperand::parse(stream) {
64                Ok(val) => Some(val),
65                Err(_) => {
66                    stream.set_position(saved_pos);
67                    None
68                }
69            };
70            stream.expect_complete()?;
71            stream.expect_complete()?;
72            stream.expect(&PtxToken::Semicolon)?;
73            Ok(AllocaType {
74                type_,
75                ptr,
76                size,
77                immalign,
78            })
79        }
80    }
81
82
83}
84
ptx_parser/parser/instruction/alloca.rs

ptx_parser/parser/instruction/
alloca.rs
