129 lines · plain
1// RUN: tco %s | FileCheck %s2// RUN: %flang_fc1 -mmlir -disable-external-name-interop -emit-llvm %s -o -| FileCheck %s3 4 5// CHECK-LABEL: define void @_QPtest_callee(6func.func @_QPtest_callee(%arg0: !fir.box<!fir.array<?xi32>>) {7 return8}9 10// CHECK-LABEL: define void @_QPtest_slice()11func.func @_QPtest_slice() {12// CHECK: %[[a1:.*]] = alloca { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] }, align 813// CHECK: %[[a2:.*]] = alloca [20 x i32], i64 1, align 414// CHECK: %[[a3:.*]] = getelementptr [20 x i32], ptr %[[a2]], i64 0, i64 015// CHECK: %[[a4:.*]] = insertvalue { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] }16// CHECK: { ptr undef, i64 4, i32 20240719, i8 1, i8 9, i8 0, i8 0, [1 x [3 x i64]]17// CHECK: [i64 1, i64 5, i64 8]] }, ptr %[[a3]], 018// CHECK: store { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] } %[[a4]], ptr %[[a1]], align 819// CHECK: call void @_QPtest_callee(ptr %[[a1]])20 %c20 = arith.constant 20 : index21 %c1_i64 = arith.constant 1 : i6422 %c10_i64 = arith.constant 10 : i6423 %c2_i64 = arith.constant 2 : i6424 %0 = fir.alloca !fir.array<20xi32> {bindc_name = "x", uniq_name = "_QFtest_sliceEx"}25 %1 = fir.shape %c20 : (index) -> !fir.shape<1>26 %2 = fir.slice %c1_i64, %c10_i64, %c2_i64 : (i64, i64, i64) -> !fir.slice<1>27 %3 = fir.embox %0(%1) [%2] : (!fir.ref<!fir.array<20xi32>>, !fir.shape<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xi32>>28 fir.call @_QPtest_callee(%3) : (!fir.box<!fir.array<?xi32>>) -> ()29 return30}31 32// CHECK-LABEL: define void @_QPtest_dt_callee(33func.func @_QPtest_dt_callee(%arg0: !fir.box<!fir.array<?xi32>>) {34 return35}36 37// CHECK-LABEL: define void @_QPtest_dt_slice()38func.func @_QPtest_dt_slice() {39// CHECK: %[[a1:.*]] = alloca { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] }, align 840// CHECK: %[[a3:.*]] = alloca [20 x %_QFtest_dt_sliceTt], i64 1, align 841// CHECK: %[[a4:.*]] = getelementptr [20 x %_QFtest_dt_sliceTt], ptr %[[a3]], i64 0, i64 0, i32 042// CHECK: %[[a5:.*]] = insertvalue { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] }43// CHECK-SAME: { ptr undef, i64 4, i32 20240719, i8 1, i8 9, i8 0, i8 0, [1 x [3 x i64]]44// CHECK-SAME: [i64 1, i64 5, i64 1645// CHECK-SAME: , ptr %[[a4]], 046 47// CHECK: store { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] } %[[a5]], ptr %[[a1]], align 848// CHECK: call void @_QPtest_dt_callee(ptr %[[a1]])49 %c20 = arith.constant 20 : index50 %c1_i64 = arith.constant 1 : i6451 %c10_i64 = arith.constant 10 : i6452 %c2_i64 = arith.constant 2 : i6453 %0 = fir.alloca i32 {bindc_name = "v", uniq_name = "_QFtest_dt_sliceEv"}54 %1 = fir.alloca !fir.array<20x!fir.type<_QFtest_dt_sliceTt{i:i32,j:i32}>> {bindc_name = "x", uniq_name = "_QFtest_dt_sliceEx"}55 %2 = fir.field_index i, !fir.type<_QFtest_dt_sliceTt{i:i32,j:i32}>56 %3 = fir.shape %c20 : (index) -> !fir.shape<1>57 %4 = fir.slice %c1_i64, %c10_i64, %c2_i64 path %2 : (i64, i64, i64, !fir.field) -> !fir.slice<1>58 %5 = fir.embox %1(%3) [%4] : (!fir.ref<!fir.array<20x!fir.type<_QFtest_dt_sliceTt{i:i32,j:i32}>>>, !fir.shape<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xi32>>59 fir.call @_QPtest_dt_callee(%5) : (!fir.box<!fir.array<?xi32>>) -> ()60 return61}62 63func.func private @takesRank2CharBox(!fir.box<!fir.array<?x?x!fir.char<1,?>>>)64 65// CHECK-LABEL: define void @emboxSubstring(66// CHECK-SAME: ptr {{[^%]*}}%[[arg0:.*]])67func.func @emboxSubstring(%arg0: !fir.ref<!fir.array<2x3x!fir.char<1,4>>>) {68 %c2 = arith.constant 2 : index69 %c3 = arith.constant 3 : index70 %c1 = arith.constant 1 : index71 %c1_i64 = arith.constant 1 : i6472 %c2_i64 = arith.constant 2 : i6473 %0 = fir.shape %c2, %c3 : (index, index) -> !fir.shape<2>74 %1 = fir.slice %c1, %c2, %c1, %c1, %c3, %c1 substr %c1_i64, %c2_i64 : (index, index, index, index, index, index, i64, i64) -> !fir.slice<2>75 %2 = fir.embox %arg0(%0) [%1] : (!fir.ref<!fir.array<2x3x!fir.char<1,4>>>, !fir.shape<2>, !fir.slice<2>) -> !fir.box<!fir.array<?x?x!fir.char<1,?>>>76 // CHECK: %[[addr:.*]] = getelementptr [3 x [2 x [4 x i8]]], ptr %[[arg0]], i64 0, i64 0, i64 0, i64 177 // CHECK: insertvalue {[[descriptorType:.*]]} { ptr undef, i64 2, i32 20240719, i8 2, i8 40, i8 0, i8 078 // CHECK-SAME: [2 x [3 x i64]] [{{\[}}3 x i64] [i64 1, i64 2, i64 4], [3 x i64] [i64 1, i64 3, i64 8]] }79 // CHECK-SAME: ptr %[[addr]], 080 81 fir.call @takesRank2CharBox(%2) : (!fir.box<!fir.array<?x?x!fir.char<1,?>>>) -> ()82 return83}84 85func.func private @do_something(!fir.box<!fir.array<?xf32>>) -> ()86// CHECK: define void @fir_dev_issue_141687// CHECK-SAME: ptr {{[^%]*}}%[[base_addr:.*]], i64 %[[low:.*]], i64 %[[up:.*]], i64 %[[at:.*]])88func.func @fir_dev_issue_1416(%arg0: !fir.ref<!fir.array<40x?xf32>>, %low: index, %up: index, %at : index) {89 // Test fir.embox with a constant interior array shape.90 %c1 = arith.constant 1 : index91 %c40 = arith.constant 40 : index92 %0 = fir.undefined index93 %1 = fir.shape_shift %c1, %c40, %low, %up : (index, index, index, index) -> !fir.shapeshift<2>94 %2 = fir.slice %c1, %c40, %c1, %at, %0, %0 : (index, index, index, index, index, index) -> !fir.slice<2>95// CHECK: %[[diff:.*]] = sub i64 %[[at]], %[[low]]96// CHECK: %[[mul:.*]] = mul i64 %[[diff]], 197// CHECK: %[[offset:.*]] = add i64 %[[mul]], 098// CHECK: %[[addr:.*]] = getelementptr [40 x float], ptr %0, i64 %[[offset]], i64 099// CHECK: %[[box:.*]] = insertvalue { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] }100// CHECK-SAME: { ptr undef, i64 4, i32 20240719, i8 1, i8 27, i8 0, i8 0, [1 x [3 x i64]] [{{.*}} [i64 1, i64 40, i64 4]] }, ptr %[[addr]], 0101 %3 = fir.embox %arg0(%1) [%2] : (!fir.ref<!fir.array<40x?xf32>>, !fir.shapeshift<2>, !fir.slice<2>) -> !fir.box<!fir.array<?xf32>>102 fir.call @do_something(%3) : (!fir.box<!fir.array<?xf32>>) -> ()103 return104}105 106// CHECK-LABEL: define void @_QPtest_allocator1()107func.func @_QPtest_allocator1() {108 %c20 = arith.constant 20 : index109 %0 = fir.alloca !fir.array<20xi32> {bindc_name = "x", uniq_name = "_QFtest_sliceEx"}110 %1 = fir.shape %c20 : (index) -> !fir.shape<1>111 %3 = fir.embox %0(%1) {allocator_idx = 1 : i32} : (!fir.ref<!fir.array<20xi32>>, !fir.shape<1>) -> !fir.box<!fir.array<?xi32>>112 fir.call @_QPtest_callee(%3) : (!fir.box<!fir.array<?xi32>>) -> ()113 return114}115 116// %{{.*}} = insertvalue { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] } { ptr undef, i64 ptrtoint (ptr getelementptr (i32, ptr null, i32 1) to i64), i32 20240719, i8 1, i8 9, i8 0, i8 2, [1 x [3 x i64]] [[3 x i64] [i64 1, i64 20, i64 ptrtoint (ptr getelementptr (i32, ptr null, i32 1) to i64)]] }117 118// CHECK-LABEL: define void @_QPtest_allocator2()119func.func @_QPtest_allocator2() {120 %c20 = arith.constant 20 : index121 %0 = fir.alloca !fir.array<20xi32> {bindc_name = "x", uniq_name = "_QFtest_sliceEx"}122 %1 = fir.shape %c20 : (index) -> !fir.shape<1>123 %3 = fir.embox %0(%1) {allocator_idx = 3 : i32} : (!fir.ref<!fir.array<20xi32>>, !fir.shape<1>) -> !fir.box<!fir.array<?xi32>>124 fir.call @_QPtest_callee(%3) : (!fir.box<!fir.array<?xi32>>) -> ()125 return126}127 128// CHECK: %{{.*}} = insertvalue { ptr, i64, i32, i8, i8, i8, i8, [1 x [3 x i64]] } { ptr undef, i64 4, i32 20240719, i8 1, i8 9, i8 0, i8 6129