onnx-mlir
onnx-mlir copied to clipboard
Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
Add one new option env var named 'menvVarName' in CompilerOptions and adjust the inputs of function 'ParseCommandLineOptions' in onnx-mlir.cpp. A python testing example named 'multi-threading-test.py' is provided under /docs/mnist_example/. Signed-off-by:...
This pull request: 1. Add option to the CMakeLists.txt so that Mhlo could be an optional library. 2. Add target as the parameter for decompose pass. 3. Add Mhlo lowering...
## Description When compiling following model with ONNX-MLIR, it crashes. But it could normally execute by onnxruntime. The model with error is as follows, check bug3.onnx in [o2m_bug3.zip](https://github.com/onnx/onnx-mlir/files/9335140/o2m_bug3.zip). ## Error...
Looking at the bertwquad12.onnx model (https://github.com/onnx/models/tree/main/text/machine_comprehension/bert-squad/model) When I do shape inference with the `?` dynamic values replace by 4, the final model is fully shaped with constant shapes ``` bash...
There are multiple impactful changes from mlir: - [LLVM] Update C++ standard to 17 (https://reviews.llvm.org/D130689) - [ADT] Deprecate Optional::{hasValue,getValue} (NFC) (https://reviews.llvm.org/D131349) - [mlir] Remove types from attributes (https://reviews.llvm.org/D130092) - [mlir]...
I ran into a problem when I was lowering an ONNX model with the onnx.SliceOp. The end input for the SliceOp is from a constant operator, but the op only...
- Implement incremental retrieval of llvm-project commit history to avoid the 5000/hour github REST API call limit - Use green (instead of orange) for the successful build color - Add...
As I implemented OptType in PR #1589 I noticed some issues with SeqType and sequence ops that are solved here: 1. SeqType required its element type to be a ShapedType,...
also added more checks to ONNXSplitToSequenceOp::verify() added lit tests for ONNXSplitToSequenceOp::verify() added SplitToSequence shape inference lit tests
If
Implemented verify() and inferShapes() for If op. Tested with lit tests. Also reordered the else and then branches so that ``` %0 = "onnx.If"(%arg0) ({ %1 = "onnx.Constant"() {value =...