Update version of Black to 22.3.0
Update version of Black to 22.3.0 due to updated dependencies.
Updates to fix reported issues due to new version.
Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: I60056aae452093ce8dcea1f499ecced22b25eef1
diff --git a/ethosu/vela/tensor_allocation.py b/ethosu/vela/tensor_allocation.py
index c8b5129..ab65740 100644
--- a/ethosu/vela/tensor_allocation.py
+++ b/ethosu/vela/tensor_allocation.py
@@ -128,7 +128,12 @@
print("\n" + "#" * 80)
sg_placement = (
sg.placement.name
- if mem_type_set.intersection((MemType.Permanent_NPU, MemType.Permanent_CPU,))
+ if mem_type_set.intersection(
+ (
+ MemType.Permanent_NPU,
+ MemType.Permanent_CPU,
+ )
+ )
else "Cpu and Npu"
)
print(
@@ -141,7 +146,15 @@
min_mem_usage_for_alloc = max(memory_hist)
print("Start Time - End Time: Start Addr - End Addr: Tensor Size: Memory Usage: Tensor Purpose: Tensor Name")
for start_time, end_time, size, start_addr, end_addr, purpose, name in sorted(
- (lr.start_time, lr.end_time, lr.size, tens.address, tens.address + lr.size, tens.purpose, tens.name,)
+ (
+ lr.start_time,
+ lr.end_time,
+ lr.size,
+ tens.address,
+ tens.address + lr.size,
+ tens.purpose,
+ tens.name,
+ )
for tens, lr in lrs.ranges.items()
):
print(
@@ -184,7 +197,11 @@
):
# Allocates addresses to tensors, returns False if tensors could not be fit within max_size
lrs = live_range.extract_live_ranges_from_cascaded_passes(
- sg, mem_area, mem_type_set, lr_graph=lr_graph, cpu_tensor_alignment=cpu_tensor_alignment,
+ sg,
+ mem_area,
+ mem_type_set,
+ lr_graph=lr_graph,
+ cpu_tensor_alignment=cpu_tensor_alignment,
)
total_sz = 0
if lrs.ranges: