getOpVersion, trait comments

Xilinx · Jun 12, 2024 · 4328b37 · 4328b37
1 parent 876e3a0
commit 4328b37
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/Dialect/ONNX/ONNX.td b/src/Dialect/ONNX/ONNX.td
@@ -232,8 +232,8 @@ def ONNXConstantOpFromDenseAttr: NativeCodeCall<
 class ONNX_Op<string mnemonic, list<Trait> traits = []> :
   Op<ONNX_Dialect, mnemonic, traits> ;
 
-// Trait to specify which operation set a given operator was introduced in.
-// For multi-versioned operators, this also appears in the operator's name.
+// 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)>;
 

diff --git a/src/Dialect/ONNX/ONNXTraits.hpp b/src/Dialect/ONNX/ONNXTraits.hpp
@@ -25,7 +25,7 @@ class OpVersionTrait {
   template <typename ConcreteType>
   class Impl : public OpTrait::TraitBase<ConcreteType, Impl> {
   public:
-    int getOpsetMinVersion() { return version; }
+    int getOpVersion() { return version; }
   };
 };