Files
clang-p2996/llvm/test/Transforms/MemCpyOpt/callslot_throw.ll
Nikita Popov bf5d96c96c [IR] Add dead_on_unwind attribute (#74289)
Add the `dead_on_unwind` attribute, which states that the caller will
not read from this argument if the call unwinds. This allows eliding
stores that could otherwise be visible on the unwind path, for example:

```
declare void @may_unwind()

define void @src(ptr noalias dead_on_unwind %out) {
    store i32 0, ptr %out
    call void @may_unwind()
    store i32 1, ptr %out
    ret void
}

define void @tgt(ptr noalias dead_on_unwind %out) {
    call void @may_unwind()
    store i32 1, ptr %out
    ret void
}
```

The optimization is not valid without `dead_on_unwind`, because the `i32
0` value might be read if `@may_unwind` unwinds.

This attribute is primarily intended to be used on sret arguments. In
fact, I previously wanted to change the semantics of sret to include
this "no read after unwind" property (see D116998), but based on the
feedback there it is better to keep these attributes orthogonal (sret is
an ABI attribute, dead_on_unwind is an optimization attribute). This is
a reboot of that change with a separate attribute.
2023-12-14 09:58:14 +01:00

106 lines
3.4 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -S -passes=memcpyopt < %s -verify-memoryssa | FileCheck %s
declare void @may_throw(ptr nocapture %x)
define void @test1(ptr nocapture noalias dereferenceable(4) %x) {
; CHECK-LABEL: @test1(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[T:%.*]] = alloca i32, align 4
; CHECK-NEXT: call void @may_throw(ptr nonnull [[T]])
; CHECK-NEXT: [[LOAD:%.*]] = load i32, ptr [[T]], align 4
; CHECK-NEXT: store i32 [[LOAD]], ptr [[X:%.*]], align 4
; CHECK-NEXT: ret void
;
entry:
%t = alloca i32, align 4
call void @may_throw(ptr nonnull %t)
%load = load i32, ptr %t, align 4
store i32 %load, ptr %x, align 4
ret void
}
declare void @always_throws()
define void @test2(ptr nocapture noalias dereferenceable(4) %x) {
; CHECK-LABEL: @test2(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[T:%.*]] = alloca i32, align 4
; CHECK-NEXT: call void @may_throw(ptr nonnull [[T]]) #[[ATTR0:[0-9]+]]
; CHECK-NEXT: [[LOAD:%.*]] = load i32, ptr [[T]], align 4
; CHECK-NEXT: call void @always_throws()
; CHECK-NEXT: store i32 [[LOAD]], ptr [[X:%.*]], align 4
; CHECK-NEXT: ret void
;
entry:
%t = alloca i32, align 4
call void @may_throw(ptr nonnull %t) nounwind
%load = load i32, ptr %t, align 4
call void @always_throws()
store i32 %load, ptr %x, align 4
ret void
}
; byval argument is not visible on unwind.
define void @test_byval(ptr nocapture noalias dereferenceable(4) byval(i32) %x) {
; CHECK-LABEL: @test_byval(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[T:%.*]] = alloca i32, align 4
; CHECK-NEXT: call void @may_throw(ptr nonnull [[X:%.*]])
; CHECK-NEXT: ret void
;
entry:
%t = alloca i32, align 4
call void @may_throw(ptr nonnull %t)
%load = load i32, ptr %t, align 4
store i32 %load, ptr %x, align 4
ret void
}
define void @test_dead_on_unwind(ptr nocapture noalias writable dead_on_unwind dereferenceable(4) %x) {
; CHECK-LABEL: @test_dead_on_unwind(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[T:%.*]] = alloca i32, align 4
; CHECK-NEXT: call void @may_throw(ptr nonnull [[X:%.*]])
; CHECK-NEXT: ret void
;
entry:
%t = alloca i32, align 4
call void @may_throw(ptr nonnull %t)
%load = load i32, ptr %t, align 4
store i32 %load, ptr %x, align 4
ret void
}
; Same as previous test, but dereferenceability information is provided by sret.
define void @test_dead_on_unwind_sret(ptr nocapture noalias writable dead_on_unwind sret(i32) %x) {
; CHECK-LABEL: @test_dead_on_unwind_sret(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[T:%.*]] = alloca i32, align 4
; CHECK-NEXT: call void @may_throw(ptr nonnull [[X:%.*]])
; CHECK-NEXT: ret void
;
entry:
%t = alloca i32, align 4
call void @may_throw(ptr nonnull %t)
%load = load i32, ptr %t, align 4
store i32 %load, ptr %x, align 4
ret void
}
define void @test_dead_on_unwind_missing_writable(ptr nocapture noalias dead_on_unwind dereferenceable(4) %x) {
; CHECK-LABEL: @test_dead_on_unwind_missing_writable(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[T:%.*]] = alloca i32, align 4
; CHECK-NEXT: call void @may_throw(ptr nonnull [[T]])
; CHECK-NEXT: [[LOAD:%.*]] = load i32, ptr [[T]], align 4
; CHECK-NEXT: store i32 [[LOAD]], ptr [[X:%.*]], align 4
; CHECK-NEXT: ret void
;
entry:
%t = alloca i32, align 4
call void @may_throw(ptr nonnull %t)
%load = load i32, ptr %t, align 4
store i32 %load, ptr %x, align 4
ret void
}