45608282e0
[MLIR] Add support for Relu ( #392 )
...
* Add support for Relu
* Add comments
2019-12-21 01:38:16 -05:00
1c3176bf9f
[MLIR] Lower ONNX element-wise unary ops: Exp, Tanh, Sinh, Cosh, Sigmoid ( #389 )
...
* Lower ExpOp
* Lower ONNXTanhOp
* Lower Exp Tanh, Sinh, and Cosh
* Lower ONNX Sigmoid op
* Merge
* Specialize template lowerScalarOp
* Unify ONNXEWUnaryOpLowering and ONNXEWBinaryOpLowering
* Support multiple types
* Reformat the code
* Add test cases
* Reformat the code
* Change names
* Apply clang-format
* Update variable names
2019-12-21 01:37:29 -05:00
c3ef1d93ae
[MLIR] Lower ONNX element-wise binary ops: Mul, Div, Sub, And, Or, Xor ( #388 )
...
* Lower ONNX element-wise binary ops: Mul, Div, Sub, And, Or, Xor
* Edit gen_doc.py to avoid changes about AnyTypeOf<[AnyMemRef, AnyTensor]>
* Miss a space
* Add tests
* Shorten ONNXElementWiseBinaryOpLowering into ONNXEWBinaryOpLowering
* Move lowering patterns into runOnModule()
* Redundant space
2019-12-21 01:35:31 -05:00
b02652dd76
[MLIR] Lowering of frontend dialect to KRNL dialect ( #382 )
...
* Partial support for lowering operations to KRNL dialect.
* Attempt to lower to KRNL IR.
* Update file.
* Add lowering.
* Address comments. Fix alloc dynamic dimensions. Correctly link StandardOps.
* Temporarily remove deallocation of locally allocated tensors.
2019-12-21 01:11:14 -05:00
53ab014a1d
[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-12-21 00:47:22 -05:00
bee32e2041
Fix rebase errors. ( #378 )
2019-12-21 00:46:29 -05:00
701c7d7508
clean up
2019-12-21 00:41:40 -05:00
3f68c5420d
[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-12-21 00:40:40 -05:00
dc36fd416b
[MLIR] Fix shape inference and enable ONNF to read in MLIR files. ( #367 )
...
* Fix inference. Enable ONNF to read in MLIR files.
* Fix input of onnx or mlir models.
* Address comments.
2019-12-21 00:12:27 -05:00
63596e723f
[MLIR] Add optimization based on TableGen pattern ( #363 )
...
* Define pattern.
* Fix file names.
* Add canonicalizer optimization based on TableGen pattern.
* Remove custom builders.
* Enable canonicalization in ONNF and ONNF-OPT.
2019-12-21 00:04:51 -05:00
03be41f7df
[MLIR] Add shape inference pass ( #361 )
...
* Add shape inference pass.
* Format code.
* Enable shape inference pass.
* Rename new files to use ONNF convention.
* Use include path from src folder.
* Replace guards with pragma once.
* Format variable names. Shuffle headers.
* Fix comment.
* Fix comments.
2019-12-19 20:14:07 -05:00
958a2fbffc
ONNX Dialect creation and registration with MLIR ( #358 )
...
* Create and register ONNX Dialect with one Add operation.
* Fix file formatting.
* Change name from ONNX to SGIR.
* Use ONNX dialect. Change SGIR to frontend placeholder dialect.
* Add comments.
* Type clean-up.
2019-12-19 16:56:30 -05:00
TUNG LEDUC
GHEORGHE-TEOD BERCEA
Tian Jin
TONG CHEN