Tags: onnx/onnx-mlir
Tags
[RunONNXModel.py] Add options to print out input/output signatures an… …d support big models (#2982) * Add options to print out input/output signatures and support big models Signed-off-by: Tung D. Le <tung@jp.ibm.com> --------- Signed-off-by: Tung D. Le <tung@jp.ibm.com>
[Cherry-pick] Fixing the location of DimAnalysis in onnx-to-zhigh pas… …s and some rules in zhigh-to-onnx pass (#2797) The onnx-to-zhigh pass has two phases: 1) converting multiple onnx ops into a single zhigh op, and 2) converting a single onnx op to a single zhigh op, where the second phase uses DimAnalysis (Patterns in the 1st phase at this moment does not use DimAnalysis) The problem is DimAnalysis is currently called before the 1st phase, which is not good because the 1st phase may change the IR so the information from DimAnalysis is obsoleted to the 2nd phase. Correct position for DimAnalysis would be just before the 2nd phase. Other than that, this PR changes slightly the rules in zhigh-to-onnx pass so that for binary ops, only one input (instead of two) that is from stick would be enough to trigger the rule to convert a zhigh op back to an onnx op. Resolves #2789 --------- (cherry picked from commit 80a63f2) Signed-off-by: Tung D. Le <tung@jp.ibm.com> Signed-off-by: Charles Volzka <cjvolzka@us.ibm.com> Co-authored-by: Tung D. Le <tung@jp.ibm.com>
Add bitwise op support (#2043) * Add e2e support for bitwise ops Signed-off-by: philass <plassen@groq.com> * Fix docs Signed-off-by: philass <plassen@groq.com> * Add lit tests Signed-off-by: philass <plassen@groq.com> --------- Signed-off-by: philass <plassen@groq.com>
PreviousNext