Add ONNX operator version as trait of dialect

Xilinx · Jun 13, 2024 · 67cb3a6 · 67cb3a6
1 parent 4818661
commit 67cb3a6
Show file tree

Hide file tree

Showing 5 changed files with 279 additions and 240 deletions.
diff --git a/src/Dialect/ONNX/ONNX.td b/src/Dialect/ONNX/ONNX.td
@@ -232,6 +232,11 @@ def ONNXConstantOpFromDenseAttr: NativeCodeCall<
 class ONNX_Op<string mnemonic, list<Trait> traits = []> :
   Op<ONNX_Dialect, mnemonic, traits> ;
 
+// Trait to specify which operation set introduced a revision of an operator.
+// For multi-versioned operators, the version also appears in the operator's name.
+class OpVersionTrait<int version>
+  : ParamNativeOpTrait<"OpVersionTrait", !cast<string>(version)>;
+
 // The tablegen code onnxop.in is generated with gen_doc.py
 // clone and install onnx
 //    git clone --recursive https://github.com/onnx/onnx.git

diff --git a/src/Dialect/ONNX/ONNXOps.hpp b/src/Dialect/ONNX/ONNXOps.hpp
@@ -18,6 +18,7 @@
 #include "src/Dialect/ONNX/ONNXAttributes.hpp"
 #include "src/Dialect/ONNX/ONNXDialect.hpp"
 #include "src/Dialect/ONNX/ONNXOps/ShapeHelper.hpp"
+#include "src/Dialect/ONNX/ONNXTraits.hpp"
 #include "src/Dialect/ONNX/ONNXTypes.hpp"
 #include "src/Interface/HasOnnxSubgraphOpInterface.hpp"
 #include "src/Interface/ResultTypeInferenceOpInterface.hpp"