Update to ThreadPinning v1

Author: oschulz

Project.toml

@@ -27,5 +27,5 @@ Logging = "1"
 Parameters = "0.12, 0.13"
 Pkg = "1"
 Sockets = "1"
-ThreadPinning = "0.7.22, 1"
+ThreadPinning = "1"
 julia = "1.10"

ext/ParallelProcessingToolsThreadPinningExt.jl

@@ -5,12 +5,18 @@ import LinearAlgebra
 import Distributed
 import ThreadPinning
 
-# ThreadPinning.jl does not support all operating systems, currently:
-const _threadpinning_supported = isdefined(ThreadPinning, :affinitymask2cpuids)
+
+# ThreadPinning.jl does not fully support all operating systems, currently:
+const _threadpinning_supported = try
+    @assert convert(Int, ThreadPinning.getcpuid()) isa Int
+    true
+catch err
+    false
+end
+
 
 @static if _threadpinning_supported
 
-    
 function ParallelProcessingTools._pinthreads_auto_impl(::Val{true})
     pid = Distributed.myid()
     if Distributed.myid() == 1
@@ -22,7 +28,7 @@ function ParallelProcessingTools._pinthreads_auto_impl(::Val{true})
         end
     else
         @debug "On process $pid, pinning threads according to affinity mask"
-        let available_cpus = ThreadPinning.affinitymask2cpuids(ThreadPinning.get_affinity_mask())
+        let available_cpus = _ThreadPinning.Utility.affinitymask2cpuids(ThreadPinning.getaffinity())
             ThreadPinning.pinthreads(:affinitymask)
             LinearAlgebra.BLAS.set_num_threads(length(available_cpus))
         end
@@ -32,7 +38,6 @@ end
 
 ParallelProcessingTools._getcpuids_impl(::Val{true}) = ThreadPinning.getcpuids()
 
-
 end # if _threadpinning_supported
 
 end # module ChangesOfVariablesInverseFunctionsExt

test/runtests.jl

@@ -3,7 +3,10 @@
 import Test
 
 import ThreadPinning
-ThreadPinning.Prefs.set_os_warning(false)
+
+if isdefined(ThreadPinning, :Prefs)
+    ThreadPinning.Prefs.set_os_warning(false)
+end
 
 Test.@testset "Package ParallelProcessingTools" begin
     @info "Testing with $(Base.Threads.nthreads()) Julia threads."
@@ -21,6 +24,7 @@ Test.@testset "Package ParallelProcessingTools" begin
     include("test_procinit.jl")
     include("test_workerpool.jl")
     include("test_onworkers.jl")
+    include("test_ext_threadpinning.jl")
     include("test_deprecated.jl")
     include("test_docs.jl")
 end # testset

test/test_ext_threadpinning.jl

@@ -0,0 +1,13 @@
+# This file is a part of ParallelProcessingTools.jl, licensed under the MIT License (MIT).
+
+using ParallelProcessingTools
+using Test
+
+import ThreadPinning
+
+@testset "ext_threadpinning" begin
+    ParallelProcessingToolsThreadPinningExt = Base.get_extension(ParallelProcessingTools, :ParallelProcessingToolsThreadPinningExt)
+    @test ParallelProcessingToolsThreadPinningExt isa Module
+
+    @test pinthreads_auto() isa Nothing
+end