update onnx opset (#3050)

* update onnx opset Signed-off-by: Sunny Anand <sunnyanand.979@gmail.com> * address feedback Signed-off-by: Sunny Anand <sunnyanand.979@gmail.com> * reformat for clang formatting Signed-off-by: Sunny Anand <sunnyanand.979@gmail.com> * reformat for black formattter Signed-off-by: Sunny Anand <sunnyanand.979@gmail.com> --------- Signed-off-by: Sunny Anand <sunnyanand.979@gmail.com>
onnx · Jan 23, 2025 · 6fb96c5 · 6fb96c5
1 parent bd41f89
commit 6fb96c5
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 7 deletions.
diff --git a/docs/SupportedONNXOps-NNPA.md b/docs/SupportedONNXOps-NNPA.md
@@ -3,11 +3,11 @@
 
 # Supported ONNX Operation for Target *NNPA*.
 
-Onnx-mlir currently supports ONNX operations targeting up to opset 21. Limitations are listed when applicable. This documentation highlights the minimum and maximum opset versions that are fully supported by onnx-mlir and not the version changes.
+Onnx-mlir currently supports ONNX operations targeting up to opset 22. Limitations are listed when applicable. This documentation highlights the minimum and maximum opset versions that are fully supported by onnx-mlir and not the version changes.
 
 * Operations are defined by the [ONNX Standard](https://github.com/onnx/onnx/blob/main/docs/Operators.md).
 * **Supported Opsets** indicates the lowest and highest opset a model may have for onnx-mlir to support compiling a model with the operator.
-   * A * indicates onnx-mlir is compatible with the latest version of that operator available as of opset 21.
+   * A * indicates onnx-mlir is compatible with the latest version of that operator available as of opset 22.
    * A ^ indicates onnx-mlir is compatible with the latest level of the NNPA Architecture which is z16.
 
 

diff --git a/docs/SupportedONNXOps-cpu.md b/docs/SupportedONNXOps-cpu.md
@@ -3,11 +3,11 @@
 
 # Supported ONNX Operation for Target *cpu*.
 
-Onnx-mlir currently supports ONNX operations targeting up to opset 21. Limitations are listed when applicable. This documentation highlights the minimum and maximum opset versions that are fully supported by onnx-mlir and not the version changes.
+Onnx-mlir currently supports ONNX operations targeting up to opset 22. Limitations are listed when applicable. This documentation highlights the minimum and maximum opset versions that are fully supported by onnx-mlir and not the version changes.
 
 * Operations are defined by the [ONNX Standard](https://github.com/onnx/onnx/blob/main/docs/Operators.md).
 * **Supported Opsets** indicates the lowest and highest opset a model may have for onnx-mlir to support compiling a model with the operator.
-   * A * indicates onnx-mlir is compatible with the latest version of that operator available as of opset 21.
+   * A * indicates onnx-mlir is compatible with the latest version of that operator available as of opset 22.
 
 
 | Op |Supported Opsets (inclusive) |Limitations |Notes |

diff --git a/src/Dialect/ONNX/ONNXOps.hpp b/src/Dialect/ONNX/ONNXOps.hpp
@@ -26,7 +26,9 @@
 
 namespace mlir {
 // OpSet level supported by onnx-mlir
-static constexpr int CURRENT_ONNX_OPSET = 20;
+// To update all occurrence of the current ONNX opset, please grep
+// "CURRENT_ONNX_OPSET" and update all locations accordingly.
+static constexpr int CURRENT_ONNX_OPSET = 22;
 } // end namespace mlir
 
 #define GET_OP_CLASSES

diff --git a/utils/pre-onnx-mlir.py b/utils/pre-onnx-mlir.py
@@ -39,7 +39,8 @@
 
 # ==UPDATE_ONNX_VERSION_OPSET==
 # Look for tag above and update all references when upgrading the ONNX support within ONNX-MLIR.
-current_onnx_opset = 21
+# To update all occurrence of the current ONNX opset, please grep "CURRENT_ONNX_OPSET" and update all locations accordingly.
+current_onnx_opset = 22
 
 converted_model = version_converter.convert_version(original_model, current_onnx_opset)
 
@@ -58,4 +59,4 @@
             + ".onnx"
         )
         onnx.save(converted_model, outFile)
-        print("The converted model is aved to " + outFile)
+        print("The converted model is saved to " + outFile)