Manually specify max_threads

CliMA · Jul 27, 2023 · 64f3bda · 64f3bda
1 parent 0446116
commit 64f3bda
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/Operators/finitedifference.jl b/src/Operators/finitedifference.jl
@@ -3448,9 +3448,9 @@ function Base.copyto!(
         Nq,
         Nh,
     )
-    kernel = @cuda launch = false copyto_stencil_kernel!(args...)
-    kernel_config = CUDA.launch_configuration(kernel.fun)
-    max_threads = kernel_config.threads
+    #kernel = @cuda launch = false copyto_stencil_kernel!(args...)
+    #kernel_config = CUDA.launch_configuration(kernel.fun)
+    max_threads = 256#kernel_config.threads
     nitems = nnodes * Nq * Nq * Nh
     nthreads = min(max_threads, nitems)
     nblocks = cld(nitems, nthreads)