// Test hlfir.cshift operation lowering to fir runtime call // RUN: fir-opt %s -lower-hlfir-intrinsics | FileCheck %s // 1d boxed vector shift by scalar func.func @cshift1(%arg0: !fir.box> {fir.bindc_name = "a"}, %arg1: !fir.ref {fir.bindc_name = "sh"}) { %0:2 = hlfir.declare %arg0 {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) %1:2 = hlfir.declare %arg1 {uniq_name = "sh"} : (!fir.ref) -> (!fir.ref, !fir.ref) %2 = hlfir.cshift %0#0 %1#0 : (!fir.box>, !fir.ref) -> !hlfir.expr hlfir.assign %2 to %0#0 : !hlfir.expr, !fir.box> return } // CHECK-LABEL: func.func @cshift1( // CHECK-SAME: %[[VAL_0:.*]]: !fir.box> {fir.bindc_name = "a"}, // CHECK-SAME: %[[VAL_1:.*]]: !fir.ref {fir.bindc_name = "sh"}) { // CHECK: %[[VAL_2:.*]] = arith.constant true // CHECK: %[[VAL_4:.*]] = arith.constant 0 : index // CHECK: %[[VAL_5:.*]] = fir.alloca !fir.box>> // CHECK: %[[VAL_6:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) // CHECK: %[[VAL_7:.*]]:2 = hlfir.declare %[[VAL_1]] {uniq_name = "sh"} : (!fir.ref) -> (!fir.ref, !fir.ref) // CHECK: %[[VAL_8:.*]] = fir.zero_bits !fir.heap> // CHECK: %[[VAL_9:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1> // CHECK: %[[VAL_10:.*]] = fir.embox %[[VAL_8]](%[[VAL_9]]) : (!fir.heap>, !fir.shape<1>) -> !fir.box>> // CHECK: fir.store %[[VAL_10]] to %[[VAL_5]] : !fir.ref>>> // CHECK: %[[VAL_11:.*]] = fir.load %[[VAL_7]]#0 : !fir.ref // CHECK: %[[VAL_13:.*]] = fir.convert %[[VAL_5]] : (!fir.ref>>>) -> !fir.ref> // CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_6]]#1 : (!fir.box>) -> !fir.box // CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_11]] : (i32) -> i64 // CHECK: fir.call @_FortranACshiftVector(%[[VAL_13]], %[[VAL_14]], %[[VAL_15]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, i64, !fir.ref, i32) -> () // 2d boxed array shift by scalar func.func @cshift2(%arg0: !fir.box> {fir.bindc_name = "a"}, %arg1: i32 {fir.bindc_name = "sh"}) { %0:2 = hlfir.declare %arg0 {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) %2 = hlfir.cshift %0#0 %arg1 : (!fir.box>, i32) -> !hlfir.expr hlfir.assign %2 to %0#0 : !hlfir.expr, !fir.box> return } // CHECK-LABEL: func.func @cshift2( // CHECK-SAME: %[[VAL_0:.*]]: !fir.box> {fir.bindc_name = "a"}, // CHECK-SAME: %[[VAL_1:.*]]: i32 {fir.bindc_name = "sh"}) { // CHECK: %[[VAL_2:.*]] = arith.constant true // CHECK: %[[VAL_4:.*]] = arith.constant 1 : index // CHECK: %[[VAL_5:.*]] = arith.constant 0 : index // CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> // CHECK: %[[VAL_7:.*]] = fir.alloca i32 // CHECK: %[[VAL_8:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) // CHECK: fir.store %[[VAL_1]] to %[[VAL_7]] : !fir.ref // CHECK: %[[VAL_9:.*]] = fir.zero_bits !fir.heap> // CHECK: %[[VAL_10:.*]] = fir.shape %[[VAL_5]], %[[VAL_5]] : (index, index) -> !fir.shape<2> // CHECK: %[[VAL_11:.*]] = fir.embox %[[VAL_9]](%[[VAL_10]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> // CHECK: fir.store %[[VAL_11]] to %[[VAL_6]] : !fir.ref>>> // CHECK: %[[VAL_12:.*]] = fir.embox %[[VAL_7]] : (!fir.ref) -> !fir.box // CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> // CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_8]]#1 : (!fir.box>) -> !fir.box // CHECK: %[[VAL_16:.*]] = fir.convert %[[VAL_12]] : (!fir.box) -> !fir.box // CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_4]] : (index) -> i32 // CHECK: fir.call @_FortranACshift(%[[VAL_14]], %[[VAL_15]], %[[VAL_16]], %[[VAL_17]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, i32, !fir.ref, i32) -> () // 2d boxed array shift by boxed array func.func @cshift3(%arg0: !fir.box> {fir.bindc_name = "a"}, %arg1: !fir.box> {fir.bindc_name = "sh"}) { %0:2 = hlfir.declare %arg0 {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) %1:2 = hlfir.declare %arg1 {uniq_name = "sh"} : (!fir.box>) -> (!fir.box>, !fir.box>) %2 = hlfir.cshift %0#0 %1#0 : (!fir.box>, !fir.box>) -> !hlfir.expr hlfir.assign %2 to %0#0 : !hlfir.expr, !fir.box> return } // CHECK-LABEL: func.func @cshift3( // CHECK-SAME: %[[VAL_0:.*]]: !fir.box> {fir.bindc_name = "a"}, // CHECK-SAME: %[[VAL_1:.*]]: !fir.box> {fir.bindc_name = "sh"}) { // CHECK: %[[VAL_2:.*]] = arith.constant true // CHECK: %[[VAL_4:.*]] = arith.constant 1 : index // CHECK: %[[VAL_5:.*]] = arith.constant 0 : index // CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> // CHECK: %[[VAL_7:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) // CHECK: %[[VAL_8:.*]]:2 = hlfir.declare %[[VAL_1]] {uniq_name = "sh"} : (!fir.box>) -> (!fir.box>, !fir.box>) // CHECK: %[[VAL_9:.*]] = fir.zero_bits !fir.heap> // CHECK: %[[VAL_10:.*]] = fir.shape %[[VAL_5]], %[[VAL_5]] : (index, index) -> !fir.shape<2> // CHECK: %[[VAL_11:.*]] = fir.embox %[[VAL_9]](%[[VAL_10]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> // CHECK: fir.store %[[VAL_11]] to %[[VAL_6]] : !fir.ref>>> // CHECK: %[[VAL_13:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> // CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_7]]#1 : (!fir.box>) -> !fir.box // CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_8]]#1 : (!fir.box>) -> !fir.box // CHECK: %[[VAL_16:.*]] = fir.convert %[[VAL_4]] : (index) -> i32 // CHECK: fir.call @_FortranACshift(%[[VAL_13]], %[[VAL_14]], %[[VAL_15]], %[[VAL_16]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, i32, !fir.ref, i32) -> () // 2d boxed array shift by array expr func.func @cshift4(%arg0: !fir.box> {fir.bindc_name = "a"}, %arg1: !hlfir.expr {fir.bindc_name = "sh"}) { %0:2 = hlfir.declare %arg0 {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) %2 = hlfir.cshift %0#0 %arg1 : (!fir.box>, !hlfir.expr) -> !hlfir.expr hlfir.assign %2 to %0#0 : !hlfir.expr, !fir.box> return } // CHECK-LABEL: func.func @cshift4( // CHECK-SAME: %[[VAL_0:.*]]: !fir.box> {fir.bindc_name = "a"}, // CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr {fir.bindc_name = "sh"}) { // CHECK: %[[VAL_2:.*]] = arith.constant true // CHECK: %[[VAL_4:.*]] = arith.constant 1 : index // CHECK: %[[VAL_5:.*]] = arith.constant 0 : index // CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> // CHECK: %[[VAL_7:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) // CHECK: %[[VAL_8:.*]] = hlfir.shape_of %[[VAL_1]] : (!hlfir.expr) -> !fir.shape<1> // CHECK: %[[VAL_9:.*]]:3 = hlfir.associate %[[VAL_1]](%[[VAL_8]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<1>) -> (!fir.box>, !fir.ref>, i1) // CHECK: %[[VAL_10:.*]] = hlfir.get_extent %[[VAL_8]] {dim = 0 : index} : (!fir.shape<1>) -> index // CHECK: %[[VAL_11:.*]] = fir.zero_bits !fir.heap> // CHECK: %[[VAL_12:.*]] = fir.shape %[[VAL_5]], %[[VAL_5]] : (index, index) -> !fir.shape<2> // CHECK: %[[VAL_13:.*]] = fir.embox %[[VAL_11]](%[[VAL_12]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> // CHECK: fir.store %[[VAL_13]] to %[[VAL_6]] : !fir.ref>>> // CHECK: %[[VAL_14:.*]] = fir.shape %[[VAL_10]] : (index) -> !fir.shape<1> // CHECK: %[[VAL_15:.*]] = fir.embox %[[VAL_9]]#1(%[[VAL_14]]) : (!fir.ref>, !fir.shape<1>) -> !fir.box> // CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> // CHECK: %[[VAL_18:.*]] = fir.convert %[[VAL_7]]#1 : (!fir.box>) -> !fir.box // CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_15]] : (!fir.box>) -> !fir.box // CHECK: %[[VAL_20:.*]] = fir.convert %[[VAL_4]] : (index) -> i32 // CHECK: fir.call @_FortranACshift(%[[VAL_17]], %[[VAL_18]], %[[VAL_19]], %[[VAL_20]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, i32, !fir.ref, i32) -> () // 2d array expr shift by array expr func.func @cshift5(%arg0: !hlfir.expr {fir.bindc_name = "a"}, %arg1: !hlfir.expr {fir.bindc_name = "sh"}) { %2 = hlfir.cshift %arg0 %arg1 : (!hlfir.expr, !hlfir.expr) -> !hlfir.expr hlfir.destroy %2 : !hlfir.expr return } // CHECK-LABEL: func.func @cshift5( // CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr {fir.bindc_name = "a"}, // CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr {fir.bindc_name = "sh"}) { // CHECK: %[[VAL_2:.*]] = arith.constant true // CHECK: %[[VAL_4:.*]] = arith.constant 1 : index // CHECK: %[[VAL_5:.*]] = arith.constant 0 : index // CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> // CHECK: %[[VAL_7:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr) -> !fir.shape<2> // CHECK: %[[VAL_8:.*]]:3 = hlfir.associate %[[VAL_0]](%[[VAL_7]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<2>) -> (!fir.box>, !fir.ref>, i1) // CHECK: %[[VAL_9:.*]] = hlfir.get_extent %[[VAL_7]] {dim = 0 : index} : (!fir.shape<2>) -> index // CHECK: %[[VAL_10:.*]] = hlfir.get_extent %[[VAL_7]] {dim = 1 : index} : (!fir.shape<2>) -> index // CHECK: %[[VAL_11:.*]] = hlfir.shape_of %[[VAL_1]] : (!hlfir.expr) -> !fir.shape<1> // CHECK: %[[VAL_12:.*]]:3 = hlfir.associate %[[VAL_1]](%[[VAL_11]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<1>) -> (!fir.box>, !fir.ref>, i1) // CHECK: %[[VAL_13:.*]] = hlfir.get_extent %[[VAL_11]] {dim = 0 : index} : (!fir.shape<1>) -> index // CHECK: %[[VAL_14:.*]] = fir.shape %[[VAL_9]], %[[VAL_10]] : (index, index) -> !fir.shape<2> // CHECK: %[[VAL_15:.*]] = fir.embox %[[VAL_8]]#1(%[[VAL_14]]) : (!fir.ref>, !fir.shape<2>) -> !fir.box> // CHECK: %[[VAL_16:.*]] = fir.zero_bits !fir.heap> // CHECK: %[[VAL_17:.*]] = fir.shape %[[VAL_5]], %[[VAL_5]] : (index, index) -> !fir.shape<2> // CHECK: %[[VAL_18:.*]] = fir.embox %[[VAL_16]](%[[VAL_17]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> // CHECK: fir.store %[[VAL_18]] to %[[VAL_6]] : !fir.ref>>> // CHECK: %[[VAL_19:.*]] = fir.shape %[[VAL_13]] : (index) -> !fir.shape<1> // CHECK: %[[VAL_20:.*]] = fir.embox %[[VAL_12]]#1(%[[VAL_19]]) : (!fir.ref>, !fir.shape<1>) -> !fir.box> // CHECK: %[[VAL_22:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> // CHECK: %[[VAL_23:.*]] = fir.convert %[[VAL_15]] : (!fir.box>) -> !fir.box // CHECK: %[[VAL_24:.*]] = fir.convert %[[VAL_20]] : (!fir.box>) -> !fir.box // CHECK: %[[VAL_25:.*]] = fir.convert %[[VAL_4]] : (index) -> i32 // CHECK: fir.call @_FortranACshift(%[[VAL_22]], %[[VAL_23]], %[[VAL_24]], %[[VAL_25]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, i32, !fir.ref, i32) -> () // 2d array expr shift by array expr with explicit dim func.func @cshift6(%arg0: !hlfir.expr {fir.bindc_name = "a"}, %arg1: !hlfir.expr {fir.bindc_name = "sh"}, %dim : i16) { %2 = hlfir.cshift %arg0 %arg1 dim %dim : (!hlfir.expr, !hlfir.expr, i16) -> !hlfir.expr hlfir.destroy %2 : !hlfir.expr return } // CHECK-LABEL: func.func @cshift6( // CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr {fir.bindc_name = "a"}, // CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr {fir.bindc_name = "sh"}, // CHECK-SAME: %[[VAL_2:.*]]: i16) { // CHECK: %[[VAL_3:.*]] = arith.constant true // CHECK: %[[VAL_4:.*]] = arith.constant 1 : index // CHECK: %[[VAL_6:.*]] = arith.constant 0 : index // CHECK: %[[VAL_7:.*]] = fir.alloca !fir.box>> // CHECK: %[[VAL_8:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr) -> !fir.shape<2> // CHECK: %[[VAL_9:.*]]:3 = hlfir.associate %[[VAL_0]](%[[VAL_8]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<2>) -> (!fir.box>, !fir.ref>, i1) // CHECK: %[[VAL_10:.*]] = hlfir.get_extent %[[VAL_8]] {dim = 0 : index} : (!fir.shape<2>) -> index // CHECK: %[[VAL_11:.*]] = hlfir.get_extent %[[VAL_8]] {dim = 1 : index} : (!fir.shape<2>) -> index // CHECK: %[[VAL_12:.*]] = hlfir.shape_of %[[VAL_1]] : (!hlfir.expr) -> !fir.shape<1> // CHECK: %[[VAL_13:.*]]:3 = hlfir.associate %[[VAL_1]](%[[VAL_12]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<1>) -> (!fir.box>, !fir.ref>, i1) // CHECK: %[[VAL_14:.*]] = hlfir.get_extent %[[VAL_12]] {dim = 0 : index} : (!fir.shape<1>) -> index // CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_2]] : (i16) -> i32 // CHECK: %[[VAL_16:.*]] = fir.shape %[[VAL_10]], %[[VAL_11]] : (index, index) -> !fir.shape<2> // CHECK: %[[VAL_17:.*]] = fir.embox %[[VAL_9]]#1(%[[VAL_16]]) : (!fir.ref>, !fir.shape<2>) -> !fir.box> // CHECK: %[[VAL_18:.*]] = fir.zero_bits !fir.heap> // CHECK: %[[VAL_19:.*]] = fir.shape %[[VAL_6]], %[[VAL_6]] : (index, index) -> !fir.shape<2> // CHECK: %[[VAL_20:.*]] = fir.embox %[[VAL_18]](%[[VAL_19]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> // CHECK: fir.store %[[VAL_20]] to %[[VAL_7]] : !fir.ref>>> // CHECK: %[[VAL_21:.*]] = fir.shape %[[VAL_14]] : (index) -> !fir.shape<1> // CHECK: %[[VAL_22:.*]] = fir.embox %[[VAL_13]]#1(%[[VAL_21]]) : (!fir.ref>, !fir.shape<1>) -> !fir.box> // CHECK: %[[VAL_24:.*]] = fir.convert %[[VAL_7]] : (!fir.ref>>>) -> !fir.ref> // CHECK: %[[VAL_25:.*]] = fir.convert %[[VAL_17]] : (!fir.box>) -> !fir.box // CHECK: %[[VAL_26:.*]] = fir.convert %[[VAL_22]] : (!fir.box>) -> !fir.box // CHECK: fir.call @_FortranACshift(%[[VAL_24]], %[[VAL_25]], %[[VAL_26]], %[[VAL_15]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, i32, !fir.ref, i32) -> () // shift of polymorphic array func.func @cshift7(%arg0: !fir.ref>>>>, %arg1: !fir.ref) { %0 = fir.dummy_scope : !fir.dscope %1:2 = hlfir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs, uniq_name = "a"} : (!fir.ref>>>>, !fir.dscope) -> (!fir.ref>>>>, !fir.ref>>>>) %2:2 = hlfir.declare %arg1 dummy_scope %0 {uniq_name = "sh"} : (!fir.ref, !fir.dscope) -> (!fir.ref, !fir.ref) %c2_i32 = arith.constant 2 : i32 %3 = fir.load %1#0 : !fir.ref>>>> %4 = hlfir.cshift %3 %c2_i32 : (!fir.class>>>, i32) -> !hlfir.expr?> hlfir.assign %4 to %1#0 realloc : !hlfir.expr?>, !fir.ref>>>> hlfir.destroy %4 : !hlfir.expr?> return } // CHECK-LABEL: func.func @cshift7( // CHECK-SAME: %[[VAL_0:.*]]: !fir.ref>>>>, // CHECK-SAME: %[[VAL_1:.*]]: !fir.ref) { // CHECK: %[[VAL_2:.*]] = arith.constant true // CHECK: %[[VAL_4:.*]] = arith.constant 0 : index // CHECK: %[[VAL_5:.*]] = arith.constant 2 : i32 // CHECK: %[[VAL_6:.*]] = fir.alloca !fir.class>>> // CHECK: %[[VAL_7:.*]] = fir.alloca i32 // CHECK: %[[VAL_8:.*]] = fir.dummy_scope : !fir.dscope // CHECK: %[[VAL_9:.*]]:2 = hlfir.declare %[[VAL_0]] dummy_scope %[[VAL_8]] {fortran_attrs = #fir.var_attrs, uniq_name = "a"} : (!fir.ref>>>>, !fir.dscope) -> (!fir.ref>>>>, !fir.ref>>>>) // CHECK: %[[VAL_10:.*]]:2 = hlfir.declare %[[VAL_1]] dummy_scope %[[VAL_8]] {uniq_name = "sh"} : (!fir.ref, !fir.dscope) -> (!fir.ref, !fir.ref) // CHECK: %[[VAL_11:.*]] = fir.load %[[VAL_9]]#0 : !fir.ref>>>> // CHECK: fir.store %[[VAL_5]] to %[[VAL_7]] : !fir.ref // CHECK: %[[VAL_12:.*]] = fir.zero_bits !fir.heap>> // CHECK: %[[VAL_13:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1> // CHECK: %[[VAL_14:.*]] = fir.embox %[[VAL_12]](%[[VAL_13]]) source_box %[[VAL_11]] : (!fir.heap>>, !fir.shape<1>, !fir.class>>>) -> !fir.class>>> // CHECK: fir.store %[[VAL_14]] to %[[VAL_6]] : !fir.ref>>>> // CHECK: %[[VAL_15:.*]] = fir.load %[[VAL_7]] : !fir.ref // CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>>) -> !fir.ref> // CHECK: %[[VAL_18:.*]] = fir.convert %[[VAL_11]] : (!fir.class>>>) -> !fir.box // CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_15]] : (i32) -> i64 // CHECK: fir.call @_FortranACshiftVector(%[[VAL_17]], %[[VAL_18]], %[[VAL_19]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, i64, !fir.ref, i32) -> ()