Skip to content

Commit 20f494e

Browse files
committed
add similar function
1 parent 6da1def commit 20f494e

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/KernelAbstractions.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -356,6 +356,10 @@ struct Kernel{Device, WorkgroupSize<:_Size, NDRange<:_Size, Fun}
356356
f::Fun
357357
end
358358

359+
function Base.similar(kernel::Kernel{D, WS, ND}, f::F) where {D, WS, ND, F}
360+
Kernel{D, WS, ND, F}(f)
361+
end
362+
359363
workgroupsize(::Kernel{D, WorkgroupSize}) where {D, WorkgroupSize} = WorkgroupSize
360364
ndrange(::Kernel{D, WorkgroupSize, NDRange}) where {D, WorkgroupSize,NDRange} = NDRange
361365

0 commit comments

Comments
 (0)