Expand description
GPU kernel provider for XLOG
Re-exports§
pub use arrow_device::ArrowDeviceArray;pub use arrow_device::ArrowDeviceArrayOwned;pub use arrow_device::ARROW_DEVICE_CUDA;pub use cuda_compat::AsKernelParam;pub use cuda_compat::CudaFunction;pub use cuda_compat::IntoKernelParamStorage;pub use cuda_compat::KernelParamStorage;pub use cuda_compat::KernelScalar;pub use cuda_compat::LaunchAsync;pub use device::CudaDevice;pub use device_pool::GpuDevicePool;pub use dlpack::DLManagedTensor;pub use dlpack::DlpackManagedTensor;pub use dlpack::DlpackTable;pub use memory::CudaBuffer;pub use memory::CudaColumn;pub use memory::GpuMemoryManager;pub use memory::RuntimeAllocBlock;pub use multi_gpu_memory::MultiGpuMemoryManager;pub use provider::circuit_kernels;pub use provider::dedup_kernels;pub use provider::filter_kernels;pub use provider::groupby_kernels;pub use provider::ilp_kernels;pub use provider::join_kernels;pub use provider::pack_kernels;pub use provider::pir_kernels;pub use provider::scan_kernels;pub use provider::set_ops_kernels;pub use provider::sort_kernels;pub use provider::CompareOp;pub use provider::CudaKernelProvider;pub use provider::FjDeltaCols;pub use provider::FjNode;pub use provider::FjPlan;pub use provider::FjSubAtom;pub use provider::JoinIndexV2;pub use provider::JoinType;pub use provider::CIRCUIT_MODULE;pub use provider::DEDUP_MODULE;pub use provider::FILTER_MODULE;pub use provider::GROUPBY_MODULE;pub use provider::ILP_MODULE;pub use provider::JOIN_MODULE;pub use provider::PACK_MODULE;pub use provider::PIR_MODULE;pub use provider::SCAN_MODULE;pub use provider::SET_OPS_MODULE;pub use provider::SORT_MODULE;pub use wcoj_metadata::HeatDist;pub use wcoj_metadata::LayoutSignature;pub use wcoj_metadata::RootMetadata;pub use wcoj_metadata::VertexId;pub use wcoj_metadata::WcojCycle4HgWorkPlanU32;pub use wcoj_metadata::WcojCycle4HgWorkPlanU64;pub use wcoj_metadata::WcojRelationMetadata;pub use wcoj_metadata::WcojTriangleHgWorkPlanU32;pub use wcoj_metadata::WcojTriangleHgWorkPlanU64;
Modules§
- arrow_
device - cuda_
compat - cuda_
graph - CUDA Graph RAII helpers for production graph capture/replay.
- device
- CUDA device management
- device_
pool - Multi-GPU device pool management
- device_
runtime - Stream-ordered device memory runtime, RMM-inspired.
- dlpack
- DLPack interop for zero-copy GPU exchange.
- kernel_
manifest_ data - launch
- Launch / use recorder for runtime-backed buffers.
- memory
- CUDA memory management
- multi_
gpu_ memory - Multi-GPU memory management
- provider
- CUDA kernel provider implementation
- sys
- type_
seam - GpuScalar — marker trait for Rust scalar types that round-trip through GPU column storage.
- wcoj_
metadata
Structs§
- Cuda
Slice Vec<T>on a cuda device. You can allocate and modify this with CudaStream.- Cuda
Stream - A wrapper around sys::CUstream that you can schedule work on.
- Cuda
View &[T]on a cuda device. An immutable sub-view into a CudaSlice created by CudaSlice::as_view()/CudaSlice::slice().- Cuda
View Mut &mut [T]on a cuda device. A mutable sub-view into a CudaSlice created by CudaSlice::as_view_mut()/CudaSlice::slice_mut().- Driver
Error - Wrapper around sys::CUresult. See nvidia’s CUresult docs
- Launch
Config - Configuration for [result::launch_kernel]
Traits§
- Device
Ptr - Abstraction over CudaSlice/CudaView
- Device
PtrMut - Abstraction over CudaSlice/CudaViewMut
- Device
Repr - Something that can be copied to device memory and turned into a parameter for [result::launch_kernel].
- Device
Slice - Base trait for abstracting over CudaSlice/CudaView/CudaViewMut.
- Valid
AsZero Bits - Marker trait to indicate that the type is valid when all of its bits are set to 0.