Add guess activity of ref of int

EnzymeAD · Sep 18, 2023 · e3e40e7 · e3e40e7
1 parent a66ede4
commit e3e40e7
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -388,6 +388,17 @@ end
     return state == ActiveState
 end
 
+@inline function Enzyme.guess_activity(::Type{T}, Mode::API.CDerivativeMode) where {T<:AbstractArray{<:Integer}}
+    return Const{T}
+end
+
+@inline function Enzyme.guess_activity(::Type{T}, Mode::API.CDerivativeMode) where {T<:Base.Ref{<:Integer}}
+    return Const{T}
+end
+@inline function Enzyme.guess_activity(::Type{T}, Mode::API.CDerivativeMode) where {T<:Base.RefValue{<:Integer}}
+    return Const{T}
+end
+
 @inline function Enzyme.guess_activity(::Type{T}, Mode::API.CDerivativeMode) where {T<:AbstractArray}
     if Mode == API.DEM_ForwardMode
         return DuplicatedNoNeed{T}