Update 3D.jl

src/ext/AMDGPU/3D.jl

@@ -92,14 +92,14 @@ function thermal_bcs!(::CUDABackendTrait, thermal::JustRelax.ThermalArrays, bcs)
 end
 
 # Phases
-function JR3D.phase_ratios_center(
+function JR3D.phase_ratios_center!(
     ::CUDABackendTrait,
     phase_ratios::JustRelax.PhaseRatio,
     particles,
     grid::Geometry,
     phases,
 )
-    return _phase_ratios_center(phase_ratios, particles, grid, phases)
+    return _phase_ratios_center!(phase_ratios, particles, grid, phases)
 end
 
 # Rheology
@@ -183,4 +183,40 @@ function JR3D.compute_dt(::CUDABackendTrait, S::JustRelax.StokesArrays, args...)
     return _compute_dt(S, args...)
 end
 
+function JR3D.subgrid_characteristic_time!(
+    subgrid_arrays,
+    particles,
+    dt₀::RocArray,
+    phases::JustRelax.PhaseRatio,
+    rheology,
+    thermal::JustRelax.ThermalArrays,
+    stokes::JustRelax.StokesArrays,
+    xci,
+    di,
+)
+    ni = size(stokes.P)
+    @parallel (@idx ni) subgrid_characteristic_time!(
+        dt₀, phases.center, rheology, thermal.Tc, stokes.P, di
+    )
+    return nothing
+end
+
+function JR3D.subgrid_characteristic_time!(
+    subgrid_arrays,
+    particles,
+    dt₀::RocArray,
+    phases::AbstractArray{Int,N},
+    rheology,
+    thermal::JustRelax.ThermalArrays,
+    stokes::JustRelax.StokesArrays,
+    xci,
+    di,
+) where {N}
+    ni = size(stokes.P)
+    @parallel (@idx ni) subgrid_characteristic_time!(
+        dt₀, phases, rheology, thermal.Tc, stokes.P, di
+    )
+    return nothing
+end
+
 end