代码拉取完成,页面将自动刷新
# This file is a part of Julia. License is MIT: https://julialang.org/license
module TestBoundsCheck
using Test, Random, InteractiveUtils
@enum BCOption bc_default bc_on bc_off
bc_opt = BCOption(Base.JLOptions().check_bounds)
# test for boundscheck block eliminated at same level
@inline function A1()
r = 0
@boundscheck r += 1
return r
end
@noinline function A1_noinline()
r = 0
@boundscheck r += 1
return r
end
function A1_inbounds()
r = 0
@inbounds begin
@boundscheck r += 1
end
return r
end
A1_wrap() = @inbounds return A1_inbounds()
if bc_opt == bc_default
@test A1() == 1
@test A1_inbounds() == 1
@test A1_wrap() == 0
elseif bc_opt == bc_on
@test A1() == 1
@test A1_inbounds() == 1
@test A1_wrap() == 1
else
@test A1() == 0
@test A1_inbounds() == 0
@test A1_wrap() == 0
end
# test for boundscheck block eliminated one layer deep, if the called method is inlined
@inline function A2()
r = A1()+1
return r
end
function A2_inbounds()
@inbounds r = A1()+1
return r
end
function A2_notinlined()
@inbounds r = A1_noinline()+1
return r
end
Base.@propagate_inbounds function A2_propagate_inbounds()
r = A1()+1
return r
end
if bc_opt == bc_default
@test A2() == 2
@test A2_inbounds() == 1
@test A2_notinlined() == 2
@test A2_propagate_inbounds() == 2
elseif bc_opt == bc_on
@test A2() == 2
@test A2_inbounds() == 2
@test A2_notinlined() == 2
@test A2_propagate_inbounds() == 2
else
@test A2() == 1
@test A2_inbounds() == 1
@test A2_notinlined() == 1
@test A2_propagate_inbounds() == 1
end
# test boundscheck NOT eliminated two layers deep, unless propagated
function A3()
r = A2()+1
return r
end
function A3_inbounds()
@inbounds r = A2()+1
return r
end
function A3_inbounds2()
@inbounds r = A2_propagate_inbounds()+1
return r
end
if bc_opt == bc_default
@test A3() == 3
@test A3_inbounds() == 3
@test A3_inbounds2() == 2
elseif bc_opt == bc_on
@test A3() == 3
@test A3_inbounds() == 3
@test A3_inbounds2() == 3
else
@test A3() == 2
@test A3_inbounds() == 2
@test A3_inbounds2() == 2
end
# swapped nesting order of @boundscheck and @inbounds
function A1_nested()
r = 0
@boundscheck @inbounds r += 1
return r
end
if bc_opt == bc_default || bc_opt == bc_on
@test A1_nested() == 1
else
@test A1_nested() == 0
end
# elide a throw
cb(x) = x > 0 || throw(BoundsError())
@inline function B1()
y = [1, 2, 3]
@inbounds begin
@boundscheck cb(0)
end
return 0
end
B1_wrap() = @inbounds return B1()
if bc_opt == bc_default
@test_throws BoundsError B1()
@test B1_wrap() == 0
elseif bc_opt == bc_off
@test B1() == 0
@test B1_wrap() == 0
else
@test_throws BoundsError B1()
@test_throws BoundsError B1_wrap()
end
# elide a simple branch
cond(x) = x > 0 ? x : -x
function B2()
y = [1, 2, 3]
@inbounds begin
@boundscheck cond(0)
end
return 0
end
@test B2() == 0
# Make sure type inference doesn't incorrectly optimize out
# `Expr(:inbounds, false)`
# Simply `return a[1]` doesn't work due to inlining bug
@inline function f1(a)
# This has to be an arrayget / arrayset since these currently have a
# implicit `Expr(:boundscheck)` that's not visible to type inference
x = a[1]
return x
end
# second level
@inline function g1(a)
x = f1(a)
return x
end
function k1(a)
# This `Expr(:inbounds, true)` shouldn't affect `f1`
@inbounds x = g1(a)
return x
end
if bc_opt != bc_off
@test_throws BoundsError k1(Int[])
end
# Ensure that broadcast doesn't use @inbounds when calling the function
if bc_opt != bc_off
let A = zeros(3,3)
@test_throws BoundsError broadcast(getindex, A, 1:3, 1:3)
end
end
# issue #19554
function f19554(a)
a[][3]
end
function f19554_2(a, b)
a[][3] = b
return a
end
a19554 = Ref{Array{Float64}}([1 2; 3 4])
@test f19554(a19554) === 2.0
@test f19554_2(a19554, 1) === a19554
@test a19554[][3] === f19554(a19554) === 1.0
# Ensure unsafe_view doesn't check bounds
function V1()
A = rand(10,10)
B = view(A, 4:7, 4:7)
C = Base.unsafe_view(B, -2:7, -2:7)
@test C == A
nothing
end
if bc_opt == bc_default || bc_opt == bc_off
@test V1() === nothing
else
@test_throws BoundsError V1()
end
# This tests both the bounds check elision and the behavior of `jl_array_isassigned`
# For `isbits` array the `ccall` should return a constant `true` and does not access
# the array
inbounds_isassigned(a, i) = @inbounds return isassigned(a, i)
if bc_opt == bc_default || bc_opt == bc_off
@test inbounds_isassigned(Int[], 2) == true
else
@test inbounds_isassigned(Int[], 2) == false
end
# Test that @inbounds annotations don't propagate too far for Array; Issue #20469
struct BadVector20469{T} <: AbstractVector{Int}
data::T
end
Base.size(X::BadVector20469) = size(X.data)
Base.getindex(X::BadVector20469, i::Int) = X.data[i-1]
if bc_opt != bc_off
@test_throws BoundsError BadVector20469([1,2,3])[:]
end
# Accumulate: do not set inbounds context for user-supplied functions
if bc_opt != bc_off
Base.@propagate_inbounds op58200(a, b) = (1, 2)[a] + (1, 2)[b]
@test_throws BoundsError accumulate(op58200, 1:10)
@test_throws BoundsError Base.accumulate_pairwise(op58200, 1:10)
end
# Ensure iteration over arrays is vectorizable
function g27079(X)
r = 0
for x in X
r += x
end
r
end
@test occursin("vector.reduce.add", sprint(code_llvm, g27079, Tuple{Vector{Int}}))
# Boundschecking removal of indices with different type, see #40281
getindex_40281(v, a, b, c) = @inbounds getindex(v, a, b, c)
llvm_40281 = sprint((io, args...) -> code_llvm(io, args...; optimize=true), getindex_40281, Tuple{Array{Float64, 3}, Int, UInt8, Int})
if bc_opt == bc_default || bc_opt == bc_off
@test !occursin("call void @ijl_bounds_error_ints", llvm_40281)
end
# Given this is a sub-processed test file, not using @testsets avoids
# leaking the report print into the Base test runner report
begin # Pass inbounds meta to getindex on CartesianIndices (#42115)
@inline getindex_42115(r, i) = @inbounds getindex(r, i)
@inline getindex_42115(r, i, j) = @inbounds getindex(r, i, j)
R = CartesianIndices((5, 5))
if bc_opt == bc_on
@test_throws BoundsError getindex_42115(R, -1, -1)
@test_throws BoundsError getindex_42115(R, 1, -1)
else
@test getindex_42115(R, -1, -1) == CartesianIndex(-1, -1)
@test getindex_42115(R, 1, -1) == CartesianIndex(1, -1)
end
if bc_opt == bc_on
@test_throws BoundsError getindex_42115(R, CartesianIndices((6, 6)))
@test_throws BoundsError getindex_42115(R, -1:3, :)
else
@test getindex_42115(R, CartesianIndices((6, 6))) == CartesianIndices((6, 6))
@test getindex_42115(R, -1:3, :) == CartesianIndices((-1:3, 1:5))
end
end
# Test that --check-bounds=off doesn't permit const prop of indices into
# function that are not dynamically reachable (the same test for @inbounds
# is in the compiler tests).
function f_boundscheck_elim(n)
# Inbounds here assumes that this is only ever called with n==0, but of
# course the compiler has no way of knowing that, so it must not attempt
# to run the @inbounds `getfield(sin, 1)`` that ntuple generates.
ntuple(x->getfield(sin, x), n)
end
@test Tuple{} <: code_typed(f_boundscheck_elim, Tuple{Int})[1][2]
# https://github.com/JuliaArrays/StaticArrays.jl/issues/1155
@test Base.return_types() do
typeintersect(Int, Integer)
end |> only === Type{Int}
if bc_opt == bc_default
# Array/Memory escape analysis
function no_allocate(T::Type{<:Union{Memory}})
v = T(undef, 2)
v[1] = 2
v[2] = 3
return v[1] + v[2]
end
function test_alloc(::Type{T}; broken=false) where T
@test (@allocated no_allocate(T)) == 0 broken=broken
end
for T in [Memory] # This requires changing the pointer_from_objref to something llvm sees through
for ET in [Int, Float32, Union{Int, Float64}]
no_allocate(T{ET}) #compile
# allocations aren't removed for Union eltypes which they theoretically could be eventually
test_alloc(T{ET}, broken=(ET==Union{Int, Float64}))
end
end
function f() # this was causing a bug on an in progress version of #55913.
m = Memory{Float64}(undef, 4)
m .= 1.0
s = 0.0
for x ∈ m
s += x
end
s
end
@test f() === 4.0
function confuse_alias_analysis()
mem0 = Memory{Int}(undef, 1)
mem1 = Memory{Int}(undef, 1)
@inbounds mem0[1] = 3
for width in 1:2
@inbounds mem1[1] = mem0[1]
mem0 = mem1
end
mem0[1]
end
@test confuse_alias_analysis() == 3
@test (@allocated confuse_alias_analysis()) == 0
function no_alias_prove(n)
m1 = Memory{Int}(undef,n)
m2 = Memory{Int}(undef,n)
m1 === m2
end
no_alias_prove(1)
@test (@allocated no_alias_prove(5)) == 0
end
@testset "automatic boundscheck elision for iteration on some important types" begin
if bc_opt != bc_on
@test !contains(sprint(code_llvm, iterate, (Memory{UInt8}, Int)), "unreachable")
@test !contains(sprint(code_llvm, iterate, (Vector{UInt8}, Int)), "unreachable")
@test !contains(sprint(code_llvm, iterate, (Matrix{UInt8}, Int)), "unreachable")
@test !contains(sprint(code_llvm, iterate, (Array{UInt8,3}, Int)), "unreachable")
@test !contains(sprint(code_llvm, iterate, (SubArray{Float64, 1, Vector{Float64}, Tuple{Base.Slice{Base.OneTo{Int64}}}, true}, Int)), "unreachable")
@test !contains(sprint(code_llvm, iterate, (SubArray{Float64, 2, Matrix{Float64}, Tuple{Base.Slice{Base.OneTo{Int64}}, Base.Slice{Base.OneTo{Int64}}}, true}, Int)), "unreachable")
@test !contains(sprint(code_llvm, iterate, (SubArray{Float64, 2, Matrix{Float64}, Tuple{Base.Slice{Base.OneTo{Int64}}, UnitRange{Int64}}, true}, Int)), "unreachable")
@test !contains(sprint(code_llvm, iterate, (Base.CodeUnits{UInt8,String}, Int)), "unreachable")
end
end
end
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。