perf: Use sum compute function (#4207)

This patch finishes two todo items to replace with `SUM` compute
function in `Validity`'s `all_valid` and `all_invalid` functions.

Signed-off-by: Liang-Chi Hsieh <[email protected]>

Author: viirya

vortex-array/src/validity.rs

@@ -91,21 +91,27 @@ impl Validity {
         Ok(match self {
             Validity::NonNullable | Validity::AllValid => true,
             Validity::AllInvalid => false,
-            Validity::Array(array) => {
-                // TODO(ngates): replace with SUM compute function
-                array.to_bool()?.boolean_buffer().count_set_bits() == array.len()
-            }
+            Validity::Array(array) => sum(array)
+                .map(|v| {
+                    v.as_primitive()
+                        .typed_value::<u64>()
+                        .map(|count| count == array.len() as u64)
+                })?
+                .ok_or_else(|| vortex_err!("Failed to compute sum for validity array"))?,
         })
     }
 
     pub fn all_invalid(&self) -> VortexResult<bool> {
         Ok(match self {
             Validity::NonNullable | Validity::AllValid => false,
             Validity::AllInvalid => true,
-            Validity::Array(array) => {
-                // TODO(ngates): replace with SUM compute function
-                array.to_bool()?.boolean_buffer().count_set_bits() == 0
-            }
+            Validity::Array(array) => sum(array)
+                .map(|v| {
+                    v.as_primitive()
+                        .typed_value::<u64>()
+                        .map(|count| count == 0u64)
+                })?
+                .ok_or_else(|| vortex_err!("Failed to compute sum for validity array"))?,
         })
     }