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/npu_performance.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/npu_performance.py')
-rw-r--r-- | ethosu/vela/npu_performance.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py index 34530ae8..8c4aee63 100644 --- a/ethosu/vela/npu_performance.py +++ b/ethosu/vela/npu_performance.py @@ -22,6 +22,7 @@ import copy from enum import auto from enum import IntEnum +from typing import Optional from typing import Set from uuid import UUID @@ -580,7 +581,7 @@ def update_summary_cycles(arch, bws, cycles): def estimate_full_op_performance( - arch, schedule: Schedule, op: SchedulerOperation, prev_op: SchedulerOperation, block_config + arch, schedule: Schedule, op: SchedulerOperation, prev_op: Optional[SchedulerOperation], block_config ): cycles_a = make_cycles_array() bws = make_bandwidth_array() |