45608282e0
[MLIR] Add support for Relu ( #392 )
...
* Add support for Relu
* Add comments
2019-12-21 01:38:16 -05:00
82f5bfec9f
Update lower_frontend_to_krnl.cpp ( #391 )
2019-12-21 01:37:50 -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
05e16dafae
Use template to support lowering all binary onnx ops to kernel ir ( #387 )
2019-12-21 01:35:17 -05:00
7fb2f80dce
[MLIR] Add support for dealloc insertion ( #386 )
...
* Add support for dealloc op.
* Check dealloc for returned result not present.
2019-12-21 01:34:48 -05:00
b2a1103915
[MLIR] Refactor Krnl Dialect and Krnl Dialect Lowering ( #375 )
...
* Store bounds as affine map attributes & check in test cases with generic printer
* Upgrading MLIR
MLIR is outdated on Buildbot, rebuilding a newer version.
* work with new version of mlir
* check-in parser tests
* custom printer
* nit
* bug fix
* enable custom asm printer test
* enable custom asm printer test
* more consistent variable naming
* test max/min
* variable naming scheme change to MLIR style
* can lower krnl to llvm
* kernel -> llvm
* comments
* bug fix
* try fixing ci
* fix ci
* deactivate model test
* fix lit test
* nit
* fix z buildbot
2019-12-21 01:34:14 -05:00
652ce4b7d4
Add test for checking lowering of Add op to KRNL IR ( #385 )
...
* Add test for checking lowering of Add op to KRNL IR.
* Add test file.
2019-12-21 01:20:36 -05:00
b46f965715
clean up
2019-12-21 01:13:50 -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
TUNG LEDUC
Tian Jin
GHEORGHE-TEOD BERCEA
Tian Jin