Skip to content

Commit 51d34d1

Browse files
authored
move scheme files around (#7342)
## Summary Just moves stuff around. ## Testing N/A Signed-off-by: Connor Tsui <connor.tsui20@gmail.com>
1 parent 5adc437 commit 51d34d1

6 files changed

Lines changed: 10 additions & 10 deletions

File tree

vortex-tensor/src/encodings/turboquant/array/mod.rs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,3 @@ pub(crate) mod slots;
99

1010
pub(crate) mod centroids;
1111
pub(crate) mod rotation;
12-
13-
pub(crate) mod scheme;

vortex-tensor/src/encodings/turboquant/mod.rs

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -108,21 +108,20 @@
108108
mod array;
109109
pub use array::data::TurboQuantArrayExt;
110110
pub use array::data::TurboQuantData;
111-
pub use array::scheme::TurboQuantScheme;
112111

113112
pub(crate) mod compute;
114113

115114
mod metadata;
116115

117116
mod vtable;
117+
118118
pub use vtable::TurboQuant;
119119
pub use vtable::TurboQuantArray;
120120

121-
mod compress;
122-
pub use compress::TurboQuantConfig;
123-
pub use compress::turboquant_encode;
124-
125-
mod decompress;
121+
mod scheme;
122+
pub use scheme::TurboQuantScheme;
123+
pub use scheme::compress::TurboQuantConfig;
124+
pub use scheme::compress::turboquant_encode;
126125

127126
#[cfg(test)]
128127
mod tests;

vortex-tensor/src/encodings/turboquant/compress.rs renamed to vortex-tensor/src/encodings/turboquant/scheme/compress.rs

File renamed without changes.

vortex-tensor/src/encodings/turboquant/decompress.rs renamed to vortex-tensor/src/encodings/turboquant/scheme/decompress.rs

File renamed without changes.

vortex-tensor/src/encodings/turboquant/array/scheme.rs renamed to vortex-tensor/src/encodings/turboquant/scheme/mod.rs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// SPDX-License-Identifier: Apache-2.0
22
// SPDX-FileCopyrightText: Copyright the Vortex contributors
33

4-
//! TurboQuant compression scheme for the pluggable compressor.
4+
//! TurboQuant compression scheme and decompression.
55
66
use vortex_array::ArrayRef;
77
use vortex_array::Canonical;
@@ -18,6 +18,9 @@ use crate::encodings::turboquant::TurboQuant;
1818
use crate::encodings::turboquant::TurboQuantConfig;
1919
use crate::encodings::turboquant::turboquant_encode;
2020

21+
pub(super) mod compress;
22+
pub(super) mod decompress;
23+
2124
/// TurboQuant compression scheme for [`Vector`] extension types.
2225
///
2326
/// Applies lossy vector quantization to [`Vector`] extension arrays using the TurboQuant

vortex-tensor/src/encodings/turboquant/vtable.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ use crate::encodings::turboquant::TurboQuantData;
3838
use crate::encodings::turboquant::array::slots::Slot;
3939
use crate::encodings::turboquant::compute::rules::PARENT_KERNELS;
4040
use crate::encodings::turboquant::compute::rules::RULES;
41-
use crate::encodings::turboquant::decompress::execute_decompress;
4241
use crate::encodings::turboquant::metadata::TurboQuantMetadata;
42+
use crate::encodings::turboquant::scheme::decompress::execute_decompress;
4343
use crate::vector::AnyVector;
4444
use crate::vector::VectorMatcherMetadata;
4545

0 commit comments

Comments
 (0)