parse inf and -inf in nnef

sonos · Aug 31, 2023 · 77231e2 · 77231e2
1 parent f53bedf
commit 77231e2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nnef/src/ast/parse.rs b/nnef/src/ast/parse.rs
@@ -388,7 +388,7 @@ pub(super) fn numeric_literal(i: &str) -> IResult<&str, String> {
         recognize(tuple((tag("."), digit0)))(i)
     }
     spaced(map(
-        recognize(tuple((opt(tag("-")), digit1, opt(frac_part), opt(exp_part)))),
+        recognize(tuple((opt(tag("-")), alt((digit1, tag("inf"))), opt(frac_part), opt(exp_part)))),
         |s: &str| s.to_owned(),
     ))(i)
 }

diff --git a/nnef/src/deser.rs b/nnef/src/deser.rs
@@ -510,7 +510,7 @@ impl RValue {
                 ))
             }
             RValue::Literal(Literal::Numeric(f)) => {
-                if f.contains('.') || f.contains('e') {
+                if f.contains('.') || f.contains('e') || f == "inf" || f == "-inf" {
                     f.parse::<f32>()
                         .map(Value::Scalar)
                         .with_context(|| format!("Can not parse {f} as f32"))