From 5961820125e90cd451c53303bcd33ad84d34e2b5 Mon Sep 17 00:00:00 2001 From: Ian Butterworth Date: Thu, 13 Mar 2025 12:08:25 -0400 Subject: [PATCH] use the correct `nthreads` --- src/dim_helpers/ConvDims.jl | 2 +- src/gemm.jl | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/dim_helpers/ConvDims.jl b/src/dim_helpers/ConvDims.jl index 9358a41a8..e8bcc08f4 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 9cbe50195..79f6f3302 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).