Merge pull request #15 from simeonschaub/sds/fix_on_master

introduce literal_getfield
FluxML · Dec 15, 2020 · 5e48881 · 5e48881 · simeonschaub · Dec 15, 2020
2 parents 85bfd61 + c69a717
commit 5e48881
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/ZygoteRules.jl b/src/ZygoteRules.jl
@@ -8,7 +8,15 @@ export @adjoint, @adjoint!
 In Zygote, differentiation of property access is defined by defining adjoint of
 `ZygoteRules.literal_getproperty` rather than of `Base.getproperty`.
 """
-literal_getproperty(x, ::Val{f}, getproperty=getproperty) where f = getproperty(x, f)
+literal_getproperty(x, ::Val{f}) where f = getproperty(x, f)
+
+"""
+    ZygoteRules.literal_getfield(x, ::Val{f})
+
+In Zygote, differentiation of property access is defined by defining adjoint of
+`ZygoteRules.literal_getfield` rather than of `Base.getfield`.
+"""
+literal_getfield(x, ::Val{f}) where f = getfield(x, f)
 
 include("adjoint.jl")