Skip to content

Commit 1ef4d96

Browse files
committed
disable forward mode testing
1 parent a0df9b7 commit 1ef4d96

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

ext/BijectorsMooncakeExt.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ using Bijectors: find_alpha, ChainRulesCore
1515
# unusual Integer type is encountered.
1616
@is_primitive(MinimalCtx, Tuple{typeof(find_alpha),P,P,Integer} where {P<:Base.IEEEFloat})
1717

18+
# TODO: This needs a corresponding frule!! as well for it to work on forward-mode Mooncake.
1819
function Mooncake.rrule!!(
1920
::CoDual{typeof(find_alpha)}, x::CoDual{P}, y::CoDual{P}, z::CoDual{I}
2021
) where {P<:Base.IEEEFloat,I<:Integer}

test/ad/chainrules.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,8 @@ end
3131

3232
if @isdefined Mooncake
3333
rng = Xoshiro(123456)
34-
@testset "$mode" for mode in (Mooncake.ForwardMode, Mooncake.ReverseMode)
34+
# TODO: Enable Mooncake.ForwardMode as well.
35+
@testset "$mode" for mode in (Mooncake.ReverseMode,)
3536
Mooncake.TestUtils.test_rule(
3637
rng,
3738
Bijectors.find_alpha,

0 commit comments

Comments
 (0)