assert check the world we encode in enzyme_extrace_world

Author: vchuravy

src/compiler.jl

@@ -1083,6 +1083,7 @@ function set_module_types!(interp, mod::LLVM.Module, primalf::Union{Nothing, LLV
         end
 
         world = enzyme_extract_world(f)
+        @assert world == interp.world
 
         if expectLen != length(parameters(f))
             continue
@@ -1665,6 +1666,7 @@ function shadow_alloc_rewrite(V::LLVM.API.LLVMValueRef, gutils::API.EnzymeGradie
        mode == API.DEM_ReverseModeCombined
         fn = LLVM.parent(LLVM.parent(V))
         world = enzyme_extract_world(fn)
+        @assert world == enzyme_gutils_context(gutils).world
         if !guaranteed_nonactive(Ty, world)
             B = LLVM.IRBuilder()
             position!(B, V)

src/errors.jl

@@ -475,6 +475,7 @@ function julia_error(
             ) #=error=#
             world = enzyme_extract_world(f)
         end
+        @assert world == enzyme_gutils_context(gutils).world
         throw(IllegalTypeAnalysisException(msg, mi, world, sval, ir, bt))
     elseif errtype == API.ET_NoType
         @assert B != C_NULL
@@ -550,6 +551,7 @@ function julia_error(
         illegal = false
         created = LLVM.Instruction[]
         world = enzyme_extract_world(LLVM.parent(position(IRBuilder(B))))
+        @assert world == enzyme_context(gutils).world
         width = get_width(gutils)
         function make_batched(@nospecialize(cur::LLVM.Value), B::LLVM.IRBuilder)::LLVM.Value
             if width == 1
@@ -944,7 +946,7 @@ end
             end
         end
 
-	mi = nothing
+        mi = nothing
         world = nothing
 
         if isa(val, LLVM.Instruction)
@@ -962,6 +964,7 @@ end
             ) #=error=#
             world = enzyme_extract_world(f)
         end
+        @assert world == enzyme_gutils_context(gutils).world
         mode = Enzyme.API.DEM_ReverseModeCombined
 
 	if mi !== nothing