83 lines · plain
1// Test hlfir.count operation parse, verify (no errors), and unparse2 3// RUN: fir-opt %s | fir-opt | FileCheck %s4 5// mask is an expression of known shape6func.func @count0(%arg0: !hlfir.expr<2x!fir.logical<4>>) {7 %count = hlfir.count %arg0 : (!hlfir.expr<2x!fir.logical<4>>) -> i328 return9}10// CHECK: func.func @count0(%[[ARRAY:.*]]: !hlfir.expr<2x!fir.logical<4>>) {11// CHECK-NEXT: %[[COUNT:.*]] = hlfir.count %[[ARRAY]] : (!hlfir.expr<2x!fir.logical<4>>) -> i3212// CHECK-NEXT: return13// CHECK-NEXT: }14 15// mask is an expression of assumed shape16func.func @count1(%arg0: !hlfir.expr<?x!fir.logical<4>>) {17 %count = hlfir.count %arg0 : (!hlfir.expr<?x!fir.logical<4>>) -> i3218 return19}20// CHECK: func.func @count1(%[[ARRAY:.*]]: !hlfir.expr<?x!fir.logical<4>>) {21// CHECK-NEXT: %[[COUNT:.*]] = hlfir.count %[[ARRAY]] : (!hlfir.expr<?x!fir.logical<4>>) -> i3222// CHECK-NEXT: return23// CHECK-NEXT: }24 25// mask is a boxed array26func.func @count2(%arg0: !fir.box<!fir.array<2x!fir.logical<4>>>) {27 %count = hlfir.count %arg0 : (!fir.box<!fir.array<2x!fir.logical<4>>>) -> i3228 return29}30// CHECK: func.func @count2(%[[ARRAY:.*]]: !fir.box<!fir.array<2x!fir.logical<4>>>) {31// CHECK-NEXT: %[[COUNT:.*]] = hlfir.count %[[ARRAY]] : (!fir.box<!fir.array<2x!fir.logical<4>>>) -> i3232// CHECK-NEXT: return33// CHECK-NEXT: }34 35// mask is an assumed shape boxed array36func.func @count3(%arg0: !fir.box<!fir.array<?x!fir.logical<4>>>) {37 %count = hlfir.count %arg0 : (!fir.box<!fir.array<?x!fir.logical<4>>>) -> i3238 return39}40// CHECK: func.func @count3(%[[ARRAY:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>) {41// CHECK-NEXT: %[[COUNT:.*]] = hlfir.count %[[ARRAY]] : (!fir.box<!fir.array<?x!fir.logical<4>>>) -> i3242// CHECK-NEXT: return43// CHECK-NEXT: }44 45// mask is a 2-dimensional array46func.func @count4(%arg0: !fir.box<!fir.array<?x?x!fir.logical<4>>>){47 %count = hlfir.count %arg0 : (!fir.box<!fir.array<?x?x!fir.logical<4>>>) -> i3248 return49}50// CHECK: func.func @count4(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?x!fir.logical<4>>>) {51// CHECK-NEXT: %[[COUNT:.*]] = hlfir.count %[[ARRAY]] : (!fir.box<!fir.array<?x?x!fir.logical<4>>>) -> i3252// CHECK-NEXT: return53// CHECK-NEXT: }54 55// mask and dim argument56func.func @count5(%arg0: !fir.box<!fir.array<?x!fir.logical<4>>>, %arg1: i32) {57 %count = hlfir.count %arg0 dim %arg1 : (!fir.box<!fir.array<?x!fir.logical<4>>>, i32) -> i3258 return59}60// CHECK: func.func @count5(%[[ARRAY:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>, %[[DIM:.*]]: i32) {61// CHECK-NEXT: %[[COUNT:.*]] = hlfir.count %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x!fir.logical<4>>>, i32) -> i3262// CHECK-NEXT: return63// CHECK-NEXT: }64 65// mask is a 2 dimensional array with dim66func.func @count6(%arg0: !fir.box<!fir.array<?x?x!fir.logical<4>>>, %arg1: i32) {67 %count = hlfir.count %arg0 dim %arg1 : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i32) -> i3268 return69}70// CHECK: func.func @count6(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?x!fir.logical<4>>>, %[[DIM:.*]]: i32) {71// CHECK-NEXT: %[[ANY:.*]] = hlfir.count %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i32) -> i3272// CHECK-NEXT: return73// CHECK-NEXT: }74 75// mask is of a different logical type76func.func @count7(%arg0: !fir.box<!fir.array<?x?x!fir.logical<8>>>, %arg1: i32) {77 %count = hlfir.count %arg0 dim %arg1 : (!fir.box<!fir.array<?x?x!fir.logical<8>>>, i32) -> i3278 return79}80// CHECK: func.func @count7(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?x!fir.logical<8>>>, %[[DIM:.*]]: i32) {81// CHECK-NEXT: %[[ANY:.*]] = hlfir.count %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?x!fir.logical<8>>>, i32) -> i3282// CHECK-NEXT: return83// CHECK-NEXT: }