flang/test/HLFIR/dot_product.fir

   1 // Test hlfir.dot_product operation parse, verify (no errors), and unparse
   2
   3 // RUN: fir-opt %s | fir-opt | FileCheck %s
   4
   5 // arguments are expressions of known shape
   6 func.func @dot_product0(%arg0: !hlfir.expr<2xi32>, %arg1: !hlfir.expr<2xi32>) {
   7   %res = hlfir.dot_product %arg0 %arg1 : (!hlfir.expr<2xi32>, !hlfir.expr<2xi32>) -> i32
   8   return
   9 }
  10 // CHECK-LABEL: func.func @dot_product0
  11 // CHECK:           %[[ARG0:.*]]: !hlfir.expr<2xi32>,
  12 // CHECK:           %[[ARG1:.*]]: !hlfir.expr<2xi32>
  13 // CHECK-NEXT:    %[[RES:.*]] = hlfir.dot_product %[[ARG0]] %[[ARG1]] : (!hlfir.expr<2xi32>, !hlfir.expr<2xi32>) -> i32
  14 // CHECK-NEXT:    return
  15 // CHECK-NEXT:  }
  16
  17 // arguments are expressions of assumed shape
  18 func.func @dot_product1(%arg0: !hlfir.expr<?xi32>, %arg1: !hlfir.expr<?xi32>) {
  19   %res = hlfir.dot_product %arg0 %arg1 : (!hlfir.expr<?xi32>, !hlfir.expr<?xi32>) -> i32
  20   return
  21 }
  22 // CHECK-LABEL: func.func @dot_product1
  23 // CHECK:           %[[ARG0:.*]]: !hlfir.expr<?xi32>,
  24 // CHECK:           %[[ARG1:.*]]: !hlfir.expr<?xi32>
  25 // CHECK-NEXT:    %[[RES:.*]] = hlfir.dot_product %[[ARG0]] %[[ARG1]] : (!hlfir.expr<?xi32>, !hlfir.expr<?xi32>) -> i32
  26 // CHECK-NEXT:    return
  27 // CHECK-NEXT:  }
  28
  29 // arguments are boxed arrays
  30 func.func @dot_product2(%arg0: !fir.box<!fir.array<2xi32>>, %arg1: !fir.box<!fir.array<2xi32>>) {
  31   %res = hlfir.dot_product %arg0 %arg1 : (!fir.box<!fir.array<2xi32>>, !fir.box<!fir.array<2xi32>>) -> i32
  32   return
  33 }
  34 // CHECK-LABEL: func.func @dot_product2
  35 // CHECK:           %[[ARG0:.*]]: !fir.box<!fir.array<2xi32>>,
  36 // CHECK:           %[[ARG1:.*]]: !fir.box<!fir.array<2xi32>>
  37 // CHECK-NEXT:    %[[RES:.*]] = hlfir.dot_product %[[ARG0]] %[[ARG1]] : (!fir.box<!fir.array<2xi32>>, !fir.box<!fir.array<2xi32>>) -> i32
  38 // CHECK-NEXT:    return
  39 // CHECK-NEXT:  }
  40
  41 // arguments are logical
  42 func.func @dot_product3(%arg0: !fir.box<!fir.array<2x!fir.logical<4>>>, %arg1: !fir.box<!fir.array<2x!fir.logical<4>>>) {
  43   %res = hlfir.dot_product %arg0 %arg1 : (!fir.box<!fir.array<2x!fir.logical<4>>>, !fir.box<!fir.array<2x!fir.logical<4>>>) -> !fir.logical<4>
  44   return
  45 }
  46 // CHECK-LABEL: func.func @dot_product3
  47 // CHECK:           %[[ARG0:.*]]: !fir.box<!fir.array<2x!fir.logical<4>>>,
  48 // CHECK:           %[[ARG1:.*]]: !fir.box<!fir.array<2x!fir.logical<4>>>
  49 // CHECK-NEXT:    %[[RES:.*]] = hlfir.dot_product %[[ARG0]] %[[ARG1]] : (!fir.box<!fir.array<2x!fir.logical<4>>>, !fir.box<!fir.array<2x!fir.logical<4>>>) -> !fir.logical<4>
  50 // CHECK-NEXT:    return
  51 // CHECK-NEXT:  }