Add threading to search

dillondaudert · Jan 24, 2019 · 45d93d7 · 45d93d7
1 parent 0a9bd2d
commit 45d93d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nn_descent.jl b/src/nn_descent.jl
@@ -125,7 +125,7 @@ function search(graph::DescentGraph,
     max_candidates = trunc(Int, n_neighbors*queue_size)
     Dtype = result_type(graph.metric, queries[1], queries[1])
     candidates = [BinaryMaxHeap{NNTuple{Int, Dtype}}() for _ in 1:length(queries)]
-    for i in eachindex(queries)
+    Threads.@threads for i in eachindex(queries)
         # init
         seen = fill(false, length(graph.data))
         j =  rand(1:length(graph.data))