ChainRulesCore 1

Author: YingboMa

Project.toml

@@ -23,7 +23,7 @@ TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
 
 [compat]
 AbstractTrees = "0.3"
-ChainRulesCore = "0.9, 0.10, 1"
+ChainRulesCore = "1"
 Combinatorics = "1.0"
 ConstructionBase = "1.1"
 DataStructures = "0.18"

src/adjoints.jl

@@ -6,7 +6,7 @@ function rrule(::typeof(Code.create_array), A::Type{<:AbstractArray}, T, u::Val{
   y = Code.create_array(A, T, u, d, elems...)
   function create_array_pullback(Δ)
     dx = Δ
-    (NO_FIELDS, DoesNotExist(), DoesNotExist(), DoesNotExist(), DoesNotExist(), dx..., ntuple(_ -> DoesNotExist(), length(elems) - prod(dims) + j)...)
+    (ZeroTangent(), NoTangent(), NoTangent(), NoTangent(), NoTangent(), dx..., ntuple(_ -> NoTangent(), length(elems) - prod(dims) + j)...)
   end
   y, create_array_pullback
 end