remove ccall & struct, fix test GPU

EnzymeAD · Dec 10, 2024 · fe1c81c · fe1c81c
1 parent ae1cd2e
commit fe1c81c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/src/Compiler.jl b/src/Compiler.jl
@@ -435,8 +435,7 @@ macro jit(args...)
 end
 
 is_a_module(s::Symbol)::Bool = begin
-    b = ccall(:jl_get_module_binding, Any, (Module, Symbol, Any), Main, s, 0)
-    isdefined(b, :value) && b.value isa Module
+    isdefined(@__MODULE__, s) && getproperty(@__MODULE__, s) isa Module
 end
 
 #create expression for more complex expression than a call
@@ -449,9 +448,7 @@ function wrapped_expression(expr::Expr)
     return (
         Expr(:tuple, args...),
         quote
-            struct $fname
-                ($fname)($(args...)) = $expr
-            end
+            ($fname)($(args...)) = $expr
         end,
         quote
             $(fname)

diff --git a/test/basic.jl b/test/basic.jl
@@ -105,7 +105,7 @@ end
     x = Reactant.to_rarray(ones(3))
     f(x) = x .+ 1
     @test @jit(x + x - x + x * float(Base.pi) * 0) ≈ x
-    @test @jit(f(f(f(f(x))))) ≈ x .+ 4
+    @test @jit(f(f(f(f(x))))) ≈ @allowscalar x .+ 4
 end
 
 function sumcos(x)