Fix justActive condition (#1936)

Closes #1935
EnzymeAD · Oct 22, 2024 · 6c23ee7 · 6c23ee7
1 parent 274e5e5
commit 6c23ee7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -469,7 +469,7 @@ end
 
     subT = typed_fieldtype(T, f)
 
-    if justActive && !allocatedinline(subT)
+    if justActive && ismutabletype(subT)
         return Val(AnyState)
     end
 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -135,6 +135,17 @@ end
     @test Enzyme.Compiler.active_reg_inner(Tuple, (), nothing) == Enzyme.Compiler.DupState
     @test Enzyme.Compiler.active_reg_inner(Tuple, (), nothing, #=justactive=#Val(false), #=unionsret=#Val(false), #=abstractismixed=#Val(true)) == Enzyme.Compiler.MixedState
     @test Enzyme.Compiler.active_reg_inner(Tuple{A,A} where A, (), nothing, #=justactive=#Val(false), #=unionsret=#Val(false), #=abstractismixed=#Val(true)) == Enzyme.Compiler.MixedState
+
+    # issue #1935
+    struct Incomplete
+        x::Float64
+        y
+        Incomplete(x) = new(x)
+        # incomplete constructor & non-bitstype field => !Base.allocatedinline(Incomplete)
+    end
+    @test Enzyme.Compiler.active_reg_inner(Tuple{Incomplete}, (), nothing, #=justActive=#Val(false)) == Enzyme.Compiler.MixedState
+    @test Enzyme.Compiler.active_reg_inner(Tuple{Incomplete}, (), nothing, #=justActive=#Val(true)) == Enzyme.Compiler.ActiveState
+
     world = codegen_world_age(typeof(f0), Tuple{Float64})
     thunk_a = Enzyme.Compiler.thunk(Val(world), Const{typeof(f0)}, Active, Tuple{Active{Float64}}, Val(API.DEM_ReverseModeCombined), Val(1), Val((false, false)), Val(false), Val(false), DefaultABI, Val(false), Val(false))
     thunk_b = Enzyme.Compiler.thunk(Val(world), Const{typeof(f0)}, Const, Tuple{Const{Float64}}, Val(API.DEM_ReverseModeCombined), Val(1), Val((false, false)), Val(false), Val(false), DefaultABI, Val(false), Val(false))