Handle constant arrays with single value

2026-05-29 20:46:07 +03:00 · 2020-03-20 11:25:39 +01:00
parent 1bab67c72b
commit d3167c63f8
1 changed files with 11 additions and 0 deletions
@@ -356,6 +356,17 @@ fn index(&mut self, lhs: &'_ Expr<'_>, index: &'_ Expr<'_>) -> Option<Constant>
                Constant::F64(x) => Some(Constant::F64(x)),
                _ => None,
            },
+            (Some(Constant::Vec(vec)), _) => {
+                if !vec.is_empty() && vec.iter().all(|x| *x == vec[0]) {
+                    match vec[0] {
+                        Constant::F32(x) => Some(Constant::F32(x)),
+                        Constant::F64(x) => Some(Constant::F64(x)),
+                        _ => None,
+                    }
+                } else {
+                    None
+                }
+            },
            _ => None,
        }
    }