Skip to content

Commit 0db1ef9

Browse files
committed
initiated experimental branch
1 parent 6d2755e commit 0db1ef9

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/ParallelKMeans.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,16 @@ TODO 2: Document function
2222
function pl_pairwise!(target, x, y, nth = Threads.nthreads())
2323
ncol = size(x, 2)
2424
nrow = size(x, 1)
25+
2526
ranges = divider(nrow, nth)
2627
waiting_list = Task[]
28+
2729
for i in 1:length(ranges) - 1
2830
push!(waiting_list, @spawn inner_pairwise!(target, x, y, ranges[i]))
2931
end
32+
3033
inner_pairwise!(target, x, y, ranges[end])
34+
3135
for i in 1:length(ranges) - 1
3236
wait(waiting_list[i])
3337
end
@@ -41,6 +45,7 @@ TODO 3: Document function
4145
"""
4246
function inner_pairwise!(target, x, y, r)
4347
ncol = size(x, 2)
48+
4449
@inbounds for k in axes(y, 1)
4550
for i in r
4651
target[i, k] = (x[i, 1] - y[k, 1])^2
@@ -61,6 +66,7 @@ TODO 4: Document function
6166
"""
6267
function pairwise!(target, x, y)
6368
ncol = size(x, 2)
69+
6470
@inbounds for k in axes(y, 1)
6571
for i in axes(x, 1)
6672
target[i, k] = (x[i, 1] - y[k, 1])^2

0 commit comments

Comments
 (0)