Skip to content

Commit 3b85689

Browse files
committed
Skip uniqueness check
1 parent 40517a6 commit 3b85689

File tree

1 file changed

+4
-11
lines changed

1 file changed

+4
-11
lines changed

src/lib/broadcast.jl

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -290,16 +290,9 @@ end
290290

291291
∇getindex(x::CUDA.CuArray, inds::Tuple{AbstractArray{<:Integer}}) = dy -> begin
292292
inds1_cpu = Array(inds[1])
293-
if allunique(inds1_cpu)
294-
dx = _zero(x, eltype(dy))
295-
dxv = view(dx, inds[1])
296-
dxv .= accum.(dxv, _droplike(dy, dxv))
297-
return _project(x, dx), nothing
298-
else
299-
dx = zeros(eltype(dy), length(x))
300-
dxv = view(dx, inds1_cpu)
301-
dxv .= accum.(dxv, _droplike(Array(dy), dxv))
302-
return _project(x, CUDA.CuArray(dx)), nothing
303-
end
293+
dx = zeros(eltype(dy), length(x))
294+
dxv = view(dx, inds1_cpu)
295+
dxv .= accum.(dxv, _droplike(Array(dy), dxv))
296+
return _project(x, CUDA.CuArray(dx)), nothing
304297
end
305298
end

0 commit comments

Comments
 (0)