update, based on review comments

Change-Id: I6175d154af235a31dd1c985827b99bdd7c0ac993
This commit is contained in:
Mateusz Poliwczak 2025-02-17 11:07:02 +01:00
parent 1bacc014a0
commit e5847b26c3
2 changed files with 16 additions and 20 deletions

View File

@ -222,19 +222,14 @@ func HeapAllocReason(n ir.Node) string {
n.Cap = s
}
}
}
if n.Len != nil {
} else if n.Len != nil {
if s := ir.StaticValue(n.Len); s.Op() == ir.OLITERAL {
len, ok := s.(*ir.BasicLit)
if !ok || len.Val().Kind() != constant.Int {
base.Fatalf("unexpected BasicLit Kind")
}
if constant.Compare(len.Val(), token.GEQ, constant.MakeInt64(0)) {
cap, ok := n.Cap.(*ir.BasicLit)
if n.Cap == nil || (ok && constant.Compare(cap.Val(), token.GEQ, len.Val())) {
n.Len = s
}
n.Len = s
}
}
}

View File

@ -14,44 +14,44 @@ var globalVarSize = 128
func testSlices() {
{
size := 128
_ = make([]byte, size) // ERROR "does not escape"
_ = make([]byte, size) // ERROR "make\(\[\]byte, 128\) does not escape"
}
{
s := 128
size := s
_ = make([]byte, size) // ERROR "does not escape"
_ = make([]byte, size) // ERROR "make\(\[\]byte, 128\) does not escape"
}
{
size := 128
_ = make([]byte, 0, size) // ERROR "does not escape"
_ = make([]byte, size) // ERROR "make\(\[\]byte, 128\) does not escape"
}
{
s := 128
size := s
_ = make([]byte, 0, size) // ERROR "does not escape"
_ = make([]byte, size) // ERROR "make\(\[\]byte, 128\) does not escape"
}
{
s1 := 128
s2 := 256
_ = make([]byte, s2, s1) // ERROR "does not escape"
_ = make([]byte, s2, s1) // ERROR "make\(\[\]byte, s2, 128\) does not escape"
}
allocLen(256) // ERROR "does not escape" "inlining call"
allocCap(256) // ERROR "does not escape" "inlining call"
_ = newT(256) // ERROR "does not escape" "inlining call"
allocLen(256) // ERROR "make\(\[\]byte, 256\) does not escape" "inlining call"
allocCap(256) // ERROR "make\(\[\]byte, 0, 256\) does not escape" "inlining call"
_ = newT(256) // ERROR "make\(\[\]byte, 256\) does not escape" "inlining call"
{
size := globalConstSize
_ = make([]byte, size) // ERROR "does not escape"
_ = make([]byte, size) // ERROR "make\(\[\]byte, 128\) does not escape"
}
allocLen(globalConstSize) // ERROR "does not escape" "inlining call"
allocCap(globalConstSize) // ERROR "does not escape" "inlining call"
_ = newT(globalConstSize) // ERROR "does not escape" "inlining call"
allocLen(globalConstSize) // ERROR "make\(\[\]byte, 128\) does not escape" "inlining call"
allocCap(globalConstSize) // ERROR "make\(\[\]byte, 0, 128\) does not escape" "inlining call"
_ = newT(globalConstSize) // ERROR "make\(\[\]byte, 128\) does not escape" "inlining call"
{
c := 128
@ -65,7 +65,8 @@ func testSlices() {
}
{
_ = make([]byte, globalVarSize) // ERROR "escapes to heap"
_ = make([]byte, globalVarSize) // ERROR "make\(\[\]byte, globalVarSize\) escapes to heap"
_ = make([]byte, globalVarSize, globalConstSize) // ERROR "make\(\[\]byte, globalVarSize, 128\) does not escape"
}
}