diff --git a/ext/JACCAMDGPU/JACCAMDGPU.jl b/ext/JACCAMDGPU/JACCAMDGPU.jl index e344702..69de6b3 100644 --- a/ext/JACCAMDGPU/JACCAMDGPU.jl +++ b/ext/JACCAMDGPU/JACCAMDGPU.jl @@ -33,8 +33,8 @@ function JACC.parallel_for( (L, M, N)::Tuple{I, I, I}, f::F, x...) where { I <: Integer, F <: Function} numThreads = 32 - Lthreads = min(L, numThreads) - Mthreads = 1 + Lthreads = 1 + Mthreads = min(L, numThreads) Nthreads = min(M, numThreads) Lblocks = ceil(Int, L / Lthreads) Mblocks = ceil(Int, M / Mthreads)