Skip to main content

kvbm_kernels/
lib.rs

1// SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2// SPDX-License-Identifier: Apache-2.0
3
4pub mod tensor_kernels;
5
6// Always available - core transfer functionality
7pub use tensor_kernels::{
8    MemcpyBatchMode, is_memcpy_batch_available, is_using_stubs, memcpy_batch, vectorized_copy,
9};
10
11// Permute kernels - data layout transformation (requires permute_kernels feature)
12#[cfg(feature = "permute_kernels")]
13pub use tensor_kernels::{BlockLayout, TensorDataType, block_from_universal, universal_from_block};