Skip to content

Commit 247d5d2

Browse files
committed
fix syntax
1 parent 3613d26 commit 247d5d2

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/host/mapreduce.jl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -133,8 +133,11 @@ function Base.:(==)(A::AnyGPUArray, B::AnyGPUArray)
133133
res.is_missing ? missing : res.is_equal
134134
end
135135

136+
137+
import KernelAbstractions: @context
138+
136139
@inline function reduce_group(@context, op, val::T, neutral, ::Val{maxitems}) where {T, maxitems}
137-
items = @groupsize[1]
140+
items = @groupsize()[1]
138141
item = @index(Local, Linear)
139142

140143
# local mem for a complete reduction
@@ -213,13 +216,11 @@ Base.@propagate_inbounds _map_getindex(args::Tuple{}, I) = ()
213216
R[Iout] = val
214217
end
215218
end
216-
217-
return
218219
end
219220

220221
## COV_EXCL_STOP
221222

222-
function GPUArrays.mapreducedim!(f::F, op::OP, R::AnyGPUArray{T}, A::AbstractArrayOrBroadcasted;
223+
function mapreducedim!(f::F, op::OP, R::AnyGPUArray{T}, A::AbstractArrayOrBroadcasted;
223224
init=nothing) where {F, OP, T}
224225
Base.check_reducedims(R, A)
225226
length(A) == 0 && return R # isempty(::Broadcasted) iterates

0 commit comments

Comments
 (0)