basic.mlir 992 字节
Newer Older
1
// RUN: infrtexec -i %s | FileCheck %s
Y
Yan Chunwei 已提交
2 3
// CHECK-LABEL: @basic_f32
func @basic_f32() -> f32 {
4 5 6
  %v0 = Infrt.constant.f32 1.0
  %v1 = Infrt.constant.f32 2.0
  %value = "Infrt.add.f32"(%v0, %v1) : (f32, f32) -> f32
Y
Yan Chunwei 已提交
7 8

  // CHECK-NEXT: 3
9
  "Infrt.print.f32"(%value) : (f32) -> ()
Y
Yan Chunwei 已提交
10

11
  Infrt.return %value : f32
Y
Yan Chunwei 已提交
12 13 14 15 16
}

/// ================================================================
/// @caller call the other function @callee
func @callee.add.f32(%x : f32, %y : f32, %y1 : f32) -> f32 {
17 18 19
  %z = "Infrt.add.f32"(%x, %y) : (f32, f32) -> f32
  %z1 = "Infrt.add.f32"(%z, %y1) : (f32, f32) -> f32
  Infrt.return %z1 : f32
Y
Yan Chunwei 已提交
20 21 22 23
}

// CHECK-LABEL: @caller.add.f32
func @caller.add.f32() -> f32 {
24 25 26 27
  %x = Infrt.constant.f32 1.0
  %y = Infrt.constant.f32 2.0
  %y1 = Infrt.constant.f32 3.0
  %z = Infrt.call @callee.add.f32(%x, %y, %y1) : (f32, f32, f32) -> f32
Y
Yan Chunwei 已提交
28 29

  // CHECK-NEXT: 6
30 31
  "Infrt.print.f32"(%z) : (f32) -> ()
  Infrt.return %z : f32
Y
Yan Chunwei 已提交
32 33
}
/// <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<