Skip another segfault

TuringLang · Feb 16, 2025 · f42a6e1 · f42a6e1
1 parent 570b493
commit f42a6e1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/ad.jl b/test/ad.jl
@@ -40,8 +40,9 @@ using EnzymeCore: set_runtime_activity, Forward, Reverse, ReverseMode
                     is_1_11 = v"1.11" <= VERSION < v"1.12"
                     is_svi_vnv = varinfo isa SimpleVarInfo{<:DynamicPPL.VarNamedVector}
                     is_svi_od = varinfo isa SimpleVarInfo{<:OrderedDict}
-                    is_svi_od_ref = varinfo isa SimpleVarInfo{<:OrderedDict,<:Ref}
-                    is_svi_od_not_ref = is_svi_od && !is_svi_od_ref
+                    is_svi_ref = varinfo isa SimpleVarInfo{<:Any,<:Ref}
+                    is_svi_od_not_ref = is_svi_od && !is_svi_ref
+                    is_svi_vnv_not_ref = is_svi_vnv && !is_svi_ref
 
                     # Mooncake doesn't work with several combinations of SimpleVarInfo.
                     if is_mooncake && is_1_11 && is_svi_vnv
@@ -57,7 +58,7 @@ using EnzymeCore: set_runtime_activity, Forward, Reverse, ReverseMode
                         )
                     elseif string(m.f) == "demo_dot_assume_dot_observe" &&
                         is_enzyme_reverse &&
-                        is_svi_od_not_ref &&
+                        (is_svi_od_not_ref || is_svi_vnv_not_ref) &&
                         is_1_11
                         # TODO: report upstream, this segfaults
                         @test_broken false