Skip to content

Commit

Permalink
[tools/mec] Test the number of operator and const nodes (Samsung#14417)
Browse files Browse the repository at this point in the history
It adds test cases for checking the number of operator and const nodes.

ONE-DCO-1.0-Signed-off-by: Jonghwa Lee <jonghwa3.lee@samsung.com>
  • Loading branch information
batcheu authored Dec 5, 2024
1 parent 81d6d2a commit f5395d8
Showing 1 changed file with 41 additions and 0 deletions.
41 changes: 41 additions & 0 deletions tools/model_explorer_circle/tests/test_circle_adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,3 +48,44 @@ def test_output_count(circle_adapter):
output_node = me_graph.nodes[-1]
assert output_node.label == 'GraphOutputs'
assert len(output_node.incomingEdges) == len(circle_graph.outputs)


def test_operator_count(circle_adapter):
"""Test if number of operators matches"""
circle_model = circle_adapter.model
circle_graph = circle_model.subgraphs[0]
me_graph = circle_adapter.graph

# Count operators in circle model
nr_operators = len(circle_graph.operators)

# Count operator nodes in model explorer graph
nr_opnodes = 0
for node in me_graph.nodes:
if node.label in ['GraphInputs', 'GraphOutputs', 'pseudo_const']:
continue
assert int(node.id) == nr_opnodes
nr_opnodes += 1

assert nr_operators == nr_opnodes


def test_const_tensor_count(circle_adapter):
"""Test if number of constant tensors matches"""
circle_model = circle_adapter.model
circle_graph = circle_model.subgraphs[0]
me_graph = circle_adapter.graph

# Count constant tensors in circle model
nr_const_tensors = len(circle_graph.tensors)
nr_const_tensors -= len(circle_graph.inputs)
nr_const_tensors -= sum([len(op.outputs) for op in circle_graph.operators])

# Count constant tensor nodes in model explorer graph
nr_pseudo_const = 0
for node in me_graph.nodes:
if node.label == 'pseudo_const':
nr_pseudo_const += 1
assert int(node.id) == (len(circle_graph.operators) + nr_pseudo_const)

assert nr_const_tensors == nr_pseudo_const

0 comments on commit f5395d8

Please sign in to comment.