| ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 4 |
| ; RUN: opt < %s -passes=slsr,gvn -S | FileCheck %s |
| |
| target datalayout = "e-i64:64-v16:16-v32:32-n16:32:64-p:64:64:64-p1:32:32:32-p2:128:128:128:32" |
| |
| ; foo(input[0]); |
| ; foo(input[s]); |
| ; foo(input[s * 2]); |
| ; => |
| ; p0 = &input[0]; |
| ; foo(*p); |
| ; p1 = p0 + s; |
| ; foo(*p1); |
| ; p2 = p1 + s; |
| ; foo(*p2); |
| define void @slsr_gep(ptr %input, i64 %s) { |
| ; CHECK-LABEL: define void @slsr_gep( |
| ; CHECK-SAME: ptr [[INPUT:%.*]], i64 [[S:%.*]]) { |
| ; CHECK-NEXT: call void @foo(ptr [[INPUT]]) |
| ; CHECK-NEXT: [[P1:%.*]] = getelementptr inbounds i32, ptr [[INPUT]], i64 [[S]] |
| ; CHECK-NEXT: call void @foo(ptr [[P1]]) |
| ; CHECK-NEXT: [[TMP1:%.*]] = shl i64 [[S]], 2 |
| ; CHECK-NEXT: [[P2:%.*]] = getelementptr inbounds i8, ptr [[P1]], i64 [[TMP1]] |
| ; CHECK-NEXT: call void @foo(ptr [[P2]]) |
| ; CHECK-NEXT: ret void |
| ; |
| ; v0 = input[0]; |
| call void @foo(ptr %input) |
| |
| ; v1 = input[s]; |
| %p1 = getelementptr inbounds i32, ptr %input, i64 %s |
| call void @foo(ptr %p1) |
| |
| ; v2 = input[s * 2]; |
| %s2 = shl nsw i64 %s, 1 |
| %p2 = getelementptr inbounds i32, ptr %input, i64 %s2 |
| call void @foo(ptr %p2) |
| |
| ret void |
| } |
| |
| ; foo(input[0]); |
| ; foo(input[(long)s]); |
| ; foo(input[(long)(s * 2)]); |
| ; => |
| ; p0 = &input[0]; |
| ; foo(*p); |
| ; p1 = p0 + (long)s; |
| ; foo(*p1); |
| ; p2 = p1 + (long)s; |
| ; foo(*p2); |
| define void @slsr_gep_sext(ptr %input, i32 %s) { |
| ; CHECK-LABEL: define void @slsr_gep_sext( |
| ; CHECK-SAME: ptr [[INPUT:%.*]], i32 [[S:%.*]]) { |
| ; CHECK-NEXT: call void @foo(ptr [[INPUT]]) |
| ; CHECK-NEXT: [[T:%.*]] = sext i32 [[S]] to i64 |
| ; CHECK-NEXT: [[P1:%.*]] = getelementptr inbounds i32, ptr [[INPUT]], i64 [[T]] |
| ; CHECK-NEXT: call void @foo(ptr [[P1]]) |
| ; CHECK-NEXT: [[TMP1:%.*]] = shl i64 [[T]], 2 |
| ; CHECK-NEXT: [[P2:%.*]] = getelementptr inbounds i8, ptr [[P1]], i64 [[TMP1]] |
| ; CHECK-NEXT: call void @foo(ptr [[P2]]) |
| ; CHECK-NEXT: ret void |
| ; |
| ; v0 = input[0]; |
| call void @foo(ptr %input) |
| |
| ; v1 = input[s]; |
| %t = sext i32 %s to i64 |
| %p1 = getelementptr inbounds i32, ptr %input, i64 %t |
| call void @foo(ptr %p1) |
| |
| ; v2 = input[s * 2]; |
| %s2 = shl nsw i32 %s, 1 |
| %t2 = sext i32 %s2 to i64 |
| %p2 = getelementptr inbounds i32, ptr %input, i64 %t2 |
| call void @foo(ptr %p2) |
| |
| ret void |
| } |
| |
| ; int input[10][5]; |
| ; foo(input[s][t]); |
| ; foo(input[s * 2][t]); |
| ; foo(input[s * 3][t]); |
| ; => |
| ; p0 = &input[s][t]; |
| ; foo(*p0); |
| ; p1 = p0 + 5s; |
| ; foo(*p1); |
| ; p2 = p1 + 5s; |
| ; foo(*p2); |
| define void @slsr_gep_2d(ptr %input, i64 %s, i64 %t) { |
| ; CHECK-LABEL: define void @slsr_gep_2d( |
| ; CHECK-SAME: ptr [[INPUT:%.*]], i64 [[S:%.*]], i64 [[T:%.*]]) { |
| ; CHECK-NEXT: [[P0:%.*]] = getelementptr inbounds [10 x [5 x i32]], ptr [[INPUT]], i64 0, i64 [[S]], i64 [[T]] |
| ; CHECK-NEXT: call void @foo(ptr [[P0]]) |
| ; CHECK-NEXT: [[TMP1:%.*]] = mul i64 [[S]], 20 |
| ; CHECK-NEXT: [[P1:%.*]] = getelementptr inbounds i8, ptr [[P0]], i64 [[TMP1]] |
| ; CHECK-NEXT: call void @foo(ptr [[P1]]) |
| ; CHECK-NEXT: [[P2:%.*]] = getelementptr inbounds i8, ptr [[P1]], i64 [[TMP1]] |
| ; CHECK-NEXT: call void @foo(ptr [[P2]]) |
| ; CHECK-NEXT: ret void |
| ; |
| ; v0 = input[s][t]; |
| %p0 = getelementptr inbounds [10 x [5 x i32]], ptr %input, i64 0, i64 %s, i64 %t |
| call void @foo(ptr %p0) |
| |
| ; v1 = input[s * 2][t]; |
| %s2 = shl nsw i64 %s, 1 |
| %p1 = getelementptr inbounds [10 x [5 x i32]], ptr %input, i64 0, i64 %s2, i64 %t |
| call void @foo(ptr %p1) |
| |
| ; v3 = input[s * 3][t]; |
| %s3 = mul nsw i64 %s, 3 |
| %p2 = getelementptr inbounds [10 x [5 x i32]], ptr %input, i64 0, i64 %s3, i64 %t |
| call void @foo(ptr %p2) |
| |
| ret void |
| } |
| |
| %struct.S = type <{ i64, i32 }> |
| |
| ; In this case, the bump |
| ; = (char *)&input[s * 2][t].f1 - (char *)&input[s][t].f1 |
| ; = 60 * s |
| ; which may not be divisible by typeof(input[s][t].f1) = 8. Therefore, we |
| ; rewrite the candidates using byte offset instead of index offset as in |
| ; @slsr_gep_2d. |
| define void @slsr_gep_uglygep(ptr %input, i64 %s, i64 %t) { |
| ; CHECK-LABEL: define void @slsr_gep_uglygep( |
| ; CHECK-SAME: ptr [[INPUT:%.*]], i64 [[S:%.*]], i64 [[T:%.*]]) { |
| ; CHECK-NEXT: [[P0:%.*]] = getelementptr inbounds [10 x [5 x %struct.S]], ptr [[INPUT]], i64 0, i64 [[S]], i64 [[T]], i32 0 |
| ; CHECK-NEXT: call void @bar(ptr [[P0]]) |
| ; CHECK-NEXT: [[TMP1:%.*]] = mul i64 [[S]], 60 |
| ; CHECK-NEXT: [[P1:%.*]] = getelementptr inbounds i8, ptr [[P0]], i64 [[TMP1]] |
| ; CHECK-NEXT: call void @bar(ptr [[P1]]) |
| ; CHECK-NEXT: [[P2:%.*]] = getelementptr inbounds i8, ptr [[P1]], i64 [[TMP1]] |
| ; CHECK-NEXT: call void @bar(ptr [[P2]]) |
| ; CHECK-NEXT: ret void |
| ; |
| ; v0 = input[s][t].f1; |
| %p0 = getelementptr inbounds [10 x [5 x %struct.S]], ptr %input, i64 0, i64 %s, i64 %t, i32 0 |
| call void @bar(ptr %p0) |
| |
| ; v1 = input[s * 2][t].f1; |
| %s2 = shl nsw i64 %s, 1 |
| %p1 = getelementptr inbounds [10 x [5 x %struct.S]], ptr %input, i64 0, i64 %s2, i64 %t, i32 0 |
| call void @bar(ptr %p1) |
| |
| ; v2 = input[s * 3][t].f1; |
| %s3 = mul nsw i64 %s, 3 |
| %p2 = getelementptr inbounds [10 x [5 x %struct.S]], ptr %input, i64 0, i64 %s3, i64 %t, i32 0 |
| call void @bar(ptr %p2) |
| |
| ret void |
| } |
| |
| define void @slsr_out_of_bounds_gep(ptr %input, i32 %s) { |
| ; CHECK-LABEL: define void @slsr_out_of_bounds_gep( |
| ; CHECK-SAME: ptr [[INPUT:%.*]], i32 [[S:%.*]]) { |
| ; CHECK-NEXT: call void @foo(ptr [[INPUT]]) |
| ; CHECK-NEXT: [[T:%.*]] = sext i32 [[S]] to i64 |
| ; CHECK-NEXT: [[P1:%.*]] = getelementptr i32, ptr [[INPUT]], i64 [[T]] |
| ; CHECK-NEXT: call void @foo(ptr [[P1]]) |
| ; CHECK-NEXT: [[TMP1:%.*]] = shl i64 [[T]], 2 |
| ; CHECK-NEXT: [[P2:%.*]] = getelementptr i8, ptr [[P1]], i64 [[TMP1]] |
| ; CHECK-NEXT: call void @foo(ptr [[P2]]) |
| ; CHECK-NEXT: ret void |
| ; |
| ; v0 = input[0]; |
| call void @foo(ptr %input) |
| |
| ; v1 = input[(long)s]; |
| %t = sext i32 %s to i64 |
| %p1 = getelementptr i32, ptr %input, i64 %t |
| call void @foo(ptr %p1) |
| |
| ; v2 = input[(long)(s * 2)]; |
| %s2 = shl nsw i32 %s, 1 |
| %t2 = sext i32 %s2 to i64 |
| %p2 = getelementptr i32, ptr %input, i64 %t2 |
| call void @foo(ptr %p2) |
| |
| ret void |
| } |
| |
| define void @slsr_gep_128bit_index(ptr %input, i128 %s) { |
| ; CHECK-LABEL: define void @slsr_gep_128bit_index( |
| ; CHECK-SAME: ptr [[INPUT:%.*]], i128 [[S:%.*]]) { |
| ; CHECK-NEXT: call void @foo(ptr [[INPUT]]) |
| ; CHECK-NEXT: [[S125:%.*]] = shl nsw i128 [[S]], 125 |
| ; CHECK-NEXT: [[P1:%.*]] = getelementptr inbounds i32, ptr [[INPUT]], i128 [[S125]] |
| ; CHECK-NEXT: call void @foo(ptr [[P1]]) |
| ; CHECK-NEXT: [[S126:%.*]] = shl nsw i128 [[S]], 126 |
| ; CHECK-NEXT: [[P2:%.*]] = getelementptr inbounds i32, ptr [[INPUT]], i128 [[S126]] |
| ; CHECK-NEXT: call void @foo(ptr [[P2]]) |
| ; CHECK-NEXT: ret void |
| ; |
| ; p0 = &input[0] |
| call void @foo(ptr %input) |
| |
| ; p1 = &input[s << 125] |
| %s125 = shl nsw i128 %s, 125 |
| %p1 = getelementptr inbounds i32, ptr %input, i128 %s125 |
| call void @foo(ptr %p1) |
| |
| ; p2 = &input[s << 126] |
| %s126 = shl nsw i128 %s, 126 |
| %p2 = getelementptr inbounds i32, ptr %input, i128 %s126 |
| call void @foo(ptr %p2) |
| |
| ret void |
| } |
| |
| define void @slsr_gep_32bit_pointer(ptr addrspace(1) %input, i64 %s) { |
| ; CHECK-LABEL: define void @slsr_gep_32bit_pointer( |
| ; CHECK-SAME: ptr addrspace(1) [[INPUT:%.*]], i64 [[S:%.*]]) { |
| ; CHECK-NEXT: [[P1:%.*]] = getelementptr inbounds i32, ptr addrspace(1) [[INPUT]], i64 [[S]] |
| ; CHECK-NEXT: call void @baz(ptr addrspace(1) [[P1]]) |
| ; CHECK-NEXT: [[S2:%.*]] = mul nsw i64 [[S]], 2 |
| ; CHECK-NEXT: [[P2:%.*]] = getelementptr inbounds i32, ptr addrspace(1) [[INPUT]], i64 [[S2]] |
| ; CHECK-NEXT: call void @baz(ptr addrspace(1) [[P2]]) |
| ; CHECK-NEXT: ret void |
| ; |
| ; p1 = &input[s] |
| %p1 = getelementptr inbounds i32, ptr addrspace(1) %input, i64 %s |
| call void @baz(ptr addrspace(1) %p1) |
| |
| ; p2 = &input[s * 2] |
| %s2 = mul nsw i64 %s, 2 |
| %p2 = getelementptr inbounds i32, ptr addrspace(1) %input, i64 %s2 |
| ; %s2 is wider than the pointer size of addrspace(1), so do not factor it. |
| call void @baz(ptr addrspace(1) %p2) |
| |
| ret void |
| } |
| |
| define void @slsr_gep_fat_pointer(ptr addrspace(2) %input, i32 %s) { |
| ; p1 = &input[s] |
| ; CHECK-LABEL: define void @slsr_gep_fat_pointer( |
| ; CHECK-SAME: ptr addrspace(2) [[INPUT:%.*]], i32 [[S:%.*]]) { |
| ; CHECK-NEXT: [[P1:%.*]] = getelementptr inbounds i32, ptr addrspace(2) [[INPUT]], i32 [[S]] |
| ; CHECK-NEXT: call void @baz2(ptr addrspace(2) [[P1]]) |
| ; CHECK-NEXT: [[TMP1:%.*]] = shl i32 [[S]], 2 |
| ; CHECK-NEXT: [[P2:%.*]] = getelementptr inbounds i8, ptr addrspace(2) [[P1]], i32 [[TMP1]] |
| ; CHECK-NEXT: call void @baz2(ptr addrspace(2) [[P2]]) |
| ; CHECK-NEXT: ret void |
| ; |
| %p1 = getelementptr inbounds i32, ptr addrspace(2) %input, i32 %s |
| call void @baz2(ptr addrspace(2) %p1) |
| |
| ; p2 = &input[s * 2] |
| %s2 = mul nsw i32 %s, 2 |
| %p2 = getelementptr inbounds i32, ptr addrspace(2) %input, i32 %s2 |
| ; Use index bitwidth, not pointer size (i128) |
| call void @baz2(ptr addrspace(2) %p2) |
| |
| ret void |
| } |
| |
| |
| declare void @foo(ptr) |
| declare void @bar(ptr) |
| declare void @baz(ptr addrspace(1)) |
| declare void @baz2(ptr addrspace(2)) |