mlir/test/Target/Cpp/stdops.mlir

   1 // RUN: mlir-translate -mlir-to-cpp %s | FileCheck %s -check-prefix=CPP-DEFAULT
   2 // RUN: mlir-translate -mlir-to-cpp -declare-variables-at-top %s | FileCheck %s -check-prefix=CPP-DECLTOP
   3
   4 func.func @std_constant() {
   5   %c0 = arith.constant 0 : i32
   6   %c1 = arith.constant 2 : index
   7   %c2 = arith.constant 2.0 : f32
   8   %c3 = arith.constant dense<0> : tensor<i32>
   9   %c4 = arith.constant dense<[0, 1]> : tensor<2xindex>
  10   %c5 = arith.constant dense<[[0.0, 1.0], [2.0, 3.0]]> : tensor<2x2xf32>
  11   return
  12 }
  13 // CPP-DEFAULT: void std_constant() {
  14 // CPP-DEFAULT-NEXT: int32_t [[V0:[^ ]*]] = 0;
  15 // CPP-DEFAULT-NEXT: size_t [[V1:[^ ]*]] = 2;
  16 // CPP-DEFAULT-NEXT: float [[V2:[^ ]*]] = (float)2.000000000e+00;
  17 // CPP-DEFAULT-NEXT: Tensor<int32_t> [[V3:[^ ]*]] = {0};
  18 // CPP-DEFAULT-NEXT: Tensor<size_t, 2> [[V4:[^ ]*]] = {0, 1};
  19 // CPP-DEFAULT-NEXT: Tensor<float, 2, 2> [[V5:[^ ]*]] = {(float)0.0e+00, (float)1.000000000e+00, (float)2.000000000e+00, (float)3.000000000e+00};
  20
  21 // CPP-DECLTOP: void std_constant() {
  22 // CPP-DECLTOP-NEXT: int32_t [[V0:[^ ]*]];
  23 // CPP-DECLTOP-NEXT: size_t [[V1:[^ ]*]];
  24 // CPP-DECLTOP-NEXT: float [[V2:[^ ]*]];
  25 // CPP-DECLTOP-NEXT: Tensor<int32_t> [[V3:[^ ]*]];
  26 // CPP-DECLTOP-NEXT: Tensor<size_t, 2> [[V4:[^ ]*]];
  27 // CPP-DECLTOP-NEXT: Tensor<float, 2, 2> [[V5:[^ ]*]];
  28 // CPP-DECLTOP-NEXT: [[V0]] = 0;
  29 // CPP-DECLTOP-NEXT: [[V1]] = 2;
  30 // CPP-DECLTOP-NEXT: [[V2]] = (float)2.000000000e+00;
  31 // CPP-DECLTOP-NEXT: [[V3]] = {0};
  32 // CPP-DECLTOP-NEXT: [[V4]] = {0, 1};
  33 // CPP-DECLTOP-NEXT: [[V5]] = {(float)0.0e+00, (float)1.000000000e+00, (float)2.000000000e+00, (float)3.000000000e+00};
  34
  35 func.func @std_call() {
  36   %0 = call @one_result () : () -> i32
  37   %1 = call @one_result () : () -> i32
  38   return
  39 }
  40 // CPP-DEFAULT: void std_call() {
  41 // CPP-DEFAULT-NEXT: int32_t [[V0:[^ ]*]] = one_result();
  42 // CPP-DEFAULT-NEXT: int32_t [[V1:[^ ]*]] = one_result();
  43
  44 // CPP-DECLTOP: void std_call() {
  45 // CPP-DECLTOP-NEXT: int32_t [[V0:[^ ]*]];
  46 // CPP-DECLTOP-NEXT: int32_t [[V1:[^ ]*]];
  47 // CPP-DECLTOP-NEXT: [[V0]] = one_result();
  48 // CPP-DECLTOP-NEXT: [[V1]] = one_result();
  49
  50
  51 func.func @std_call_two_results() {
  52   %c = arith.constant 0 : i8
  53   %0:2 = call @two_results () : () -> (i32, f32)
  54   %1:2 = call @two_results () : () -> (i32, f32)
  55   return
  56 }
  57 // CPP-DEFAULT: void std_call_two_results() {
  58 // CPP-DEFAULT-NEXT: int8_t  [[V0:[^ ]*]] = 0;
  59 // CPP-DEFAULT-NEXT: int32_t [[V1:[^ ]*]];
  60 // CPP-DEFAULT-NEXT: float [[V2:[^ ]*]];
  61 // CPP-DEFAULT-NEXT: std::tie([[V1]], [[V2]]) = two_results();
  62 // CPP-DEFAULT-NEXT: int32_t [[V3:[^ ]*]];
  63 // CPP-DEFAULT-NEXT: float [[V4:[^ ]*]];
  64 // CPP-DEFAULT-NEXT: std::tie([[V3]], [[V4]]) = two_results();
  65
  66 // CPP-DECLTOP: void std_call_two_results() {
  67 // CPP-DECLTOP-NEXT: int8_t [[V0:[^ ]*]];
  68 // CPP-DECLTOP-NEXT: int32_t [[V1:[^ ]*]];
  69 // CPP-DECLTOP-NEXT: float [[V2:[^ ]*]];
  70 // CPP-DECLTOP-NEXT: int32_t [[V3:[^ ]*]];
  71 // CPP-DECLTOP-NEXT: float [[V4:[^ ]*]];
  72 // CPP-DECLTOP-NEXT: [[V0]] = 0;
  73 // CPP-DECLTOP-NEXT: std::tie([[V1]], [[V2]]) = two_results();
  74 // CPP-DECLTOP-NEXT: std::tie([[V3]], [[V4]]) = two_results();
  75
  76
  77 func.func @one_result() -> i32 {
  78   %0 = arith.constant 0 : i32
  79   return %0 : i32
  80 }
  81 // CPP-DEFAULT: int32_t one_result() {
  82 // CPP-DEFAULT-NEXT: int32_t [[V0:[^ ]*]] = 0;
  83 // CPP-DEFAULT-NEXT: return [[V0]];
  84
  85 // CPP-DECLTOP: int32_t one_result() {
  86 // CPP-DECLTOP-NEXT: int32_t [[V0:[^ ]*]];
  87 // CPP-DECLTOP-NEXT: [[V0]] = 0;
  88 // CPP-DECLTOP-NEXT: return [[V0]];
  89
  90
  91 func.func @two_results() -> (i32, f32) {
  92   %0 = arith.constant 0 : i32
  93   %1 = arith.constant 1.0 : f32
  94   return %0, %1 : i32, f32
  95 }
  96 // CPP-DEFAULT: std::tuple<int32_t, float> two_results() {
  97 // CPP-DEFAULT: int32_t [[V0:[^ ]*]] = 0;
  98 // CPP-DEFAULT: float [[V1:[^ ]*]] = (float)1.000000000e+00;
  99 // CPP-DEFAULT: return std::make_tuple([[V0]], [[V1]]);
 100
 101 // CPP-DECLTOP: std::tuple<int32_t, float> two_results() {
 102 // CPP-DECLTOP: int32_t [[V0:[^ ]*]];
 103 // CPP-DECLTOP: float [[V1:[^ ]*]];
 104 // CPP-DECLTOP: [[V0]] = 0;
 105 // CPP-DECLTOP: [[V1]] = (float)1.000000000e+00;
 106 // CPP-DECLTOP: return std::make_tuple([[V0]], [[V1]]);
 107
 108
 109 func.func @single_return_statement(%arg0 : i32) -> i32 {
 110   return %arg0 : i32
 111 }
 112 // CPP-DEFAULT: int32_t single_return_statement(int32_t [[V0:[^ ]*]]) {
 113 // CPP-DEFAULT-NEXT: return [[V0]];
 114
 115 // CPP-DECLTOP: int32_t single_return_statement(int32_t [[V0:[^ ]*]]) {
 116 // CPP-DECLTOP-NEXT: return [[V0]];