43 lines · plain
1// Test hlfir.count operation lowering with different default integer kinds.2// RUN: fir-opt %s -lower-hlfir-intrinsics | FileCheck %s3 4module attributes {fir.defaultkind = "a1c4d8i8l4r4", fir.kindmap = ""} {5 func.func @test_i8(%arg0: !fir.box<!fir.array<?x?x!fir.logical<4>>> {fir.bindc_name = "x"}, %arg1: i64) -> !hlfir.expr<?xi64> {6 %4 = hlfir.count %arg0 dim %arg1 : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i64) -> !hlfir.expr<?xi64>7 return %4 : !hlfir.expr<?xi64>8 }9}10// CHECK-LABEL: func.func @test_i811// CHECK: %[[KIND:.*]] = arith.constant 8 : i3212// CHECK: fir.call @_FortranACountDim(%{{.*}}, %{{.*}}, %{{.*}}, %[[KIND]], %{{.*}}, %{{.*}}) : (!fir.ref<!fir.box<none>>, !fir.box<none>, i32, i32, !fir.ref<i8>, i32) -> ()13 14module attributes {fir.defaultkind = "a1c4d8i4l4r4", fir.kindmap = ""} {15 func.func @test_i4(%arg0: !fir.box<!fir.array<?x?x!fir.logical<4>>> {fir.bindc_name = "x"}, %arg1: i64) -> !hlfir.expr<?xi32> {16 %4 = hlfir.count %arg0 dim %arg1 : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i64) -> !hlfir.expr<?xi32>17 return %4 : !hlfir.expr<?xi32>18 }19}20// CHECK-LABEL: func.func @test_i421// CHECK: %[[KIND:.*]] = arith.constant 4 : i3222// CHECK: fir.call @_FortranACountDim(%{{.*}}, %{{.*}}, %{{.*}}, %[[KIND]], %{{.*}}, %{{.*}}) : (!fir.ref<!fir.box<none>>, !fir.box<none>, i32, i32, !fir.ref<i8>, i32) -> ()23 24module attributes {fir.defaultkind = "a1c4d8i2l4r4", fir.kindmap = ""} {25 func.func @test_i2(%arg0: !fir.box<!fir.array<?x?x!fir.logical<4>>> {fir.bindc_name = "x"}, %arg1: i64) -> !hlfir.expr<?xi16> {26 %4 = hlfir.count %arg0 dim %arg1 : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i64) -> !hlfir.expr<?xi16>27 return %4 : !hlfir.expr<?xi16>28 }29}30// CHECK-LABEL: func.func @test_i231// CHECK: %[[KIND:.*]] = arith.constant 2 : i3232// CHECK: fir.call @_FortranACountDim(%{{.*}}, %{{.*}}, %{{.*}}, %[[KIND]], %{{.*}}, %{{.*}}) : (!fir.ref<!fir.box<none>>, !fir.box<none>, i32, i32, !fir.ref<i8>, i32) -> ()33 34module attributes {fir.defaultkind = "a1c4d8i1l4r4", fir.kindmap = ""} {35 func.func @test_i1(%arg0: !fir.box<!fir.array<?x?x!fir.logical<4>>> {fir.bindc_name = "x"}, %arg1: i64) -> !hlfir.expr<?xi8> {36 %4 = hlfir.count %arg0 dim %arg1 : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i64) -> !hlfir.expr<?xi8>37 return %4 : !hlfir.expr<?xi8>38 }39}40// CHECK-LABEL: func.func @test_i141// CHECK: %[[KIND:.*]] = arith.constant 1 : i3242// CHECK: fir.call @_FortranACountDim(%{{.*}}, %{{.*}}, %{{.*}}, %[[KIND]], %{{.*}}, %{{.*}}) : (!fir.ref<!fir.box<none>>, !fir.box<none>, i32, i32, !fir.ref<i8>, i32) -> ()43