Skip to content

Commit f42a6e1

Browse files
committed
Skip another segfault
1 parent 570b493 commit f42a6e1

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

test/ad.jl

+4-3
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,9 @@ using EnzymeCore: set_runtime_activity, Forward, Reverse, ReverseMode
4040
is_1_11 = v"1.11" <= VERSION < v"1.12"
4141
is_svi_vnv = varinfo isa SimpleVarInfo{<:DynamicPPL.VarNamedVector}
4242
is_svi_od = varinfo isa SimpleVarInfo{<:OrderedDict}
43-
is_svi_od_ref = varinfo isa SimpleVarInfo{<:OrderedDict,<:Ref}
44-
is_svi_od_not_ref = is_svi_od && !is_svi_od_ref
43+
is_svi_ref = varinfo isa SimpleVarInfo{<:Any,<:Ref}
44+
is_svi_od_not_ref = is_svi_od && !is_svi_ref
45+
is_svi_vnv_not_ref = is_svi_vnv && !is_svi_ref
4546

4647
# Mooncake doesn't work with several combinations of SimpleVarInfo.
4748
if is_mooncake && is_1_11 && is_svi_vnv
@@ -57,7 +58,7 @@ using EnzymeCore: set_runtime_activity, Forward, Reverse, ReverseMode
5758
)
5859
elseif string(m.f) == "demo_dot_assume_dot_observe" &&
5960
is_enzyme_reverse &&
60-
is_svi_od_not_ref &&
61+
(is_svi_od_not_ref || is_svi_vnv_not_ref) &&
6162
is_1_11
6263
# TODO: report upstream, this segfaults
6364
@test_broken false

0 commit comments

Comments
 (0)