Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 2 additions & 5 deletions ext/ReactantKernelAbstractionsExt.jl
Original file line number Diff line number Diff line change
Expand Up @@ -109,12 +109,9 @@ function (obj::KA.Kernel{ReactantBackend})(args...; ndrange=nothing, workgroupsi
return nothing
end

Reactant.@reactant_overlay @noinline Base.@nospecializeinfer function (
obj::KA.Kernel{ReactantBackend}
)(
args...; ndrange=nothing, workgroupsize=nothing
Reactant.@reactant_overlay Base.@nospecializeinfer @noinline function (obj::KA.Kernel{ReactantBackend})(
@nospecialize args...; ndrange=nothing, workgroupsize=nothing
)
@nospecialize
return Reactant.call_with_reactant(
Reactant.ka_with_reactant, ndrange, workgroupsize, obj, args...
)
Expand Down
2 changes: 1 addition & 1 deletion src/utils.jl
Original file line number Diff line number Diff line change
Expand Up @@ -610,7 +610,7 @@ end
# using a custom interpreter in type unstable code.
# `redub_arguments` is `(typeof(original_function), map(typeof, original_args_tuple)...)`
function call_with_reactant_generator(
world::UInt, source::LineNumberNode, self, @nospecialize(redub_arguments)
world::UInt, source::Union{LineNumberNode, Core.Method}, self, @nospecialize(redub_arguments)
)
@nospecialize
args = redub_arguments
Expand Down
Loading