chore: further optimization to comparison function

Author: madhav-madhusoodanan

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

@@ -1,6 +1,7 @@
 use itertools::Itertools;
 use rayon::prelude::*;
-use std::{collections::HashMap, process::Command};
+use std::collections::{HashMap, HashSet};
+use std::process::Command;
 
 pub const INTRINSIC_DELIMITER: &str = "############";
 fn runner_command(runner: &str) -> Command {
@@ -56,23 +57,34 @@ pub fn compare_outputs(intrinsic_name_list: &Vec<String>, runner: &str, target:
         .to_lowercase()
         .replace("-nan", "nan");
 
-    let c_output_map = c
+    let c_output_set = c
         .split(INTRINSIC_DELIMITER)
         .filter_map(|output| output.trim().split_once("\n"))
-        .collect::<HashMap<&str, &str>>();
-    let rust_output_map = rust
+        .collect::<HashSet<(&str, &str)>>();
+    let rust_output_set = rust
         .split(INTRINSIC_DELIMITER)
         .filter_map(|output| output.trim().split_once("\n"))
+        .collect::<HashSet<(&str, &str)>>();
+
+    let intrinsics = rust_output_set
+        .symmetric_difference(&c_output_set)
+        .map(|&(x, _)| x)
+        .collect::<HashSet<&str>>();
+
+    let c_output_map = c_output_set
+        .into_iter()
+        .filter_map(|(key, value)| intrinsics.get(key).zip(Some(value)))
+        .map(|(&key, value)| (key, value))
+        .collect::<HashMap<&str, &str>>();
+    let rust_output_map = rust_output_set
+        .into_iter()
+        .filter_map(|(key, value)| intrinsics.get(key).zip(Some(value)))
+        .map(|(&key, value)| (key, value))
         .collect::<HashMap<&str, &str>>();
 
-    let intrinsics = c_output_map
-        .keys()
-        .chain(rust_output_map.keys())
-        .unique()
-        .collect_vec();
     let intrinsics_diff_count = intrinsics
         .par_iter()
-        .filter_map(|&&intrinsic| {
+        .filter_map(|&intrinsic| {
             let c_output = c_output_map.get(intrinsic).unwrap();
             let rust_output = rust_output_map.get(intrinsic).unwrap();
             if rust_output.eq(c_output) {