ensore on device

Signed-off-by: Andrew Duffy <andrew@a10y.dev>

Author: a10y

vortex-cuda/src/kernel/encodings/bitpacked.rs

@@ -18,18 +18,15 @@ use vortex::array::match_each_integer_ptype;
 use vortex::dtype::NativePType;
 use vortex::encodings::fastlanes::BitPacked;
 use vortex::encodings::fastlanes::BitPackedArray;
-use vortex::encodings::fastlanes::BitPackedArrayExt;
 use vortex::encodings::fastlanes::BitPackedDataParts;
 use vortex::encodings::fastlanes::unpack_iter::BitPacked as BitPackedUnpack;
 use vortex::error::VortexResult;
 use vortex::error::vortex_ensure;
 use vortex::error::vortex_err;
-use vortex_array::patches::Patches;
 use vortex_error::vortex_bail;
 
 use crate::CudaBufferExt;
 use crate::CudaDeviceBuffer;
-use crate::executor::CudaArrayExt;
 use crate::executor::CudaExecute;
 use crate::executor::CudaExecutionCtx;
 use crate::kernel::patches::gpu::GPUPatches;

vortex-cuda/src/kernel/encodings/for_.rs

@@ -18,7 +18,6 @@ use vortex::array::match_each_integer_ptype;
 use vortex::array::match_each_native_simd_ptype;
 use vortex::dtype::NativePType;
 use vortex::encodings::fastlanes::BitPacked;
-use vortex::encodings::fastlanes::BitPackedArrayExt;
 use vortex::encodings::fastlanes::FoR;
 use vortex::encodings::fastlanes::FoRArray;
 use vortex::encodings::fastlanes::FoRArrayExt;

vortex-cuda/src/kernel/encodings/patched.rs

@@ -98,9 +98,9 @@ impl CudaExecute for PatchedExecutor {
                     bitpacked.into_owned(),
                     P::default(),
                     Some(DevicePatches {
-                        lane_offsets,
-                        indices: patch_indices,
-                        values: patch_values,
+                        lane_offsets: ctx.ensure_on_device(lane_offsets).await?,
+                        indices: ctx.ensure_on_device(patch_indices).await?,
+                        values: ctx.ensure_on_device(patch_values).await?,
                     }),
                     ctx,
                 )