Skip to main content

Crate xlog_cuda

Crate xlog_cuda 

Source
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§

CudaSlice
Vec<T> on a cuda device. You can allocate and modify this with CudaStream.
CudaStream
A wrapper around sys::CUstream that you can schedule work on.
CudaView
&[T] on a cuda device. An immutable sub-view into a CudaSlice created by CudaSlice::as_view()/CudaSlice::slice().
CudaViewMut
&mut [T] on a cuda device. A mutable sub-view into a CudaSlice created by CudaSlice::as_view_mut()/CudaSlice::slice_mut().
DriverError
Wrapper around sys::CUresult. See nvidia’s CUresult docs
LaunchConfig
Configuration for [result::launch_kernel]

Traits§

DevicePtr
Abstraction over CudaSlice/CudaView
DevicePtrMut
Abstraction over CudaSlice/CudaViewMut
DeviceRepr
Something that can be copied to device memory and turned into a parameter for [result::launch_kernel].
DeviceSlice
Base trait for abstracting over CudaSlice/CudaView/CudaViewMut.
ValidAsZeroBits
Marker trait to indicate that the type is valid when all of its bits are set to 0.