fix Stokes3D

PTsolvers · Feb 19, 2024 · 6cf180a · 6cf180a
1 parent acd3a10
commit 6cf180a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/src/stokes/Stokes2D.jl b/src/stokes/Stokes2D.jl
@@ -515,7 +515,6 @@ function JustRelax.solve!(
 
     # solver loop
     @copy stokes.P0 stokes.P
-    @copy stokes.P0 stokes.P
     wtime0 = 0.0
     θ = @zeros(ni...)
     λ = @zeros(ni...)

diff --git a/src/stokes/Stokes3D.jl b/src/stokes/Stokes3D.jl
@@ -22,6 +22,7 @@ import JustRelax: mean_mpi, norm_mpi, minimum_mpi, maximum_mpi, backend
 
 include("../rheology/GeoParams.jl")
 include("StressRotation.jl")
+include("StressKernels.jl")
 include("PressureKernels.jl")
 include("VelocityKernels.jl")
 
@@ -460,6 +461,8 @@ function JustRelax.solve!(
                 stokes.τ.II,
                 @tensor_center(stokes.τ_o),
                 @strain(stokes),
+                @tensor_center(stokes.ε_pl),
+                stokes.EII_pl,
                 stokes.P,
                 θ,
                 η,

diff --git a/src/stokes/StressKernels.jl b/src/stokes/StressKernels.jl
@@ -304,7 +304,7 @@ end
     τ_old,  # @ centers
     ε,      # @ vertices
     ε_pl,   # @ centers
-    EII,   # accumulated plastic strain rate @ centers
+    EII,    # accumulated plastic strain rate @ centers
     P,
     θ,
     η,