Optimize active only rev grad (#1911)

* Optimize active only rev grad * Update Project.toml * add makezero s/marray
EnzymeAD · Sep 28, 2024 · 259f161 · 259f161
1 parent 327558b
commit 259f161
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ext/EnzymeStaticArraysExt.jl b/ext/EnzymeStaticArraysExt.jl
@@ -23,4 +23,11 @@ end
     end
 end
 
+@inline function Enzyme.EnzymeCore.make_zero(x::FT)::FT where {FT<:SArray}
+    return Base.zero(x)
+end
+@inline function Enzyme.EnzymeCore.make_zero(x::FT)::FT where {FT<:MArray}
+    return Base.zero(x)
+end
+
 end