ptx_parser/parser/instruction/
tanh.rs

1//! Original PTX specification:
2//!
3//! tanh.approx.type d, a;
4//! .type = {.f16, .f32, .f16x2, .bf16, .bf16x2};
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::tanh::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 Bf16x2
23            {
24                let saved_pos = stream.position();
25                if stream.expect_string(".bf16x2").is_ok() {
26                    return Ok(Type::Bf16x2);
27                }
28                stream.set_position(saved_pos);
29            }
30            let saved_pos = stream.position();
31            // Try F16x2
32            {
33                let saved_pos = stream.position();
34                if stream.expect_string(".f16x2").is_ok() {
35                    return Ok(Type::F16x2);
36                }
37                stream.set_position(saved_pos);
38            }
39            stream.set_position(saved_pos);
40            let saved_pos = stream.position();
41            // Try Bf16
42            {
43                let saved_pos = stream.position();
44                if stream.expect_string(".bf16").is_ok() {
45                    return Ok(Type::Bf16);
46                }
47                stream.set_position(saved_pos);
48            }
49            stream.set_position(saved_pos);
50            let saved_pos = stream.position();
51            // Try F16
52            {
53                let saved_pos = stream.position();
54                if stream.expect_string(".f16").is_ok() {
55                    return Ok(Type::F16);
56                }
57                stream.set_position(saved_pos);
58            }
59            stream.set_position(saved_pos);
60            let saved_pos = stream.position();
61            // Try F32
62            {
63                let saved_pos = stream.position();
64                if stream.expect_string(".f32").is_ok() {
65                    return Ok(Type::F32);
66                }
67                stream.set_position(saved_pos);
68            }
69            stream.set_position(saved_pos);
70            let span = stream
71                .peek()
72                .map(|(_, s)| s.clone())
73                .unwrap_or(Span { start: 0, end: 0 });
74            let expected = &[".bf16x2", ".f16x2", ".bf16", ".f16", ".f32"];
75            let found = stream
76                .peek()
77                .map(|(t, _)| format!("{:?}", t))
78                .unwrap_or_else(|_| "<end of input>".to_string());
79            Err(crate::parser::unexpected_value(span, expected, found))
80        }
81    }
82
83    impl PtxParser for TanhApproxType {
84        fn parse(stream: &mut PtxTokenStream) -> Result<Self, PtxParseError> {
85            stream.expect_string("tanh")?;
86            stream.expect_string(".approx")?;
87            let approx = ();
88            stream.expect_complete()?;
89            let type_ = Type::parse(stream)?;
90            stream.expect_complete()?;
91            let d = GeneralOperand::parse(stream)?;
92            stream.expect_complete()?;
93            stream.expect(&PtxToken::Comma)?;
94            let a = GeneralOperand::parse(stream)?;
95            stream.expect_complete()?;
96            stream.expect_complete()?;
97            stream.expect(&PtxToken::Semicolon)?;
98            Ok(TanhApproxType {
99                approx,
100                type_,
101                d,
102                a,
103            })
104        }
105    }
106}
ptx_parser/parser/instruction/tanh.rs

ptx_parser/parser/instruction/
tanh.rs
