Skip to content

Commit 5fe59ae

Browse files
committed
from_rrule -> from_chainrules
1 parent 369183c commit 5fe59ae

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ext/BijectorsMooncakeExt.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@ using Mooncake:
44
@is_primitive, MinimalCtx, Mooncake, CoDual, primal, tangent_type, @from_rrule
55
using Bijectors: find_alpha, ChainRulesCore
66

7-
@from_rrule(MinimalCtx, Tuple{typeof(find_alpha),Float16,Float16,Float16})
8-
@from_rrule(MinimalCtx, Tuple{typeof(find_alpha),Float32,Float32,Float32})
9-
@from_rrule(MinimalCtx, Tuple{typeof(find_alpha),Float64,Float64,Float64})
7+
@from_chainrules(MinimalCtx, Tuple{typeof(find_alpha),Float16,Float16,Float16})
8+
@from_chainrules(MinimalCtx, Tuple{typeof(find_alpha),Float32,Float32,Float32})
9+
@from_chainrules(MinimalCtx, Tuple{typeof(find_alpha),Float64,Float64,Float64})
1010

1111
# The final argument could be an Integer of some kind. This should be fine provided that
1212
# it has tangent type equal to `NoTangent`, which means that it's non-differentiable and

0 commit comments

Comments
 (0)