add some options of parallelism, for testing

Project.toml

@@ -4,10 +4,12 @@ authors = ["Leandro Martinez <[email protected]> and contributors"]
 version = "0.9.6-DEV"
 
 [deps]
+ChunkSplitters = "ae650224-84b6-46f8-82ea-d812ca08434e"
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Parameters = "d96e819e-fc66-5662-9728-84c9c7592b0a"
+Polyester = "f517fe37-dbe3-4b94-8317-1923a5111588"
 PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 ProgressMeter = "92933f4c-e287-5a05-a399-4b506db050ca"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

src/CellListMap.jl

@@ -10,6 +10,8 @@ using Setfield: @set!
 using LinearAlgebra: cross, diagm, I
 using Base.Threads: nthreads, @spawn 
 using Base: @lock # not exported in 1.6
+using Polyester: @batch
+using ChunkSplitters: chunks
 
 export Box
 export CellList, UpdateCellList!

src/CoreComputing.jl

@@ -187,8 +187,20 @@ function map_pairwise_parallel!(
         output_threaded = [deepcopy(output) for i in 1:nbatches]
     end
     p = show_progress ? Progress(length(cl.ref), dt=1) : nothing
-    @sync for ibatch in 1:nbatches
-        @spawn batch($f, $ibatch, $nbatches, $output_threaded, $box, $cl, $p)
+    #@sync for ibatch in 1:nbatches
+    #    @spawn batch($f, $ibatch, $nbatches, $output_threaded, $box, $cl, $p)
+    #end
+    #Threads.@threads :static for ibatch in 1:nbatches
+    #    batch(f, ibatch, nbatches, output_threaded, box, cl, p)
+    #end
+    #@batch for i in eachindex(cl.ref)
+    #    inner_loop!(f, output_threaded[1], i, box, cl)
+    #    #_next!(p)
+    #end
+    @sync for (ibatch, inds) in enumerate(chunks(cl.ref; n=nbatches))
+        @spawn for i in inds
+            inner_loop!(f, output_threaded[ibatch], i, box, cl)
+        end
     end
     return reduce(output, output_threaded)
 end