Skip to content

Commit

Permalink
Fix swapped version numbers in version converter (onnx#5734)
Browse files Browse the repository at this point in the history
Fix swapped version numbers in version converter.

Signed-off-by: Ganesan Ramalingam <grama@microsoft.com>
Signed-off-by: isdanni <leedanni@gmail.com>
  • Loading branch information
gramalingam authored and isdanni committed Nov 2, 2023
1 parent 9928add commit 483c772
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions onnx/version_converter/convert.h
Expand Up @@ -591,15 +591,15 @@ class DefaultVersionConverter : public BaseVersionConverter {
TensorProto_DataType_FLOAT8E4M3FNUZ,
TensorProto_DataType_FLOAT8E5M2,
TensorProto_DataType_FLOAT8E5M2FNUZ};
registerAdapter(std::make_unique<TypeRestriction>("IsNaN", OpSetID(19), OpSetID(20), is_nan_13_unallowed_types));
registerAdapter(std::make_unique<TypeRestriction>("IsNaN", OpSetID(20), OpSetID(19), is_nan_13_unallowed_types));
const std::vector<TensorProto_DataType> is_inf_10_unallowed_types = {
TensorProto_DataType_FLOAT16,
TensorProto_DataType_BFLOAT16,
TensorProto_DataType_FLOAT8E4M3FN,
TensorProto_DataType_FLOAT8E4M3FNUZ,
TensorProto_DataType_FLOAT8E5M2,
TensorProto_DataType_FLOAT8E5M2FNUZ};
registerAdapter(std::make_unique<TypeRestriction>("IsInf", OpSetID(19), OpSetID(20), is_inf_10_unallowed_types));
registerAdapter(std::make_unique<TypeRestriction>("IsInf", OpSetID(20), OpSetID(19), is_inf_10_unallowed_types));
registerAdapter(std::make_unique<AxisInputToAttribute>("DFT", OpSetID(20), OpSetID(19), 2, -2));
const std::vector<TensorProto_DataType> reduce_min_max_18_unallowed_types = {TensorProto_DataType_BOOL};
registerAdapter(
Expand Down

0 comments on commit 483c772

Please sign in to comment.