Merge pull request #107 from ygtangg/main

swapped thread dimension
JuliaORNL · Jul 9, 2024 · 256aec0 · 256aec0
2 parents 3ea080c + 04ee85d
commit 256aec0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/JACCAMDGPU/JACCAMDGPU.jl b/ext/JACCAMDGPU/JACCAMDGPU.jl
@@ -34,8 +34,8 @@ function JACC.parallel_for(
         I <: Integer, F <: Function}
     numThreads = 32
     Lthreads = min(L, numThreads)
-    Mthreads = min(M, numThreads)
-    Nthreads = 1
+    Mthreads = 1
+    Nthreads = min(M, numThreads)
     Lblocks = ceil(Int, L / Lthreads)
     Mblocks = ceil(Int, M / Mthreads)
     Nblocks = ceil(Int, N / Nthreads)