deving local ClimaCore

CliMA · Jul 12, 2024 · 4683bc6 · 4683bc6
1 parent b3df3a2
commit 4683bc6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/shared_utilities/implicit_timestepping.jl b/src/shared_utilities/implicit_timestepping.jl
@@ -145,7 +145,7 @@ function ImplicitEquationJacobian(Y::ClimaCore.Fields.FieldVector)
     function get_j_field(
         space::Union{Spaces.PointSpace, Spaces.SpectralElementSpace2D},
     )
-        diag_type = (I,)
+        diag_type = MatrixFields.DiagonalMatrixRow{FT}
         diag_field = ClimaCore.Fields.Field(diag_type, space)
         fill!(parent(diag_field), 0)
         return diag_field

diff --git a/src/standalone/Vegetation/canopy_energy.jl b/src/standalone/Vegetation/canopy_energy.jl
@@ -161,7 +161,7 @@ function ClimaLand.make_update_jacobian(
 
         # The derivative of the residual with respect to the prognostic variable
         ∂Tres∂T = matrix[@name(canopy.energy.T), @name(canopy.energy.T)]
-        @. ∂Tres∂T = dtγ * 1 - (I,)
+        @. ∂Tres∂T = dtγ *  MatrixFields.DiagonalMatrixRow(Y.canopy.energy.T) - (I,)
     end
     return update_jacobian!
 end