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
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
0c36057404
[MLIR] Add a constraint for the rewriting rule of add_matmult to fullgemm ( #374 )
...
* Add a constraint for the rewriting rule of matmul_add to fullgemm
* Edit comments
2019-12-21 00:42:19 -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
GHEORGHE-TEOD BERCEA
Tian Jin
Tian Jin
TUNG LEDUC