colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
onnx-mlir/test/mlir/onnx/onnx_canonicalization.mlir

30 lines
2.1 KiB
MLIR
Raw Normal View History

[MLIR] compartmentalize build script (#369) * compartmentalize build script, temporarily remove dependency of onnf_opt on helper.cpp * fix test includes * fix op directory include * compiler -> op * compiler test depends on boost system * fix function name * specify libcompiler dependencies * let cmake take care of transitive dependencies * remove unnecessary includes * use ONNF_SRC_ROOT and ONNF_BIN_ROOT * allow whole-archive linked libraries to be appended * [MLIR] Support filecheck (#371) * support lit+FileCheck * add lit into build script * format MLIR.cmake * format cmake * [MLIR] Remove input/output ops (#372) * remove input/output ops * get output tensor type from symbol table
2019-11-19 08:37:58 +08:00
// RUN: onnf-opt --canonicalize %s -split-input-file | FileCheck %s
[MLIR] Remove module from test (#379) * Remove module from test * Update onnx_canonicalization.mlir
2019-11-26 10:36:38 +08:00
func @test_matmul_add_simplification(%a0: tensor<10x10xf32>, %a1: tensor<10x10xf32>, %a2: tensor<10x10xf32>) -> tensor<10x10xf32> {
// CHECK-LABEL: test_matmul_add_simplification
Add attributes as operation parameters (#45) * add attributes of Op into parameters * fix rewrite rule for GemmOp with attributes * use I64Attr instead of I32Attr and modify test cases for the changes in attributes * add output name (prefixed with o_) to Op definition * update shape inference for the new attributes
2020-01-27 23:09:14 +08:00
// CHECK: %{{[0-9]+}} = "onnx.Gemm"(%{{.*}}, %{{.*}}, %{{.*}}) {alpha = 1.000000e+00 : f32, beta = 1.000000e+00 : f32, transA = 0 : i64, transB = 0 : i64} : (tensor<10x10xf32>, tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
[MLIR] Remove module from test (#379) * Remove module from test * Update onnx_canonicalization.mlir
2019-11-26 10:36:38 +08:00
%0 = "onnx.MatMul"(%a0, %a1) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
%1 = "onnx.Add"(%0, %a2) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
"std.return"(%1) : (tensor<10x10xf32>) -> ()
[MLIR] generate op from onnx document (#366) * generate op from onnx document * Restore FullGemm * update the op attribute for shape inference and canonicalizer * Update onnx_canonicalization.mlir
2019-11-19 10:08:21 +08:00
}
[MLIR] Canonicalization pattern for eliminating identity ops (#377) * Canonicalization pattern for eliminating identity ops * Add a test for the identity elimination rule * Remove frontend from test * Use CHECK-NEXT instead of CHECK
2019-11-21 10:57:13 +08:00
[MLIR] Add one more test case for MatMul-Add fusion (#380) * Add one more testcase for matmul-add fusion * Code format for identity elimination testcase
2019-11-26 10:49:48 +08:00
// onnx.MatMul ops with more than one result uses should not get fused
// CHECK-LABEL: func @test_sigmoid_add(%{{.*}}: tensor<10x10xf32>, %{{.*}}: tensor<10x10xf32>, %{{.*}}: tensor<10x10xf32>) -> tensor<10x10xf32>
func @test_sigmoid_add(%a0: tensor<10x10xf32>, %a1: tensor<10x10xf32>, %a2: tensor<10x10xf32>) -> tensor<10x10xf32> {
// CHECK: %{{[0-9]+}} = "onnx.MatMul"(%{{.*}}, %{{.*}}) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
%0 = "onnx.MatMul"(%a0, %a1) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
%1 = "onnx.Add"(%0, %a2) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
%2 = "onnx.Add"(%0, %a1) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
%3 = "onnx.Add"(%1, %2) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
"std.return"(%3) : (tensor<10x10xf32>) -> ()
}
// CHECK-LABEL: @test_identity_identity(%{{.*}}: tensor<10x10xf32>, %{{.*}}: tensor<10x10xf32>) -> tensor<10x10xf32>
[MLIR] Canonicalization pattern for eliminating identity ops (#377) * Canonicalization pattern for eliminating identity ops * Add a test for the identity elimination rule * Remove frontend from test * Use CHECK-NEXT instead of CHECK
2019-11-21 10:57:13 +08:00
func @test_identity_identity(%a0: tensor<10x10xf32>, %a1: tensor<10x10xf32>) -> tensor<10x10xf32> {
[MLIR] Add one more test case for MatMul-Add fusion (#380) * Add one more testcase for matmul-add fusion * Code format for identity elimination testcase
2019-11-26 10:49:48 +08:00
// CHECK-NEXT: %{{[0-9]+}} = "onnx.Add"(%{{.*}}, %{{.*}}) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
%0 = "onnx.Identity"(%a0) : (tensor<10x10xf32>) -> tensor<10x10xf32>
%1 = "onnx.Identity"(%a1) : (tensor<10x10xf32>) -> tensor<10x10xf32>
%2 = "onnx.Add"(%0, %1) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
"std.return"(%2) : (tensor<10x10xf32>) -> ()
[MLIR] Canonicalization pattern for eliminating identity ops (#377) * Canonicalization pattern for eliminating identity ops * Add a test for the identity elimination rule * Remove frontend from test * Use CHECK-NEXT instead of CHECK
2019-11-21 10:57:13 +08:00
}