Files
clang-p2996/llvm/test/Transforms/Mem2Reg/single-store.ll
Nikita Popov f58930f705 [Mem2Reg] Don't use single store optimization for potentially poison value (#97711)
If there is a single store, then loads must either load the stored value
or uninitialized memory (undef). If the stored value may be poison, then
replacing an uninitialized memory load with it would be incorrect. Fall
back to the generic code in that case.

This PR only fixes the case where there is a literal poison store -- the
case where the value is non-trivially poison will still get miscompiled
by phi simplification later, see #96631.

Fixes https://github.com/llvm/llvm-project/issues/97702.
2024-07-04 14:41:12 +02:00

67 lines
1.7 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
; RUN: opt -S -passes=mem2reg < %s | FileCheck %s
define i8 @single_store_literal_poison(i1 %cond) {
; CHECK-LABEL: define i8 @single_store_literal_poison(
; CHECK-SAME: i1 [[COND:%.*]]) {
; CHECK-NEXT: br i1 [[COND]], label %[[IF:.*]], label %[[EXIT:.*]]
; CHECK: [[IF]]:
; CHECK-NEXT: br label %[[EXIT]]
; CHECK: [[EXIT]]:
; CHECK-NEXT: ret i8 undef
;
%a = alloca i8, align 1
br i1 %cond, label %if, label %exit
if:
store i8 poison, ptr %a, align 1
br label %exit
exit:
%v = load i8, ptr %a, align 1
ret i8 %v
}
; FIXME: This is a miscompile.
define i8 @single_store_maybe_poison(i1 %cond, i8 %x) {
; CHECK-LABEL: define i8 @single_store_maybe_poison(
; CHECK-SAME: i1 [[COND:%.*]], i8 [[X:%.*]]) {
; CHECK-NEXT: br i1 [[COND]], label %[[IF:.*]], label %[[EXIT:.*]]
; CHECK: [[IF]]:
; CHECK-NEXT: br label %[[EXIT]]
; CHECK: [[EXIT]]:
; CHECK-NEXT: ret i8 [[X]]
;
%a = alloca i8, align 1
br i1 %cond, label %if, label %exit
if:
store i8 %x, ptr %a, align 1
br label %exit
exit:
%v = load i8, ptr %a, align 1
ret i8 %v
}
define i8 @single_store_cant_be_poison(i1 %cond, i8 noundef %x) {
; CHECK-LABEL: define i8 @single_store_cant_be_poison(
; CHECK-SAME: i1 [[COND:%.*]], i8 noundef [[X:%.*]]) {
; CHECK-NEXT: br i1 [[COND]], label %[[IF:.*]], label %[[EXIT:.*]]
; CHECK: [[IF]]:
; CHECK-NEXT: br label %[[EXIT]]
; CHECK: [[EXIT]]:
; CHECK-NEXT: ret i8 [[X]]
;
%a = alloca i8, align 1
br i1 %cond, label %if, label %exit
if:
store i8 %x, ptr %a, align 1
br label %exit
exit:
%v = load i8, ptr %a, align 1
ret i8 %v
}