Skip to content

Commit

Permalink
missing interpolation method
Browse files Browse the repository at this point in the history
  • Loading branch information
albert-de-montserrat committed May 15, 2024
1 parent bedd831 commit 306477a
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 0 deletions.
5 changes: 5 additions & 0 deletions src/ext/CUDA/2D.jl
Original file line number Diff line number Diff line change
Expand Up @@ -168,6 +168,11 @@ function JR2D.center2vertex!(
return center2vertex!(vertex_yz, vertex_xz, vertex_xy, center_yz, center_xz, center_xy)
end

function JR2D.velocity2vertex!(Vx_v::CuArray, Vy_v::CuArray, Vx::CuArray, Vy::CuArray; ghost_nodes=true)
velocity2vertex!(Vx_v, Vy_v, Vx, Vy; ghost_nodes=ghost_nodes)
return nothing
end

# Solvers
function JR2D.solve!(::CUDABackendTrait, stokes, args...; kwargs)
return _solve!(stokes, args...; kwargs...)
Expand Down
5 changes: 5 additions & 0 deletions src/ext/CUDA/3D.jl
Original file line number Diff line number Diff line change
Expand Up @@ -170,6 +170,11 @@ function JR3D.center2vertex!(
return center2vertex!(vertex_yz, vertex_xz, vertex_xy, center_yz, center_xz, center_xy)
end

function JR3D.velocity2vertex!(Vx_v::CuArray, Vy_v::CuArray, Vz_v::CuArray, Vx::CuArray, Vy::CuArray, Vz::CuArray)
velocity2vertex!(Vx_v, Vy_v, Vz_v, Vx, Vy, Vz)
return nothing
end

# Solvers
function JR3D.solve!(::CUDABackendTrait, stokes, args...; kwargs)
return _solve!(stokes, args...; kwargs...)
Expand Down

0 comments on commit 306477a

Please sign in to comment.