feat: added a new variant of Constraint struct, to enable support for

discrete constant values

Author: madhav-madhusoodanan

crates/intrinsic-test/src/common/constraint.rs

@@ -5,13 +5,15 @@ use std::ops::Range;
 pub enum Constraint {
     Equal(i64),
     Range(Range<i64>),
+    Set(Vec<i64>),
 }
 
 impl Constraint {
-    pub fn to_range(&self) -> Range<i64> {
+    pub fn to_vector(&self) -> Vec<i64> {
         match self {
-            Constraint::Equal(eq) => *eq..*eq + 1,
-            Constraint::Range(range) => range.clone(),
+            Constraint::Equal(eq) => vec![*eq],
+            Constraint::Range(range) => range.clone().collect::<Vec<i64>>(),
+            Constraint::Set(values) => values.clone(),
         }
     }
 }

crates/intrinsic-test/src/common/gen_c.rs

@@ -40,7 +40,7 @@ pub fn generate_c_constraint_blocks<'a, T: IntrinsicTypeDefinition + 'a>(
     };
 
     let body_indentation = indentation.nested();
-    for i in current.constraint.iter().flat_map(|c| c.to_range()) {
+    for i in current.constraint.iter().flat_map(|c| c.to_vector()) {
         let ty = current.ty.c_type();
 
         writeln!(w, "{indentation}{{")?;

crates/intrinsic-test/src/common/gen_rust.rs

@@ -176,11 +176,7 @@ pub fn generate_rust_constraint_blocks<T: IntrinsicTypeDefinition>(
     name: String,
 ) -> String {
     if let Some((current, constraints)) = constraints.split_last() {
-        let range = current
-            .constraint
-            .iter()
-            .map(|c| c.to_range())
-            .flat_map(|r| r.into_iter());
+        let range = current.constraint.iter().flat_map(|c| c.to_vector());
 
         let body_indentation = indentation.nested();
         range