Further absint improvements (#2140)

* Further absint improvements * Update Project.toml --------- Co-authored-by: William Moses <[email protected]>
EnzymeAD · Nov 28, 2024 · 6746e5a · 6746e5a
1 parent 30b6b2d
commit 6746e5a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/src/absint.jl b/src/absint.jl
@@ -576,12 +576,20 @@ function abs_typeof(
                     end
                 end
                 if !seen && typed_fieldcount(typ) > 0
-                    offset = offset - typed_fieldoffset(typ, lasti)
-                    typ = typed_fieldtype(typ, lasti)
-                    @assert Base.isconcretetype(typ)
-                    if !Base.allocatedinline(typ)
-                        legal = false
-                    end
+                    offset = offset - typed_fieldoffset(typ, lasti) 
+		    typ = typed_fieldtype(typ, lasti)
+		    if offset == 0
+                        if !Base.allocatedinline(typ)
+                            if byref != GPUCompiler.BITS_VALUE
+                                legal = false
+                            end
+                            byref = GPUCompiler.MUT_REF
+                        end
+		    else
+			    if !Base.isconcretetype(typ) || !Base.allocatedinline(typ)
+				legal = false
+			    end
+		    end
                     seen = true
                 end
                 if !seen

diff --git a/src/compiler.jl b/src/compiler.jl
@@ -6177,7 +6177,7 @@ function GPUCompiler.codegen(
 )
     params = job.config.params
     if params.run_enzyme
-        @assert eltype(params.rt) != Union{}
+        # @assert eltype(params.rt) != Union{}
     end
     expectedTapeType = params.expectedTapeType
     mode = params.mode