Array VTables 5

Signed-off-by: Nicholas Gates <nick@nickgates.com>

Author: gatesn

encodings/fastlanes/src/bitpacking/vtable/operations.rs

@@ -30,7 +30,6 @@ impl OperationsVTable<BitPacked> for BitPacked {
 #[cfg(test)]
 mod test {
     use std::ops::Range;
-    
 
     use vortex_array::ArrayRef;
     use vortex_array::DynArray;
@@ -45,7 +44,6 @@ mod test {
     use vortex_array::dtype::PType;
     use vortex_array::patches::Patches;
     use vortex_array::scalar::Scalar;
-    
     use vortex_array::validity::Validity;
     use vortex_buffer::Alignment;
     use vortex_buffer::Buffer;

java/testfiles/src/main.rs

@@ -5,16 +5,20 @@
 
 use std::path::Path;
 
+use vortex::VortexSessionDefault;
 use vortex::array::arrays::StructArray;
-use vortex::array::builders::{ArrayBuilder, DecimalBuilder, VarBinViewBuilder};
+use vortex::array::builders::ArrayBuilder;
+use vortex::array::builders::DecimalBuilder;
+use vortex::array::builders::VarBinViewBuilder;
 use vortex::array::validity::Validity;
-use vortex::dtype::{DType, DecimalDType, Nullability};
+use vortex::dtype::DType;
+use vortex::dtype::DecimalDType;
+use vortex::dtype::Nullability;
 use vortex::file::WriteOptionsSessionExt;
-use vortex::io::runtime::current::CurrentThreadRuntime;
 use vortex::io::runtime::BlockingRuntime;
+use vortex::io::runtime::current::CurrentThreadRuntime;
 use vortex::io::session::RuntimeSessionExt;
 use vortex::session::VortexSession;
-use vortex::VortexSessionDefault;
 
 /// Generate a test dataset with the following small set of rows:
 ///
@@ -74,7 +78,8 @@ fn main() {
         10,
         Validity::NonNullable,
     )
-    .expect("Could not create struct array");
+    .expect("Could not create struct array")
+    .to_array_ref();
 
     // Save to file
     let minimal_path = Path::new(env!("CARGO_MANIFEST_DIR"))

vortex-cuda/src/hybrid_dispatch/mod.rs

@@ -88,7 +88,7 @@ pub async fn try_gpu_dispatch(
             // TODO(0ax1): execute subtrees concurrently using separate CUDA streams.
             for subtree in &pending_subtrees {
                 let canonical = subtree.clone().execute_cuda(ctx).await?;
-                subtree_buffers.push(canonical.into_primitive().into_parts().buffer);
+                subtree_buffers.push(canonical.into_primitive().into_data().into_parts().buffer);
             }
 
             let num_subtrees = subtree_buffers.len();

vortex/benches/common_encoding_tree_throughput.rs

@@ -23,7 +23,6 @@ use vortex::array::arrays::TemporalArray;
 use vortex::array::arrays::VarBinArray;
 use vortex::array::arrays::VarBinViewArray;
 use vortex::array::builtins::ArrayBuiltins;
-use vortex::array::vtable::ValidityHelper;
 use vortex::dtype::DType;
 use vortex::dtype::PType;
 use vortex::encodings::alp::ALP;