flang/test/HLFIR/set_length.fir

   1 // Test hlfir.set_length operation parse, verify (no errors), and unparse.
   2 // RUN: fir-opt %s | fir-opt | FileCheck %s
   3
   4 func.func @test_cst_len(%str : !fir.boxchar<1>) {
   5   %c10 = arith.constant 10 : index
   6   %0 = hlfir.set_length %str len %c10 : (!fir.boxchar<1>, index) -> !hlfir.expr<!fir.char<1,10>>
   7   return
   8 }
   9 // CHECK-LABEL:   func.func @test_cst_len(
  10 // CHECK-SAME:    %[[VAL_0:.*]]: !fir.boxchar<1>) {
  11 // CHECK:  %[[VAL_1:.*]] = arith.constant 10 : index
  12 // CHECK:  %[[VAL_2:.*]] = hlfir.set_length %[[VAL_0]] len %[[VAL_1]] : (!fir.boxchar<1>, index) -> !hlfir.expr<!fir.char<1,10>>
  13
  14 func.func @test_dyn_len(%str : !fir.ref<!fir.char<1,10>>, %len : index) {
  15   %0 = hlfir.set_length %str len %len : (!fir.ref<!fir.char<1,10>>, index) -> !hlfir.expr<!fir.char<1,?>>
  16   return
  17 }
  18 // CHECK-LABEL:   func.func @test_dyn_len(
  19 // CHECK-SAME:    %[[VAL_0:.*]]: !fir.ref<!fir.char<1,10>>,
  20 // CHECK-SAME:    %[[VAL_1:.*]]: index) {
  21 // CHECK:  %[[VAL_2:.*]] = hlfir.set_length %[[VAL_0]] len %[[VAL_1]] : (!fir.ref<!fir.char<1,10>>, index) -> !hlfir.expr<!fir.char<1,?>>
  22
  23 func.func @test_from_expr(%str : !hlfir.expr<!fir.char<1,?>>, %len : index) {
  24   %0 = hlfir.set_length %str len %len : (!hlfir.expr<!fir.char<1,?>>, index) -> !hlfir.expr<!fir.char<1,?>>
  25   return
  26 }
  27 // CHECK-LABEL:   func.func @test_from_expr(
  28 // CHECK-SAME:    %[[VAL_0:.*]]: !hlfir.expr<!fir.char<1,?>>,
  29 // CHECK-SAME:    %[[VAL_1:.*]]: index) {
  30 // CHECK:  %[[VAL_2:.*]] = hlfir.set_length %[[VAL_0]] len %[[VAL_1]] : (!hlfir.expr<!fir.char<1,?>>, index) -> !hlfir.expr<!fir.char<1,?>>