diff options
author | Diqing Zhong <diqing.zhong@arm.com> | 2021-11-15 13:53:10 +0100 |
---|---|---|
committer | Diqing Zhong <diqing.zhong@arm.com> | 2021-11-15 14:21:15 +0100 |
commit | e3d18b0a77caa20ccc0c32fbaba6ff1564e4476d (patch) | |
tree | d0fd4395066692f17a3602688676cb5e141cbd73 /ethosu/vela/rawdata_writer.py | |
parent | 399c4a2d77df791e5d988c51d7fb1824ac4f266f (diff) | |
download | ethos-u-vela-e3d18b0a77caa20ccc0c32fbaba6ff1564e4476d.tar.gz |
TOSA: Add ifm ofm elem size into raw output3.2.0.rc2
Change-Id: I645496536a6bddf2bd289a87be9d7cef11693954
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>
Diffstat (limited to 'ethosu/vela/rawdata_writer.py')
-rw-r--r-- | ethosu/vela/rawdata_writer.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/ethosu/vela/rawdata_writer.py b/ethosu/vela/rawdata_writer.py index 76765e60..247cfe80 100644 --- a/ethosu/vela/rawdata_writer.py +++ b/ethosu/vela/rawdata_writer.py @@ -37,9 +37,11 @@ def write_rawdata_output(nng, arch, filename): if custom_op: ifm_shapes = [] + ifm_elem_sizes = [] ifm_regions = [] ifm_offsets = [] ofm_shapes = [] + ofm_elem_sizes = [] ofm_regions = [] ofm_offsets = [] cmd_stream_tensor, weight_tensor, scratch_tensor, scratch_fast_tensor = custom_op.inputs[:4] @@ -50,10 +52,12 @@ def write_rawdata_output(nng, arch, filename): ifm_shapes.append(ifm.shape) ifm_regions.append(get_region(ifm.mem_type, arch)) ifm_offsets.append(ifm.address) + ifm_elem_sizes.append(ifm.element_size()) for ofm in custom_op.outputs: ofm_shapes.append(ofm.shape) ofm_regions.append(get_region(ofm.mem_type, arch)) ofm_offsets.append(ofm.address) + ofm_elem_sizes.append(ofm.element_size()) filename_sg = f"{filename}_sg{sg_idx}_vela.npz" np.savez( @@ -66,9 +70,11 @@ def write_rawdata_output(nng, arch, filename): scratch_fast_shape=scratch_fast_tensor.shape, scratch_fast_region=scratch_fast_region, input_shape=ifm_shapes, + input_elem_size=ifm_elem_sizes, input_region=ifm_regions, input_offset=ifm_offsets, output_shape=ofm_shapes, + output_elem_size=ofm_elem_sizes, output_region=ofm_regions, output_offset=ofm_offsets, ) |