Skip to content

Commit

Permalink
feat: build ivf partition using disk based shuffler (#1312)
Browse files Browse the repository at this point in the history
* Reduce the memory consumption to run IVF_PQ
* Building block for distributed IVF_PQ indexing
* Better multi-thread support during IVF assignment
  • Loading branch information
eddyxu authored Oct 2, 2023
1 parent 499b590 commit fbe5798
Show file tree
Hide file tree
Showing 10 changed files with 411 additions and 232 deletions.
2 changes: 2 additions & 0 deletions benchmarks/wiki/index.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,12 +13,14 @@ def main():
help="The metric type (L2, cosine, dot) to use for indexing",
)
parser.add_argument(
"-p",
"--num-partitions",
default=2048,
type=int,
help="The number of partitions to use for indexing",
)
parser.add_argument(
"-s",
"--num-sub-vectors",
default=96,
type=int,
Expand Down
1 change: 0 additions & 1 deletion rust/lance/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,6 @@ url = "2.3"
rand = { version = "0.8.3", features = ["small_rng"] }
futures = { workspace = true }
uuid = { version = "1.2", features = ["v4"] }
path-absolutize = "3.0.14"
shellexpand = "3.0.0"
arrow = { version = "43.0.0", features = ["prettyprint"] }
num_cpus = "1.0"
Expand Down
Loading

0 comments on commit fbe5798

Please sign in to comment.