Improve perf of _set_continuous! & _set_discrete! (#25)

PTsolvers · Jun 18, 2024 · 8bd81f7 · 8bd81f7
1 parent 9ba161f
commit 8bd81f7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Chmy"
 uuid = "33a72cf0-4690-46d7-b987-06506c2248b9"
 authors = ["Ivan Utkin <[email protected]>, Ludovic Raess <[email protected]>, and contributors"]
-version = "0.1.15"
+version = "0.1.16"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

diff --git a/src/Fields/field.jl b/src/Fields/field.jl
@@ -73,12 +73,12 @@ end
     dst[I...] = src[I...]
 end
 
-@kernel inbounds = true function _set_continuous!(dst, grid, loc, fun::F, args...) where {F}
+@kernel inbounds = true function _set_continuous!(dst, grid, loc, fun::F, args::Vararg{Any, N}) where {F, N}
     I = @index(Global, NTuple)
     dst[I...] = fun(coord(grid, loc, I...)..., args...)
 end
 
-@kernel inbounds = true function _set_discrete!(dst, grid, loc, fun::F, args...) where {F}
+@kernel inbounds = true function _set_discrete!(dst, grid, loc, fun::F, args::Vararg{Any, N}) where {F, N}
     I = @index(Global, NTuple)
     dst[I...] = fun(grid, loc, I..., args...)
 end