Pass Nq and Nv throuh Val in copyto stencil

CliMA · Jun 24, 2024 · 7f6b373 · 7f6b373
1 parent 93194ba
commit 7f6b373
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/ext/cuda/operators_finite_difference.jl b/ext/cuda/operators_finite_difference.jl
@@ -36,9 +36,9 @@ function Base.copyto!(
         strip_space(bc, space),
         axes(out),
         bounds,
-        Nq,
+        Val(Nv),
+        Val(Nq),
         Nh,
-        Nv,
     )
     auto_launch!(
         copyto_stencil_kernel!,
@@ -50,7 +50,15 @@ function Base.copyto!(
     return out
 end
 
-function copyto_stencil_kernel!(out, bc, space, bds, Nq, Nh, Nv)
+function copyto_stencil_kernel!(
+    out,
+    bc,
+    space,
+    bds,
+    Nh,
+    ::Val{Nq},
+    ::Val{Nv},
+) where {Nv, Nq}
     @inbounds begin
         gid = threadIdx().x + (blockIdx().x - 1) * blockDim().x
         if gid ≤ Nv * Nq * Nq * Nh