fix

Signed-off-by: Joe Isaacs <joe.isaacs@live.co.uk>

Author: joseph-isaacs

vortex/benches/common_encoding_tree_throughput.rs

@@ -95,10 +95,11 @@ mod setup {
 
     /// Create FoR <- BitPacked encoding tree for u64
     pub fn for_bp_u64() -> ArrayRef {
+        let mut ctx = LEGACY_SESSION.create_execution_ctx();
         let (uint_array, ..) = setup_primitive_arrays();
         let compressed = FoR::encode(uint_array).unwrap();
         let inner = compressed.encoded();
-        let bp = BitPacked::encode(inner, 8).unwrap();
+        let bp = BitPacked::encode(inner, 8, &mut ctx).unwrap();
         FoR::try_new(bp.into_array(), compressed.reference_scalar().clone())
             .unwrap()
             .into_array()
@@ -118,7 +119,7 @@ mod setup {
             .unwrap();
         let for_array = FoR::encode(alp_encoded_prim).unwrap();
         let inner = for_array.encoded();
-        let bp = BitPacked::encode(inner, 8).unwrap();
+        let bp = BitPacked::encode(inner, 8, &mut ctx).unwrap();
         let for_with_bp =
             FoR::try_new(bp.into_array(), for_array.reference_scalar().clone()).unwrap();
 
@@ -153,7 +154,8 @@ mod setup {
         let codes_prim = PrimitiveArray::from_iter(codes);
 
         // Compress codes with BitPacked (6 bits should be enough for ~50 unique values)
-        let codes_bp = BitPacked::encode(&codes_prim.into_array(), 6)
+        let mut ctx = LEGACY_SESSION.create_execution_ctx();
+        let codes_bp = BitPacked::encode(&codes_prim.into_array(), 6, &mut ctx)
             .unwrap()
             .into_array();
 
@@ -195,7 +197,7 @@ mod setup {
             .unwrap();
         let ends_for = FoR::encode(ends_prim).unwrap();
         let ends_inner = ends_for.encoded();
-        let ends_bp = BitPacked::encode(ends_inner, 8).unwrap();
+        let ends_bp = BitPacked::encode(ends_inner, 8, &mut ctx).unwrap();
         let compressed_ends =
             FoR::try_new(ends_bp.into_array(), ends_for.reference_scalar().clone())
                 .unwrap()
@@ -207,7 +209,7 @@ mod setup {
             .clone()
             .execute::<PrimitiveArray>(&mut ctx)
             .unwrap();
-        let compressed_values = BitPacked::encode(&values_prim.into_array(), 8)
+        let compressed_values = BitPacked::encode(&values_prim.into_array(), 8, &mut ctx)
             .unwrap()
             .into_array();
 
@@ -289,7 +291,7 @@ mod setup {
             .clone()
             .execute::<PrimitiveArray>(&mut ctx)
             .unwrap();
-        let offsets_bp = BitPacked::encode(&offsets_prim.into_array(), 20).unwrap();
+        let offsets_bp = BitPacked::encode(&offsets_prim.into_array(), 20, &mut ctx).unwrap();
 
         // Rebuild VarBin with compressed offsets
         let compressed_codes = VarBinArray::try_new(
@@ -350,7 +352,7 @@ mod setup {
             .unwrap();
         let days_for = FoR::encode(days_prim).unwrap();
         let days_inner = days_for.encoded();
-        let days_bp = BitPacked::encode(days_inner, 16).unwrap();
+        let days_bp = BitPacked::encode(days_inner, 16, &mut ctx).unwrap();
         let compressed_days =
             FoR::try_new(days_bp.into_array(), days_for.reference_scalar().clone())
                 .unwrap()
@@ -364,7 +366,7 @@ mod setup {
             .unwrap();
         let seconds_for = FoR::encode(seconds_prim).unwrap();
         let seconds_inner = seconds_for.encoded();
-        let seconds_bp = BitPacked::encode(seconds_inner, 17).unwrap();
+        let seconds_bp = BitPacked::encode(seconds_inner, 17, &mut ctx).unwrap();
         let compressed_seconds = FoR::try_new(
             seconds_bp.into_array(),
             seconds_for.reference_scalar().clone(),
@@ -380,7 +382,7 @@ mod setup {
             .unwrap();
         let subseconds_for = FoR::encode(subseconds_prim).unwrap();
         let subseconds_inner = subseconds_for.encoded();
-        let subseconds_bp = BitPacked::encode(subseconds_inner, 20).unwrap();
+        let subseconds_bp = BitPacked::encode(subseconds_inner, 20, &mut ctx).unwrap();
         let compressed_subseconds = FoR::try_new(
             subseconds_bp.into_array(),
             subseconds_for.reference_scalar().clone(),