Skip to content

Commit 45625a0

Browse files
committed
fix colon indexing
1 parent 6a6f354 commit 45625a0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/abstractarray.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,13 +59,15 @@ function to_cartesian(A, indices::Tuple)
5959
isa(val, Integer) && return val
6060
isa(val, UnitRange) && return first(val)
6161
isa(val, Colon) && return 1
62+
isa(val, Base.Slice{Base.OneTo{Int}}) && return 1
6263
error("GPU indexing only defined for integers or unit ranges. Found: $val")
6364
end)
6465
stop = CartesianIndex(ntuple(length(indices)) do i
6566
val = indices[i]
6667
isa(val, Integer) && return val
6768
isa(val, UnitRange) && return last(val)
6869
isa(val, Colon) && return size(A, i)
70+
isa(val, Base.Slice{Base.OneTo{Int}}) && return size(A, i)
6971
error("GPU indexing only defined for integers or unit ranges. Found: $val")
7072
end)
7173
CartesianRange(start, stop)

0 commit comments

Comments
 (0)