From 0bd4ec89d52cc1fd36e92dff2fb496b3550ee7f5 Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Mon, 1 May 2023 18:36:43 +0000 Subject: Refactor ref_model rank checking and add level check to argmax Signed-off-by: Jerry Ge Change-Id: Iad035b31d5e5e83040068e6311501490765bfff7 --- reference_model/src/graph_node.h | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) (limited to 'reference_model/src/graph_node.h') diff --git a/reference_model/src/graph_node.h b/reference_model/src/graph_node.h index 3433192..aafc07f 100644 --- a/reference_model/src/graph_node.h +++ b/reference_model/src/graph_node.h @@ -270,20 +270,16 @@ protected: int setRequiredRank(const int min, const int max = -1) { - if (max == -1) - { - requiredRankMin = requiredRankMax = min; - } - else + requiredRankMin = min; + requiredRankMax = max; + + if (requiredRankMin >= 0 && requiredRankMax >= 0) { - requiredRankMin = min; - requiredRankMax = max; + ASSERT_MSG(requiredRankMin <= requiredRankMax, + "GraphNode::setRequiredRank: requiredRankMin %d must be <= requiredRankMax %d", requiredRankMin, + requiredRankMax); } - ASSERT_MSG(requiredRankMin <= requiredRankMax, - "GraphNode::setRequiredRank: requiredRankMin %d must be <= requiredRankMax %d", requiredRankMin, - requiredRankMax); - return 0; } -- cgit v1.2.1