aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tensor_allocation.py
diff options
context:
space:
mode:
authorJonas Ohlsson <jonas.ohlsson@arm.com>2022-03-01 12:39:55 +0100
committerJonas Ohlsson <jonas.ohlsson@arm.com>2022-03-21 11:09:39 +0100
commit845e23200d471e44f274940846e400d170b5ff37 (patch)
tree28a01492bf11f0ff69309ead9bd8a1bad9e14cbb /ethosu/vela/tensor_allocation.py
parentd2b5510697e7789f5a416f9d80d3cb640eecc092 (diff)
downloadethos-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/tensor_allocation.py')
-rw-r--r--ethosu/vela/tensor_allocation.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tensor_allocation.py b/ethosu/vela/tensor_allocation.py
index c82140c5..c8b5129d 100644
--- a/ethosu/vela/tensor_allocation.py
+++ b/ethosu/vela/tensor_allocation.py
@@ -91,7 +91,7 @@ def verify_allocation(live_ranges: LiveRangeGraph, alignment: int):
verify_alignment(live_ranges, alignment)
nr_time_slots = 1 + max(lr.end_time for lr in live_ranges.lrs)
# Contains active live ranges at each timestamp
- lrs_at_time = [[] for i in range(nr_time_slots)]
+ lrs_at_time: List[List[LiveRange]] = [[] for i in range(nr_time_slots)]
for lr in live_ranges.lrs:
for t in range(lr.start_time, lr.end_time + 1):
lrs_at_time[t].append(lr)