-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathadd.onnx.mlir
More file actions
15 lines (15 loc) · 1.16 KB
/
add.onnx.mlir
File metadata and controls
15 lines (15 loc) · 1.16 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
module attributes {llvm.data_layout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-i128:128-f80:128-n8:16:32:64-S128", llvm.target_triple = "x86_64-unknown-linux-gnu", "onnx-mlir.symbol-postfix" = "add"} {
func.func @main_graph(%arg0: memref<3x2xf32> {onnx.name = "X1"}, %arg1: memref<3x2xf32> {onnx.name = "X2"}) -> (memref<3x2xf32> {onnx.name = "Y"}) attributes {llvm.emit_c_interface} {
%alloc = memref.alloc() {alignment = 16 : i64} : memref<3x2xf32>
affine.for %arg2 = 0 to 3 {
affine.for %arg3 = 0 to 2 {
%0 = affine.load %arg0[%arg2, %arg3] : memref<3x2xf32>
%1 = affine.load %arg1[%arg2, %arg3] : memref<3x2xf32>
%2 = arith.addf %0, %1 : f32
affine.store %2, %alloc[%arg2, %arg3] : memref<3x2xf32>
}
}
return %alloc : memref<3x2xf32>
}
"krnl.entry_point"() {func = @main_graph, numInputs = 2 : i32, numOutputs = 1 : i32, signature = "[ { \22type\22 : \22f32\22 , \22dims\22 : [3 , 2] , \22name\22 : \22X1\22 }\0A , { \22type\22 : \22f32\22 , \22dims\22 : [3 , 2] , \22name\22 : \22X2\22 }\0A\0A]\00@[ { \22type\22 : \22f32\22 , \22dims\22 : [3 , 2] , \22name\22 : \22Y\22 }\0A\0A]\00"} : () -> ()
}