diff options
| author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-01 12:39:55 +0100 |
|---|---|---|
| committer | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-21 11:09:39 +0100 |
| commit | 845e23200d471e44f274940846e400d170b5ff37 (patch) | |
| tree | 28a01492bf11f0ff69309ead9bd8a1bad9e14cbb /ethosu/vela/nn_graph.py | |
| parent | d2b5510697e7789f5a416f9d80d3cb640eecc092 (diff) | |
| download | ethos-u-vela-845e23200d471e44f274940846e400d170b5ff37.tar.gz | |
MLBEDSW-3367 Add mypy to pre-commit
Add mypy to pre-commit and clean up all reported errors.
Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: If7dc869f5fecdb0e2db40f14e7d9db21aa33df71
Diffstat (limited to 'ethosu/vela/nn_graph.py')
| -rw-r--r-- | ethosu/vela/nn_graph.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index 8a2517de..671843f3 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -130,6 +130,7 @@ class CascadedPass: self.predecessors = [] self.successors = [] self.sram_used = 0 + self.time = 0 def __str__(self): return "<nng.CascadedPass strategy=%s x %s '%s', passes=%s, block_configs=%s>" % ( @@ -537,6 +538,9 @@ class Graph: self.total_npu_weights = 0 self.total_npu_encoded_weights = 0 self.weight_cache = None # See CompressedWeightCache + self.bandwidths = 0 + self.macs = 0 + self.cycles = 0 def get_root_subgraph(self): return self.subgraphs[0] |