111 lines · plain
1// Test hlfir.forall operation parse, verify (no errors), and unparse.2// RUN: fir-opt %s | fir-opt | FileCheck %s3 4func.func @forall_test(%x: !fir.box<!fir.array<?xf32>>, %x2: !fir.box<!fir.array<?x?xf32>>) {5 %c1 = arith.constant 1 : index6 %c10 = arith.constant 10 : index7 hlfir.forall lb {8 hlfir.yield %c1 : index9 } ub {10 hlfir.yield %c10 : index11 } (%i : index) {12 hlfir.region_assign {13 %res = fir.call @foo(%i) : (index) -> f3214 hlfir.yield %res : f3215 } to {16 %xi = hlfir.designate %x(%i) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>17 hlfir.yield %xi : !fir.ref<f32>18 }19 hlfir.forall lb {20 hlfir.yield %c1 : index21 } ub {22 hlfir.yield %c10 : index23 } (%j : index) {24 hlfir.region_assign {25 %jf = fir.convert %j : (index) -> f3226 hlfir.yield %jf : f3227 } to {28 %xij = hlfir.designate %x2(%i, %j) : (!fir.box<!fir.array<?x?xf32>>, index, index) -> !fir.ref<f32>29 hlfir.yield %xij : !fir.ref<f32>30 }31 }32 }33 return34}35// CHECK-LABEL: func.func @forall_test(36// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?xf32>>,37// CHECK-SAME: %[[VAL_1:.*]]: !fir.box<!fir.array<?x?xf32>>) {38// CHECK: %[[VAL_2:.*]] = arith.constant 1 : index39// CHECK: %[[VAL_3:.*]] = arith.constant 10 : index40// CHECK: hlfir.forall lb {41// CHECK: hlfir.yield %[[VAL_2]] : index42// CHECK: } ub {43// CHECK: hlfir.yield %[[VAL_3]] : index44// CHECK: } (%[[VAL_4:.*]]: index) {45// CHECK: hlfir.region_assign {46// CHECK: %[[VAL_5:.*]] = fir.call @foo(%[[VAL_4]]) : (index) -> f3247// CHECK: hlfir.yield %[[VAL_5]] : f3248// CHECK: } to {49// CHECK: %[[VAL_6:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_4]]) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>50// CHECK: hlfir.yield %[[VAL_6]] : !fir.ref<f32>51// CHECK: }52// CHECK: hlfir.forall lb {53// CHECK: hlfir.yield %[[VAL_2]] : index54// CHECK: } ub {55// CHECK: hlfir.yield %[[VAL_3]] : index56// CHECK: } (%[[VAL_7:.*]]: index) {57// CHECK: hlfir.region_assign {58// CHECK: %[[VAL_8:.*]] = fir.convert %[[VAL_7]] : (index) -> f3259// CHECK: hlfir.yield %[[VAL_8]] : f3260// CHECK: } to {61// CHECK: %[[VAL_9:.*]] = hlfir.designate %[[VAL_1]] (%[[VAL_4]], %[[VAL_7]]) : (!fir.box<!fir.array<?x?xf32>>, index, index) -> !fir.ref<f32>62// CHECK: hlfir.yield %[[VAL_9]] : !fir.ref<f32>63// CHECK: }64// CHECK: }65// CHECK: }66 67func.func @forall_test_step(%x : !fir.box<!fir.array<10xf32>>, %y: !fir.box<!fir.array<?xf32>>) {68 hlfir.forall lb {69 %c1 = arith.constant 1 : index70 hlfir.yield %c1 : index71 } ub {72 %c10 = arith.constant 10 : index73 hlfir.yield %c10 : index74 } step {75 %c2 = arith.constant 2 : index76 hlfir.yield %c2 : index77 } (%i : index) {78 hlfir.region_assign {79 %yi = hlfir.designate %y(%i) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>80 %val = fir.load %yi : !fir.ref<f32>81 hlfir.yield %val : f3282 } to {83 %xi = hlfir.designate %x(%i) : (!fir.box<!fir.array<10xf32>>, index) -> !fir.ref<f32>84 hlfir.yield %xi : !fir.ref<f32>85 }86 }87 return88}89// CHECK-LABEL: func.func @forall_test_step(90// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<10xf32>>,91// CHECK-SAME: %[[VAL_1:.*]]: !fir.box<!fir.array<?xf32>>) {92// CHECK: hlfir.forall lb {93// CHECK: %[[VAL_2:.*]] = arith.constant 1 : index94// CHECK: hlfir.yield %[[VAL_2]] : index95// CHECK: } ub {96// CHECK: %[[VAL_3:.*]] = arith.constant 10 : index97// CHECK: hlfir.yield %[[VAL_3]] : index98// CHECK: } step {99// CHECK: %[[VAL_4:.*]] = arith.constant 2 : index100// CHECK: hlfir.yield %[[VAL_4]] : index101// CHECK: } (%[[VAL_5:.*]]: index) {102// CHECK: hlfir.region_assign {103// CHECK: %[[VAL_6:.*]] = hlfir.designate %[[VAL_1]] (%[[VAL_5]]) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>104// CHECK: %[[VAL_7:.*]] = fir.load %[[VAL_6]] : !fir.ref<f32>105// CHECK: hlfir.yield %[[VAL_7]] : f32106// CHECK: } to {107// CHECK: %[[VAL_8:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_5]]) : (!fir.box<!fir.array<10xf32>>, index) -> !fir.ref<f32>108// CHECK: hlfir.yield %[[VAL_8]] : !fir.ref<f32>109// CHECK: }110// CHECK: }111