Merge pull request #201 from Alexander-Barth/master

fix allocate_threadpool when NNPACK_CPU_THREADS is set

Author: CarloLucibello

src/nnpack/NNPACK.jl

@@ -34,8 +34,8 @@ Allows NNPACK to intelligently choose which threadpool to use for getting the be
 performance.
 """
 function allocate_threadpool()
-    global NNPACK_CPU_THREADS = NNPACK_CPU_THREADS > 8 ? UInt64(8) : floor(log2(NNPACK_CPU_THREADS))
-    for i in 1:Int(NNPACK_CPU_THREADS)
+    global NNPACK_CPU_THREADS = NNPACK_CPU_THREADS > 8 ? UInt64(8) : UInt64(exp2(floor(log2(NNPACK_CPU_THREADS))))
+    for i in 0:Int(log2(NNPACK_CPU_THREADS))
         threads = UInt64(2^i)
         push!(shared_threadpool_dict, threads => Ref(pthreadpool_create(threads)))
     end