Add copy to and from functions

trixi-gpu · Aug 8, 2024 · 453c845 · 453c845
1 parent c228449
commit 453c845
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/solvers/common.jl b/src/solvers/common.jl
@@ -1,4 +1,5 @@
 # Define common functions for all solvers
+# fixme: Dispatch different functions for Flaot64 and Float32 to improve performance, maybe exists a better way for Float32
 
 # Copy matrices from host to device (Float64)
 function copy_to_device!(du::PtrArray{Float64}, u::PtrArray{Float64})
@@ -9,9 +10,10 @@ function copy_to_device!(du::PtrArray{Float64}, u::PtrArray{Float64})
 end
 
 # Copy matrices from device to host (Float64)
-function copy_to_host!(du::PtrArray{Float64}, u::PtrArray{Float64})
-    du = Array(du)
-    u = Array(u)
+function copy_to_host!(du::CuArray{Float64}, u::CuArray{Float64})
+    # fixme: maybe direct PtrArray to CuArray conversion is possible
+    du = PtrArray(Array(du))
+    u = PtrArray(Array(u))
 
     return (du, u)
 end
diff --git a/src/solvers/solvers.jl b/src/solvers/solvers.jl
@@ -0,0 +1 @@
+include("common.jl")