namespace

Author: ChrisRackauckas

ext/RecursiveArrayToolsZygoteExt.jl

@@ -13,7 +13,7 @@ end
 # Define a new species of projection operator for this type:
 ChainRulesCore.ProjectTo(x::VectorOfArray) = ChainRulesCore.ProjectTo{VectorOfArray}()
 
-function ChainRulesCore.rrule(T::Type{<:GPUArraysCore.AbstractGPUArray},
+function ChainRulesCore.rrule(T::Type{<:RecursiveArrayTools.GPUArraysCore.AbstractGPUArray},
                               xs::AbstractVectorOfArray)
     T(xs), ȳ -> (NoTangent(), ȳ)
 end