flang/test/HLFIR/get_length.fir

   1 // Test hlfir.get_length operation parse, verify (no errors), and unparse.
   2
   3 // RUN: fir-opt %s | fir-opt | FileCheck %s --check-prefix CHECK --check-prefix CHECK-ALL
   4
   5 // Test canonicalization.
   6 // RUN: fir-opt %s --canonicalize | FileCheck %s --check-prefix CHECK-CANON --check-prefix CHECK-ALL
   7
   8 func.func @fixed_len_scalar(%arg0: !hlfir.expr<!fir.char<1,10>>) -> index {
   9   %1 = hlfir.get_length %arg0 : (!hlfir.expr<!fir.char<1,10>>) -> index
  10   return %1 : index
  11 }
  12 // CHECK-ALL-LABEL:   func.func @fixed_len_scalar(
  13 // CHECK-ALL-SAME:      %[[VAL_0:.*]]: !hlfir.expr<!fir.char<1,10>>) -> index {
  14 // CHECK:               %[[VAL_1:.*]] = hlfir.get_length %[[VAL_0]] : (!hlfir.expr<!fir.char<1,10>>) -> index
  15 // CHECK-CANON:         %[[VAL_1:.*]] = arith.constant 10 : index
  16 // CHECK-ALL:           return %[[VAL_1]] : index
  17 // CHECK-ALL:         }
  18
  19 func.func @assumed_len_scalar(%arg0: !hlfir.expr<!fir.char<1,?>>) -> index {
  20   %1 = hlfir.get_length %arg0 : (!hlfir.expr<!fir.char<1,?>>) -> index
  21   return %1 : index
  22 }
  23 // CHECK-ALL-LABEL:   func.func @assumed_len_scalar(
  24 // CHECK-ALL-SAME:      %[[VAL_0:.*]]: !hlfir.expr<!fir.char<1,?>>) -> index {
  25 // CHECK-ALL:           %[[VAL_1:.*]] = hlfir.get_length %[[VAL_0]] : (!hlfir.expr<!fir.char<1,?>>) -> index
  26 // CHECK-ALL:           return %[[VAL_1]] : index
  27 // CHECK-ALL:         }
  28
  29 func.func @fixed_len_array(%arg0: !hlfir.expr<?x!fir.char<1,10>>) -> index {
  30   %1 = hlfir.get_length %arg0 : (!hlfir.expr<?x!fir.char<1,10>>) -> index
  31   return %1 : index
  32 }
  33 // CHECK-ALL-LABEL:   func.func @fixed_len_array(
  34 // CHECK-ALL-SAME:      %[[VAL_0:.*]]: !hlfir.expr<?x!fir.char<1,10>>) -> index {
  35 // CHECK:               %[[VAL_1:.*]] = hlfir.get_length %[[VAL_0]] : (!hlfir.expr<?x!fir.char<1,10>>) -> index
  36 // CHECK-CANON:         %[[VAL_1:.*]] = arith.constant 10 : index
  37 // CHECK-ALL:           return %[[VAL_1]] : index
  38 // CHECK-ALL:         }
  39
  40 func.func @assumed_len_array(%arg0: !hlfir.expr<?x!fir.char<1,?>>) -> index {
  41   %1 = hlfir.get_length %arg0 : (!hlfir.expr<?x!fir.char<1,?>>) -> index
  42   return %1 : index
  43 }
  44 // CHECK-ALL-LABEL:   func.func @assumed_len_array(
  45 // CHECK-ALL-SAME:      %[[VAL_0:.*]]: !hlfir.expr<?x!fir.char<1,?>>) -> index {
  46 // CHECK-ALL:           %[[VAL_1:.*]] = hlfir.get_length %[[VAL_0]] : (!hlfir.expr<?x!fir.char<1,?>>) -> index
  47 // CHECK-ALL:           return %[[VAL_1]] : index
  48 // CHECK-ALL:         }