Skip to content

Commit ce80713

Browse files
committed
forgot about capabilities...
1 parent 9624103 commit ce80713

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/fasteval.jl

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -262,6 +262,16 @@ function LogDensityProblems.logdensity_and_gradient(
262262
)
263263
end
264264

265+
function LogDensityProblems.capabilities(
266+
::Type{<:DynamicPPL.Experimental.FastLDF{M,Nothing}}
267+
) where {M}
268+
return LogDensityProblems.LogDensityOrder{0}()
269+
end
270+
function LogDensityProblems.capabilities(
271+
::Type{<:DynamicPPL.Experimental.FastLDF{M,<:ADTypes.AbstractADType}}
272+
) where {M}
273+
return LogDensityProblems.LogDensityOrder{1}()
274+
end
265275
function LogDensityProblems.dimension(fldf::FastLDF)
266276
return fldf._dim
267277
end

0 commit comments

Comments
 (0)