Skip to content
Snippets Groups Projects
Commit 1aa8defc authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Planes are garanteed to exist

parent 60f51b7a
No related branches found
No related tags found
2 merge requests!318[Upd] release verision 0.5.0,!290[Add] support for auto-concatenation and Fix multiple adaptToBackend() issues
Pipeline #62640 passed
......@@ -144,19 +144,18 @@ def generate_optimized_memory_info(scheduler: aidge_core.Scheduler, stats_folder
})
else:
for out_id in range(node.get_nb_outputs()):
if node in mem_planes:
plane = mem_planes[node][out_id]
node_mem_info.append({
"size": plane.size,
"offset": plane.get_contiguous_offset(),
"stride": plane.stride,
"length": plane.length,
"count": plane.count,
"cont_offset": plane.get_contiguous_offset(),
"cont_size": plane.get_contiguous_size(),
"wrap_offset": plane.get_wrapped_offset(),
"wrap_size": plane.get_wrapped_size()
})
plane = mem_planes[node][out_id]
node_mem_info.append({
"size": plane.size,
"offset": plane.get_contiguous_offset(),
"stride": plane.stride,
"length": plane.length,
"count": plane.count,
"cont_offset": plane.get_contiguous_offset(),
"cont_size": plane.get_contiguous_size(),
"wrap_offset": plane.get_wrapped_offset(),
"wrap_size": plane.get_wrapped_size()
})
mem_info[node] = node_mem_info
return mem_size, mem_info
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment