fix displacement conversion on CUDA

PTsolvers · Jul 5, 2024 · f0e1aeb · f0e1aeb
1 parent f29651e
commit f0e1aeb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/ext/CUDA/2D.jl b/src/ext/CUDA/2D.jl
@@ -199,10 +199,18 @@ function JR2D.velocity2displacement!(::CUDABackendTrait, stokes::JustRelax.Stoke
     return _velocity2displacement!(stokes, dt)
 end
 
+function velocity2displacement!(::CUDABackendTrait, stokes::JustRelax.StokesArrays, dt)
+    return _velocity2displacement!(stokes, dt)
+end
+
 function JR2D.displacement2velocity!(::CUDABackendTrait, stokes::JustRelax.StokesArrays, dt)
     return _displacement2velocity!(stokes, dt)
 end
 
+function displacement2velocity!(::CUDABackendTrait, stokes::JustRelax.StokesArrays, dt)
+    return _displacement2velocity!(stokes, dt)
+end
+
 # Solvers
 function JR2D.solve!(::CUDABackendTrait, stokes, args...; kwargs)
     return _solve!(stokes, args...; kwargs...)