Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions vortex-tensor/src/encodings/turboquant/array/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,5 +9,3 @@ pub(crate) mod slots;

pub(crate) mod centroids;
pub(crate) mod rotation;

pub(crate) mod scheme;
11 changes: 5 additions & 6 deletions vortex-tensor/src/encodings/turboquant/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -108,21 +108,20 @@
mod array;
pub use array::data::TurboQuantArrayExt;
pub use array::data::TurboQuantData;
pub use array::scheme::TurboQuantScheme;

pub(crate) mod compute;

mod metadata;

mod vtable;

pub use vtable::TurboQuant;
pub use vtable::TurboQuantArray;

mod compress;
pub use compress::TurboQuantConfig;
pub use compress::turboquant_encode;

mod decompress;
mod scheme;
pub use scheme::TurboQuantScheme;
pub use scheme::compress::TurboQuantConfig;
pub use scheme::compress::turboquant_encode;

#[cfg(test)]
mod tests;
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: Copyright the Vortex contributors

//! TurboQuant compression scheme for the pluggable compressor.
//! TurboQuant compression scheme and decompression.

use vortex_array::ArrayRef;
use vortex_array::Canonical;
Expand All @@ -18,6 +18,9 @@ use crate::encodings::turboquant::TurboQuant;
use crate::encodings::turboquant::TurboQuantConfig;
use crate::encodings::turboquant::turboquant_encode;

pub(super) mod compress;
pub(super) mod decompress;

/// TurboQuant compression scheme for [`Vector`] extension types.
///
/// Applies lossy vector quantization to [`Vector`] extension arrays using the TurboQuant
Expand Down
2 changes: 1 addition & 1 deletion vortex-tensor/src/encodings/turboquant/vtable.rs
Original file line number Diff line number Diff line change
Expand Up @@ -38,8 +38,8 @@ use crate::encodings::turboquant::TurboQuantData;
use crate::encodings::turboquant::array::slots::Slot;
use crate::encodings::turboquant::compute::rules::PARENT_KERNELS;
use crate::encodings::turboquant::compute::rules::RULES;
use crate::encodings::turboquant::decompress::execute_decompress;
use crate::encodings::turboquant::metadata::TurboQuantMetadata;
use crate::encodings::turboquant::scheme::decompress::execute_decompress;
use crate::vector::AnyVector;
use crate::vector::VectorMatcherMetadata;

Expand Down
Loading