diff --git a/src/dim_helpers/ConvDims.jl b/src/dim_helpers/ConvDims.jl index 9358a41a..e8bcc08f 100644 --- a/src/dim_helpers/ConvDims.jl +++ b/src/dim_helpers/ConvDims.jl @@ -73,7 +73,7 @@ function im2col_dims(c::ConvDims) # Size of single dotproduct within convolution prod(kernel_size(c))*channels_in(c), # One workspace per thread - VERSION > v"1.9.0-0" ? Threads.maxthreadid() : Threads.nthreads(), + VERSION > v"1.9.0-0" ? Threads.nthreads(:default) : Threads.nthreads(), ) end diff --git a/src/gemm.jl b/src/gemm.jl index 9cbe5019..79f6f330 100644 --- a/src/gemm.jl +++ b/src/gemm.jl @@ -95,7 +95,7 @@ for (gemm, elt) in gemm_datatype_mappings strC = Base.stride(C, 3) n_threads = min( - VERSION > v"1.9.0-0" ? Threads.maxthreadid() : Threads.nthreads(), + VERSION > v"1.9.0-0" ? Threads.nthreads(:default) : Threads.nthreads(), 1 + max(length(A), length(B)) รท 8000) # In some tests, size (20,20,20) is worth splitting between two threads, # as is size (32,32,8).