flang/test/HLFIR/order_assignments/forall-codegen-fuse-assignments.fir

   1 // Test code generation of hlfir.forall when assignment fusing is enabled
   2 // and possible.
   3 // RUN: fir-opt %s --lower-hlfir-ordered-assignments=fuse-assignments=true | FileCheck %s --check-prefix=FUSE
   4 // RUN: fir-opt %s --lower-hlfir-ordered-assignments=fuse-assignments=false | FileCheck %s --check-prefix=NOFUSE
   5
   6 func.func @test_assignment_fusing(%x: !fir.ref<!fir.array<10xi32>>, %y : !fir.box<!fir.array<?xi32>>) {
   7   %c42 = arith.constant 42 : i32
   8   hlfir.forall lb {
   9     %c1 = arith.constant 1 : index
  10     hlfir.yield %c1 : index
  11   } ub {
  12     %c10 = arith.constant 10 : index
  13     hlfir.yield %c10 : index
  14   }  (%i: index) {
  15     hlfir.region_assign {
  16       hlfir.yield %c42 : i32
  17     } to {
  18       %2 = hlfir.designate %x (%i)  : (!fir.ref<!fir.array<10xi32>>, index) -> !fir.ref<i32>
  19       hlfir.yield %2 : !fir.ref<i32>
  20     }
  21     hlfir.region_assign {
  22       hlfir.yield %c42 : i32
  23     } to {
  24       %2 = hlfir.designate %y (%i)  : (!fir.box<!fir.array<?xi32>>, index) -> !fir.ref<i32>
  25       hlfir.yield %2 : !fir.ref<i32>
  26     }
  27   }
  28   return
  29 }
  30 // FUSE-LABEL:   func.func @test_assignment_fusing(
  31 // FUSE-SAME:                                      %[[VAL_0:.*]]: !fir.ref<!fir.array<10xi32>>,
  32 // FUSE-SAME:                                      %[[VAL_1:.*]]: !fir.box<!fir.array<?xi32>>) {
  33 // FUSE:           %[[VAL_2:.*]] = arith.constant 42 : i32
  34 // FUSE:           %[[VAL_3:.*]] = arith.constant 1 : index
  35 // FUSE:           %[[VAL_4:.*]] = arith.constant 10 : index
  36 // FUSE:           %[[VAL_5:.*]] = arith.constant 1 : index
  37 // FUSE:           fir.do_loop %[[VAL_6:.*]] = %[[VAL_3]] to %[[VAL_4]] step %[[VAL_5]] {
  38 // FUSE-NEXT:         %[[VAL_7:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_6]])  : (!fir.ref<!fir.array<10xi32>>, index) -> !fir.ref<i32>
  39 // FUSE-NEXT:         hlfir.assign %[[VAL_2]] to %[[VAL_7]] : i32, !fir.ref<i32>
  40 // FUSE-NEXT:         %[[VAL_8:.*]] = hlfir.designate %[[VAL_1]] (%[[VAL_6]])  : (!fir.box<!fir.array<?xi32>>, index) -> !fir.ref<i32>
  41 // FUSE-NEXT:         hlfir.assign %[[VAL_2]] to %[[VAL_8]] : i32, !fir.ref<i32>
  42 // FUSE-NEXT:      }
  43 // FUSE-NEXT:      return
  44
  45 // NOFUSE-LABEL: func.func @test_assignment_fusing(
  46 // NOFUSE:         fir.do_loop
  47 // NOFUSE:           hlfir.assign
  48 // NOFUSE:         }
  49 // NOFUSE:         fir.do_loop
  50 // NOFUSE:           hlfir.assign
  51 // NOFUSE:         }