From bc82c359d4fe08750915f5db84fa83bbef063629 Mon Sep 17 00:00:00 2001 From: Jeff Bezanson Date: Tue, 3 Dec 2019 04:39:45 -0500 Subject: [PATCH] fix #33974, wrong integer types used in `jl_array_sizehint` (#34005) --- src/array.c | 6 +++--- test/arrayops.jl | 11 +++++++++++ 2 files changed, 14 insertions(+), 3 deletions(-) diff --git a/src/array.c b/src/array.c index c38b9953e0751..4410858f8fd8b 100644 --- a/src/array.c +++ b/src/array.c @@ -933,8 +933,8 @@ STATIC_INLINE void jl_array_shrink(jl_array_t *a, size_t dec) if (a->flags.how == 0) return; size_t elsz = a->elsize; - int newbytes = (a->maxsize - dec) * a->elsize; - int oldnbytes = (a->maxsize) * a->elsize; + size_t newbytes = (a->maxsize - dec) * a->elsize; + size_t oldnbytes = (a->maxsize) * a->elsize; int isbitsunion = jl_array_isbitsunion(a); if (isbitsunion) { newbytes += a->maxsize - dec; @@ -1107,7 +1107,7 @@ JL_DLLEXPORT void jl_array_sizehint(jl_array_t *a, size_t sz) { size_t n = jl_array_nrows(a); - int min = a->offset + a->length; + size_t min = a->offset + a->length; sz = (sz < min) ? min : sz; if (sz <= a->maxsize) { diff --git a/test/arrayops.jl b/test/arrayops.jl index c1eb1156bad61..4f4f9c53b865b 100644 --- a/test/arrayops.jl +++ b/test/arrayops.jl @@ -2657,3 +2657,14 @@ end C = hcat(A, B) @test typeof(C) == Array{Array{Float64,2},2} end + +# issue #33974 +let n = 12000000, k = 257000000 + # tests skipped since they use a lot of memory + @test_skip filter(x -> x[2] < 1.0, collect(enumerate(vcat(fill(0.5, n), fill(NaN, k)))))[end] == (n, 0.5) + @test_skip let v = collect(enumerate(vcat(fill(0.5, n), fill(NaN, k)))) + resize!(v, n) + sizehint!(v, n) + v[end] == (n, 0.5) + end +end