added setindex! for AD primitives (#59)

ThummeTo · Nov 24, 2023 · 4e53a05 · 4e53a05 · ThummeTo · Nov 24, 2023
1 parent 5dec4c1
commit 4e53a05
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,15 +1,15 @@
 name = "FMICore"
 uuid = "8af89139-c281-408e-bce2-3005eb87462f"
 authors = ["TT <tobias.thummerer@informatik.uni-augsburg.de>", "LM <lars.mikelsons@informatik.uni-augsburg.de>", "JK <josef.kircher@student.uni-augsburg.de>"]
-version = "0.19.1"
+version = "0.19.2"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
 Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 
 [compat]
-ChainRulesCore = "1.16.0"
+ChainRulesCore = "1.16.0 - 1.18"
 Dates = "1"
 Requires = "1.3.0"
 julia = "1.6"
diff --git a/src/extensions/FMISensitivity.jl b/src/extensions/FMISensitivity.jl
@@ -54,3 +54,12 @@ function unsense(e::ForwardDiff.Dual)
     return ForwardDiff.value(e)
 end
 
+# set sensitive primitives (this is intentionally NO additional dispatch for `setindex!`) 
+function sense_setindex!(A::Vector{Float64}, x::ForwardDiff.Dual, i::Int64)
+    return setindex!(A, undual(x), i)
+end
+
+function sense_setindex!(A::Vector{Float64}, x::ReverseDiff.TrackedReal, i::Int64)
+    return setindex!(A, untrack(x), i)
+end
+
diff --git a/src/sense.jl b/src/sense.jl
@@ -82,3 +82,8 @@ end
 function unsense(e)
     return e
 end
+
+# set sensitive primitives (this is intentionally NO additional dispatch for `setindex!`) 
+function sense_setindex!(A, x, i)
+    return setindex!(A, x, i)
+end