ptx_parser/parser/instruction/
dp4a.rs

1//! Original PTX specification:
2//!
3//! dp4a.atype.btype  d, a, b, c;
4//! .atype = .btype = { .u32, .s32 };
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::dp4a::section_0::*;
15
16    // ============================================================================
17    // Generated enum parsers
18    // ============================================================================
19
20    impl PtxParser for Atype {
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(Atype::U32);
27                }
28                stream.set_position(saved_pos);
29            }
30            let saved_pos = stream.position();
31            // Try S32
32            {
33                let saved_pos = stream.position();
34                if stream.expect_string(".s32").is_ok() {
35                    return Ok(Atype::S32);
36                }
37                stream.set_position(saved_pos);
38            }
39            stream.set_position(saved_pos);
40            let span = stream
41                .peek()
42                .map(|(_, s)| s.clone())
43                .unwrap_or(Span { start: 0, end: 0 });
44            let expected = &[".u32", ".s32"];
45            let found = stream
46                .peek()
47                .map(|(t, _)| format!("{:?}", t))
48                .unwrap_or_else(|_| "<end of input>".to_string());
49            Err(crate::parser::unexpected_value(span, expected, found))
50        }
51    }
52
53    impl PtxParser for Btype {
54        fn parse(stream: &mut PtxTokenStream) -> Result<Self, PtxParseError> {
55            // Try U32
56            {
57                let saved_pos = stream.position();
58                if stream.expect_string(".u32").is_ok() {
59                    return Ok(Btype::U32);
60                }
61                stream.set_position(saved_pos);
62            }
63            let saved_pos = stream.position();
64            // Try S32
65            {
66                let saved_pos = stream.position();
67                if stream.expect_string(".s32").is_ok() {
68                    return Ok(Btype::S32);
69                }
70                stream.set_position(saved_pos);
71            }
72            stream.set_position(saved_pos);
73            let span = stream
74                .peek()
75                .map(|(_, s)| s.clone())
76                .unwrap_or(Span { start: 0, end: 0 });
77            let expected = &[".u32", ".s32"];
78            let found = stream
79                .peek()
80                .map(|(t, _)| format!("{:?}", t))
81                .unwrap_or_else(|_| "<end of input>".to_string());
82            Err(crate::parser::unexpected_value(span, expected, found))
83        }
84    }
85
86    impl PtxParser for Dp4aAtypeBtype {
87        fn parse(stream: &mut PtxTokenStream) -> Result<Self, PtxParseError> {
88            stream.expect_string("dp4a")?;
89            let atype = Atype::parse(stream)?;
90            stream.expect_complete()?;
91            let btype = Btype::parse(stream)?;
92            stream.expect_complete()?;
93            let d = GeneralOperand::parse(stream)?;
94            stream.expect_complete()?;
95            stream.expect(&PtxToken::Comma)?;
96            let a = GeneralOperand::parse(stream)?;
97            stream.expect_complete()?;
98            stream.expect(&PtxToken::Comma)?;
99            let b = GeneralOperand::parse(stream)?;
100            stream.expect_complete()?;
101            stream.expect(&PtxToken::Comma)?;
102            let c = GeneralOperand::parse(stream)?;
103            stream.expect_complete()?;
104            stream.expect_complete()?;
105            stream.expect(&PtxToken::Semicolon)?;
106            Ok(Dp4aAtypeBtype {
107                atype,
108                btype,
109                d,
110                a,
111                b,
112                c,
113            })
114        }
115    }
116}
ptx_parser/parser/instruction/dp4a.rs

ptx_parser/parser/instruction/
dp4a.rs
