phi_pass.mlir 482 字节
Newer Older
1
// RUN: phi-ir-exec %s
2 3 4 5 6 7 8 9 10
// CHECK-LABEL: @ops
func @ops() {
  %a = pd.feed() {name="input0"} : !infrt.lod_tensor<?xf32,0>
  %b = pd.feed() {name="input1"} : !infrt.lod_tensor<?xf32,0>
  %d = pd.feed() {name="input3"} : !infrt.lod_tensor<3x4x9xf32, 0>
  %g = "pd.elementwise_add"(%a, %b) {axis=1:si32} : (!infrt.lod_tensor<?xf32,0>, !infrt.lod_tensor<?xf32>) -> tensor<?xf32>
  %h = "pd.abs"(%g):(tensor<?xf32>) -> tensor<?xf32>
  "pd.fetch"(%h) {name="output"} :(tensor<?xf32>)->()
}