87 lines · plain
1// Test hlfir.reshape operation parse, verify (no errors), and unparse2// RUN: fir-opt %s | fir-opt | FileCheck %s3 4// Operands are expressions of known shape5func.func @reshape1(%arg0: !hlfir.expr<42xi32>, %arg1: !hlfir.expr<1xi32>) -> !hlfir.expr<?xi32> {6 %0 = hlfir.reshape %arg0 %arg1 : (!hlfir.expr<42xi32>, !hlfir.expr<1xi32>) -> !hlfir.expr<?xi32>7 return %0 : !hlfir.expr<?xi32>8}9// CHECK-LABEL: func.func @reshape1(10// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<42xi32>,11// CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr<1xi32>) -> !hlfir.expr<?xi32> {12// CHECK: %[[VAL_2:.*]] = hlfir.reshape %[[VAL_0]] %[[VAL_1]] : (!hlfir.expr<42xi32>, !hlfir.expr<1xi32>) -> !hlfir.expr<?xi32>13// CHECK: return %[[VAL_2]] : !hlfir.expr<?xi32>14// CHECK: }15 16// Operands are expressions of assumed shape17func.func @reshape2(%arg0: !hlfir.expr<?xi32>, %arg1: !hlfir.expr<1xi32>) -> !hlfir.expr<4xi32> {18 %0 = hlfir.reshape %arg0 %arg1 pad %arg0 order %arg0 : (!hlfir.expr<?xi32>, !hlfir.expr<1xi32>, !hlfir.expr<?xi32>, !hlfir.expr<?xi32>) -> !hlfir.expr<4xi32>19 return %0 : !hlfir.expr<4xi32>20}21// CHECK-LABEL: func.func @reshape2(22// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<?xi32>,23// CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr<1xi32>) -> !hlfir.expr<4xi32> {24// CHECK: %[[VAL_2:.*]] = hlfir.reshape %[[VAL_0]] %[[VAL_1]] pad %[[VAL_0]] order %[[VAL_0]] : (!hlfir.expr<?xi32>, !hlfir.expr<1xi32>, !hlfir.expr<?xi32>, !hlfir.expr<?xi32>) -> !hlfir.expr<4xi32>25// CHECK: return %[[VAL_2]] : !hlfir.expr<4xi32>26// CHECK: }27 28// Operands are boxed array29func.func @reshape3(%arg0: !fir.box<!fir.array<42xi32>>, %arg1: !fir.box<!fir.array<2xi32>>) -> !hlfir.expr<?x?xi32> {30 %0 = hlfir.reshape %arg0 %arg1 : (!fir.box<!fir.array<42xi32>>, !fir.box<!fir.array<2xi32>>) -> !hlfir.expr<?x?xi32>31 return %0 : !hlfir.expr<?x?xi32>32}33// CHECK-LABEL: func.func @reshape3(34// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<42xi32>>,35// CHECK-SAME: %[[VAL_1:.*]]: !fir.box<!fir.array<2xi32>>) -> !hlfir.expr<?x?xi32> {36// CHECK: %[[VAL_2:.*]] = hlfir.reshape %[[VAL_0]] %[[VAL_1]] : (!fir.box<!fir.array<42xi32>>, !fir.box<!fir.array<2xi32>>) -> !hlfir.expr<?x?xi32>37// CHECK: return %[[VAL_2]] : !hlfir.expr<?x?xi32>38// CHECK: }39 40// Operands are assumed shape boxed arrays41func.func @reshape4(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<1xi32>>) -> !hlfir.expr<?xi32> {42 %0 = hlfir.reshape %arg0 %arg1 pad %arg0 order %arg0 : (!fir.box<!fir.array<?xi32>>, !fir.box<!fir.array<1xi32>>, !fir.box<!fir.array<?xi32>>, !fir.box<!fir.array<?xi32>>) -> !hlfir.expr<?xi32>43 return %0 : !hlfir.expr<?xi32>44}45// CHECK-LABEL: func.func @reshape4(46// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?xi32>>,47// CHECK-SAME: %[[VAL_1:.*]]: !fir.box<!fir.array<1xi32>>) -> !hlfir.expr<?xi32> {48// CHECK: %[[VAL_2:.*]] = hlfir.reshape %[[VAL_0]] %[[VAL_1]] pad %[[VAL_0]] order %[[VAL_0]] : (!fir.box<!fir.array<?xi32>>, !fir.box<!fir.array<1xi32>>, !fir.box<!fir.array<?xi32>>, !fir.box<!fir.array<?xi32>>) -> !hlfir.expr<?xi32>49// CHECK: return %[[VAL_2]] : !hlfir.expr<?xi32>50// CHECK: }51 52// Operands are ref<array<>> of known shape53func.func @reshape5(%arg0: !fir.ref<!fir.array<1xi32>>) -> !hlfir.expr<?xi32> {54 %0 = hlfir.reshape %arg0 %arg0 pad %arg0 order %arg0 : (!fir.ref<!fir.array<1xi32>>, !fir.ref<!fir.array<1xi32>>, !fir.ref<!fir.array<1xi32>>, !fir.ref<!fir.array<1xi32>>) -> !hlfir.expr<?xi32>55 return %0 : !hlfir.expr<?xi32>56}57// CHECK-LABEL: func.func @reshape5(58// CHECK-SAME: %[[VAL_0:.*]]: !fir.ref<!fir.array<1xi32>>) -> !hlfir.expr<?xi32> {59// CHECK: %[[VAL_1:.*]] = hlfir.reshape %[[VAL_0]] %[[VAL_0]] pad %[[VAL_0]] order %[[VAL_0]] : (!fir.ref<!fir.array<1xi32>>, !fir.ref<!fir.array<1xi32>>, !fir.ref<!fir.array<1xi32>>, !fir.ref<!fir.array<1xi32>>) -> !hlfir.expr<?xi32>60// CHECK: return %[[VAL_1]] : !hlfir.expr<?xi32>61// CHECK: }62 63// Polymorphic operands64func.func @reshape6(%arg0: !fir.class<!fir.heap<!fir.array<?x!fir.type<whatever>>>>, %arg1: !hlfir.expr<1xi32>) -> !hlfir.expr<?x!fir.type<whatever>?> {65 %0 = hlfir.reshape %arg0 %arg1 pad %arg0 : (!fir.class<!fir.heap<!fir.array<?x!fir.type<whatever>>>>, !hlfir.expr<1xi32>, !fir.class<!fir.heap<!fir.array<?x!fir.type<whatever>>>>) -> !hlfir.expr<?x!fir.type<whatever>?>66 return %0 : !hlfir.expr<?x!fir.type<whatever>?>67}68// CHECK-LABEL: func.func @reshape6(69// CHECK-SAME: %[[VAL_0:.*]]: !fir.class<!fir.heap<!fir.array<?x!fir.type<whatever>>>>,70// CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr<1xi32>) -> !hlfir.expr<?x!fir.type<whatever>?> {71// CHECK: %[[VAL_2:.*]] = hlfir.reshape %[[VAL_0]] %[[VAL_1]] pad %[[VAL_0]] : (!fir.class<!fir.heap<!fir.array<?x!fir.type<whatever>>>>, !hlfir.expr<1xi32>, !fir.class<!fir.heap<!fir.array<?x!fir.type<whatever>>>>) -> !hlfir.expr<?x!fir.type<whatever>?>72// CHECK: return %[[VAL_2]] : !hlfir.expr<?x!fir.type<whatever>?>73// CHECK: }74 75// Allow character LEN mismatch for ARRAY/PAD and the result76func.func @reshape7(%arg0: !hlfir.expr<1x!fir.char<2,2>>, %arg1: !hlfir.expr<1xi32>, %arg2: !hlfir.expr<1x!fir.char<2,1>>) -> !hlfir.expr<?x!fir.char<2,3>> {77 %0 = hlfir.reshape %arg0 %arg1 pad %arg2 : (!hlfir.expr<1x!fir.char<2,2>>, !hlfir.expr<1xi32>, !hlfir.expr<1x!fir.char<2,1>>) -> !hlfir.expr<?x!fir.char<2,3>>78 return %0 : !hlfir.expr<?x!fir.char<2,3>>79}80// CHECK-LABEL: func.func @reshape7(81// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<1x!fir.char<2,2>>,82// CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr<1xi32>,83// CHECK-SAME: %[[VAL_2:.*]]: !hlfir.expr<1x!fir.char<2>>) -> !hlfir.expr<?x!fir.char<2,3>> {84// CHECK: %[[VAL_3:.*]] = hlfir.reshape %[[VAL_0]] %[[VAL_1]] pad %[[VAL_2]] : (!hlfir.expr<1x!fir.char<2,2>>, !hlfir.expr<1xi32>, !hlfir.expr<1x!fir.char<2>>) -> !hlfir.expr<?x!fir.char<2,3>>85// CHECK: return %[[VAL_3]] : !hlfir.expr<?x!fir.char<2,3>>86// CHECK: }87