Skip to content

Commit d7f487b

Browse files
vchuravymaleadt
authored andcommitted
fix syntax
1 parent 36cbcee commit d7f487b

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/host/mapreduce.jl

+5-4
Original file line numberDiff line numberDiff line change
@@ -143,8 +143,11 @@ function Base.:(==)(A::AnyGPUArray, B::AnyGPUArray)
143143
res.is_missing ? missing : res.is_equal
144144
end
145145

146+
147+
import KernelAbstractions: @context
148+
146149
@inline function reduce_group(@context, op, val::T, neutral, ::Val{maxitems}) where {T, maxitems}
147-
items = @groupsize[1]
150+
items = @groupsize()[1]
148151
item = @index(Local, Linear)
149152

150153
# local mem for a complete reduction
@@ -223,13 +226,11 @@ Base.@propagate_inbounds _map_getindex(args::Tuple{}, I) = ()
223226
R[Iout] = val
224227
end
225228
end
226-
227-
return
228229
end
229230

230231
## COV_EXCL_STOP
231232

232-
function GPUArrays.mapreducedim!(f::F, op::OP, R::AnyGPUArray{T}, A::AbstractArrayOrBroadcasted;
233+
function mapreducedim!(f::F, op::OP, R::AnyGPUArray{T}, A::AbstractArrayOrBroadcasted;
233234
init=nothing) where {F, OP, T}
234235
Base.check_reducedims(R, A)
235236
length(A) == 0 && return R # isempty(::Broadcasted) iterates

0 commit comments

Comments
 (0)