Add compress/_z to Dual/Compensated

Author: novacrazy

crates/thermite-compensated/src/lib.rs

@@ -1408,6 +1408,22 @@ impl<V: CompensatedFloatVector> GenericVector for Compensated<V> {
         }
     }
 
+    #[inline(always)]
+    fn compress(self, mask: Self::Mask) -> Self {
+        Self {
+            value: self.value.compress(mask),
+            error: self.error.compress(mask),
+        }
+    }
+
+    #[inline(always)]
+    fn compress_z(self, mask: Self::Mask) -> Self {
+        Self {
+            value: self.value.compress_z(mask),
+            error: self.error.compress_z(mask),
+        }
+    }
+
 
     fn map<F>(mut self, f: F) -> Self
     where

crates/thermite-dual/src/vector.rs

@@ -576,6 +576,22 @@ impl<V: DualFloatVector, const N: usize> GenericVector for Dual<V, N> {
         }
     }
 
+    #[inline(always)]
+    fn compress(self, mask: Self::Mask) -> Self {
+        Self {
+            re: self.re.compress(mask),
+            dual: array_each!([V::ZERO; N], |j| self.dual[j].compress(mask)),
+        }
+    }
+
+    #[inline(always)]
+    fn compress_z(self, mask: Self::Mask) -> Self {
+        Self {
+            re: self.re.compress_z(mask),
+            dual: array_each!([V::ZERO; N], |j| self.dual[j].compress_z(mask)),
+        }
+    }
+
     #[inline(always)] fn map<F>(mut self, f: F) -> Self
     where
         F: Fn(Self::Element) -> Self::Element,