Skip to content

Commit 17cae90

Browse files
authored
Merge pull request #28 from iamed2/patch-1
Fix reshape on 32-bit
2 parents b3b0106 + 2fff053 commit 17cae90

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/lib/array.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -239,9 +239,9 @@ end
239239
end
240240
end
241241

242-
Base.reshape(xs::TrackedArray, dims::Union{Colon,Int64}...) = reshape(xs, dims)
243-
Base.reshape(xs::TrackedArray, dims::Tuple{Vararg{Union{Int64,Colon}}}) = reshape(xs, Base._reshape_uncolon(xs, dims))
244-
Base.reshape(xs::TrackedArray, dims::Tuple{Vararg{Int64}}) = track(reshape, xs, dims)
242+
Base.reshape(xs::TrackedArray, dims::Union{Colon,Int}...) = reshape(xs, dims)
243+
Base.reshape(xs::TrackedArray, dims::Tuple{Vararg{Union{Int,Colon}}}) = reshape(xs, Base._reshape_uncolon(xs, dims))
244+
Base.reshape(xs::TrackedArray, dims::Tuple{Vararg{Int}}) = track(reshape, xs, dims)
245245

246246
@grad reshape(xs, dims) = reshape(data(xs), dims), Δ -> (reshape(Δ, size(xs)),nothing)
247247

0 commit comments

Comments
 (0)