From edd3ee67ce2526aeb1089eae8e6759232e4cd5be Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 23 Sep 2022 19:54:19 +0800 Subject: [PATCH] fix(mgb): add error infomation for old version load new elemwise mode GitOrigin-RevId: d425425d2f7137db3dd3ea5ddc49866d04ea75f8 --- dnn/src/common/elemwise/opr_impl.cpp | 5 ++++- dnn/src/common/elemwise_multi_type/opr_impl.cpp | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/dnn/src/common/elemwise/opr_impl.cpp b/dnn/src/common/elemwise/opr_impl.cpp index ed994866..b5981b8c 100644 --- a/dnn/src/common/elemwise/opr_impl.cpp +++ b/dnn/src/common/elemwise/opr_impl.cpp @@ -141,7 +141,10 @@ const ModeTrait& ModeTrait::from_mode(Mode mode) { #pragma message "elemwise mode stripped" #endif } - + megdnn_assert( + static_cast(mode) < traits.size(), + "Invalid elemwise mode in this version. " + "Maybe this version is too old, and you may need to update."); auto&& ret = traits.at(static_cast(mode)); #if !MEGDNN_ELEMWISE_MODE_ENABLE_ALL megdnn_assert(ret.arity); diff --git a/dnn/src/common/elemwise_multi_type/opr_impl.cpp b/dnn/src/common/elemwise_multi_type/opr_impl.cpp index f5bc9ec1..01acea16 100644 --- a/dnn/src/common/elemwise_multi_type/opr_impl.cpp +++ b/dnn/src/common/elemwise_multi_type/opr_impl.cpp @@ -274,7 +274,10 @@ const ModeTrait& ModeTrait::from_mode(Mode mode) { SET(init_bool_unary_op, ISINF); #undef SET } - + megdnn_assert( + static_cast(mode) < traits.size(), + "Invalid elemwise multitype mode in this version. " + "Maybe this version is too old, and you may need to update."); return traits.at(static_cast(mode)); }