diff --git a/src/Reactant.jl b/src/Reactant.jl index 6be3aa10..ba2da588 100644 --- 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 index 3a5f6feb..b65077c0 100644 --- 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