add a sensitivity example of Pine Island Glacier

DJ4Earth · May 29, 2024 · d140317 · d140317
1 parent 64401e6
commit d140317
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/data/PIG_Control_B_dJUICE.mat b/data/PIG_Control_B_dJUICE.mat
diff --git a/data/PIG_Control_drag_dJUICE.mat b/data/PIG_Control_drag_dJUICE.mat
diff --git a/examples/sensitivity_PIG.jl b/examples/sensitivity_PIG.jl
@@ -0,0 +1,26 @@
+using Enzyme
+Enzyme.API.typeWarning!(false)
+Enzyme.Compiler.RunAttributor[] = false
+
+using DJUICE
+using MAT
+
+#Load model from MATLAB file
+file = matopen(joinpath(@__DIR__, "..", "Data","PIG_Control_drag_dJUICE.mat"))
+
+mat  = read(file, "md")
+close(file)
+md = model(mat)
+
+#make model run faster 
+md.stressbalance.maxiter = 20
+
+#Now call AD!
+md.inversion.iscontrol = 1
+md.inversion.independent = md.friction.coefficient
+md.inversion.independent_string = "FrictionCoefficient"
+
+md = solve(md, :grad)
+
+# the gradient
+g = md.results["StressbalanceSolution"]["Gradient"]