fix

EnzymeAD · Dec 15, 2024 · 8a0bfb0 · 8a0bfb0
1 parent 5a43ae8
commit 8a0bfb0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/src/Reactant.jl b/src/Reactant.jl
@@ -163,7 +163,4 @@ function set_default_backend(backend::String)
     return set_default_backend(XLA.backends[backend])
 end
 
-# include("../ext/ReactantCUDAExt.jl")
-
 end # module
-
diff --git a/src/utils.jl b/src/utils.jl
@@ -584,7 +584,6 @@ function call_with_reactant_generator(
         push!(overdubbed_codelocs, code_info.codelocs[1])
         Core.SSAValue(length(overdubbed_code))
     end
-    end
 
     push!(overdubbed_code, Expr(:(call), oc, fn_args[2:end]...))
 
@@ -611,7 +610,7 @@ function call_with_reactant_generator(
     return code_info
 end
 
-@eval function call_with_reactant(redub_arguments...)
+@eval function call_with_reactant($REDUB_ARGUMENTS_NAME...)
     $(Expr(:meta, :generated_only))
     return $(Expr(:meta, :generated, call_with_reactant_generator))
 end