Fix for Enzyme.

JuliaGPU · Aug 23, 2024 · d53854d · d53854d
1 parent cc0c716
commit d53854d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/compiler/execution.jl b/src/compiler/execution.jl
@@ -245,7 +245,8 @@ function Base.show(io::IO, ::MIME"text/plain", k::AbstractKernel{F,TT}) where {F
     print(io, "CUDA.$(nameof(typeof(k))) for $(k.f)($(join(TT.parameters, ", ")))")
 end
 
-@inline @generated function (kernel::AbstractKernel{F,TT})(args...; call_kwargs...) where {F,TT}
+@inline @generated function (kernel::AbstractKernel{F,TT})(args::Vararg{Any,N};
+                                                           call_kwargs...) where {F,TT,N}
     sig = Tuple{F, TT.parameters...}    # Base.signature_type with a function type
 
     # determine argument expressions