52 lines · plain
1// Test code generation of hlfir.forall when assignment fusing is enabled2// and possible.3// RUN: fir-opt %s --lower-hlfir-ordered-assignments=fuse-assignments=true | FileCheck %s --check-prefix=FUSE4// RUN: fir-opt %s --lower-hlfir-ordered-assignments=fuse-assignments=false | FileCheck %s --check-prefix=NOFUSE5 6func.func @test_assignment_fusing(%x: !fir.ref<!fir.array<10xi32>>, %y : !fir.box<!fir.array<?xi32>>) {7 %c42 = arith.constant 42 : i328 hlfir.forall lb {9 %c1 = arith.constant 1 : index10 hlfir.yield %c1 : index11 } ub {12 %c10 = arith.constant 10 : index13 hlfir.yield %c10 : index14 } (%i: index) {15 hlfir.region_assign {16 hlfir.yield %c42 : i3217 } 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 : i3223 } 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 return29}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 : i3234// FUSE: %[[VAL_3:.*]] = arith.constant 1 : index35// FUSE: %[[VAL_4:.*]] = arith.constant 10 : index36// FUSE: %[[VAL_5:.*]] = arith.constant 1 : index37// 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: return44 45// NOFUSE-LABEL: func.func @test_assignment_fusing(46// NOFUSE: fir.do_loop47// NOFUSE: hlfir.assign48// NOFUSE: }49// NOFUSE: fir.do_loop50// NOFUSE: hlfir.assign51// NOFUSE: }52