aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/graph_node.h
diff options
context:
space:
mode:
authorJerry Ge <jerry.ge@arm.com>2023-05-01 18:36:43 +0000
committerJerry Ge <jerry.ge@arm.com>2023-05-10 02:40:49 +0000
commit0bd4ec89d52cc1fd36e92dff2fb496b3550ee7f5 (patch)
treed2662a0e62aec08a648edf61da62ee789a481080 /reference_model/src/graph_node.h
parenta4d748b08accce06fab93e2d2b96e499b35ae89b (diff)
downloadreference_model-0bd4ec89d52cc1fd36e92dff2fb496b3550ee7f5.tar.gz
Refactor ref_model rank checking and add level check to argmax
Signed-off-by: Jerry Ge <jerry.ge@arm.com> Change-Id: Iad035b31d5e5e83040068e6311501490765bfff7
Diffstat (limited to 'reference_model/src/graph_node.h')
-rw-r--r--reference_model/src/graph_node.h18
1 files changed, 7 insertions, 11 deletions
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;
}