Remove custom adjoint

Author: devmotion

src/basekernels/matern.jl

@@ -17,7 +17,7 @@ end
 
 @inline function kappa(κ::MaternKernel, d::Real)
     ν = first(κ.ν)
-    iszero(d) ? one(d) : _matern(ν, d)
+    return ifelse(iszero(d), one(d), _matern(ν, d))
 end
 
 function _matern(ν::Real, d::Real)

src/zygote_adjoints.jl

@@ -59,16 +59,6 @@ end
   end
 end
 
-@adjoint function kappa(κ::MaternKernel, d::Real)
-    ν = first(κ.ν)
-    val, grad = pullback(_matern, ν, d)
-    return ((iszero(d) ? one(d) : val),
-    Δ -> begin
-        ∇ = grad(Δ)
-        return ((ν = [∇[1]],), iszero(d) ? zero(d) : ∇[2])
-    end)
-end
-
 @adjoint function ColVecs(X::AbstractMatrix)
     back(Δ::NamedTuple) = (Δ.X,)
     back(Δ::AbstractMatrix) = (Δ,)