xref: /aosp_15_r20/external/llvm/test/Transforms/Reassociate/wrap-flags.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker; RUN: opt < %s -reassociate -dce -S | FileCheck %s
2*9880d681SAndroid Build Coastguard Worker; PR12985
3*9880d681SAndroid Build Coastguard Worker
4*9880d681SAndroid Build Coastguard Worker; Verify the nsw flags are preserved when converting shl to mul.
5*9880d681SAndroid Build Coastguard Worker
6*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: @shl_to_mul_nsw(
7*9880d681SAndroid Build Coastguard Worker; CHECK: %mul = mul i32 %i, -2147483648
8*9880d681SAndroid Build Coastguard Worker; CHECK: add i32 %mul, 1
9*9880d681SAndroid Build Coastguard Workerdefine i32 @shl_to_mul_nsw(i32 %i) {
10*9880d681SAndroid Build Coastguard Workerentry:
11*9880d681SAndroid Build Coastguard Worker  %mul = shl nsw i32 %i, 31
12*9880d681SAndroid Build Coastguard Worker  %mul2 = add i32 %mul, 1
13*9880d681SAndroid Build Coastguard Worker  ret i32 %mul2
14*9880d681SAndroid Build Coastguard Worker}
15*9880d681SAndroid Build Coastguard Worker
16*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: @shl_to_mul_nuw(
17*9880d681SAndroid Build Coastguard Worker; CHECK: %mul = mul nuw i32 %i, 4
18*9880d681SAndroid Build Coastguard Worker; CHECK: add i32 %mul, 1
19*9880d681SAndroid Build Coastguard Workerdefine i32 @shl_to_mul_nuw(i32 %i) {
20*9880d681SAndroid Build Coastguard Workerentry:
21*9880d681SAndroid Build Coastguard Worker  %mul = shl nuw i32 %i, 2
22*9880d681SAndroid Build Coastguard Worker  %mul2 = add i32 %mul, 1
23*9880d681SAndroid Build Coastguard Worker  ret i32 %mul2
24*9880d681SAndroid Build Coastguard Worker}
25*9880d681SAndroid Build Coastguard Worker
26*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: @shl_to_mul_nuw_nsw(
27*9880d681SAndroid Build Coastguard Worker; CHECK: %mul = mul nuw nsw i32 %i, 4
28*9880d681SAndroid Build Coastguard Worker; CHECK: add i32 %mul, 1
29*9880d681SAndroid Build Coastguard Workerdefine i32 @shl_to_mul_nuw_nsw(i32 %i) {
30*9880d681SAndroid Build Coastguard Workerentry:
31*9880d681SAndroid Build Coastguard Worker  %mul = shl nuw nsw i32 %i, 2
32*9880d681SAndroid Build Coastguard Worker  %mul2 = add i32 %mul, 1
33*9880d681SAndroid Build Coastguard Worker  ret i32 %mul2
34*9880d681SAndroid Build Coastguard Worker}
35*9880d681SAndroid Build Coastguard Worker
36*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: @pr23926(
37*9880d681SAndroid Build Coastguard Worker; CHECK:       %[[X1_neg:.*]] = sub i2 0, %X1
38*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:  %[[sub_one:.*]] = add i2 %[[X1_neg]], -1
39*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:  %[[add:.*]] = add i2 %[[sub_one]], %X2
40*9880d681SAndroid Build Coastguard Worker; CHECK-NEXT:  ret i2 %[[add]]
41*9880d681SAndroid Build Coastguard Workerdefine i2 @pr23926(i2 %X1, i2 %X2) {
42*9880d681SAndroid Build Coastguard Worker  %add = add nuw i2 %X1, 1
43*9880d681SAndroid Build Coastguard Worker  %sub = sub nuw nsw i2 %X2, %add
44*9880d681SAndroid Build Coastguard Worker  ret i2 %sub
45*9880d681SAndroid Build Coastguard Worker}
46