diff options
author | erik.andersson@arm.com <erik.andersson@arm.com> | 2022-03-22 15:35:30 +0100 |
---|---|---|
committer | erik.andersson@arm.com <erik.andersson@arm.com> | 2022-07-11 11:27:47 +0200 |
commit | 6b2a0b4a64d01c8b038050a87c29f38a4909515c (patch) | |
tree | 0bd213a78debbfbe8465fcbf1c87eadd1f44fc2f /ethosu/vela/high_level_command_stream_generator.py | |
parent | 25f48dd70aebeecd490de71eed3d4f7fbad1b121 (diff) | |
download | ethos-u-vela-6b2a0b4a64d01c8b038050a87c29f38a4909515c.tar.gz |
MLBEDSW-6261: Elementwise cascading
Enabled elementwise cascading for binary/single variable IFM operators.
Signed-off-by: erik.andersson@arm.com <erik.andersson@arm.com>
Change-Id: I1c0867875fdc5c4980224fb570185c11e719d5cd
Diffstat (limited to 'ethosu/vela/high_level_command_stream_generator.py')
-rw-r--r-- | ethosu/vela/high_level_command_stream_generator.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/ethosu/vela/high_level_command_stream_generator.py b/ethosu/vela/high_level_command_stream_generator.py index 81c0d5b4..95068081 100644 --- a/ethosu/vela/high_level_command_stream_generator.py +++ b/ethosu/vela/high_level_command_stream_generator.py @@ -134,7 +134,6 @@ def generate_high_level_commands_for_sched_op(sched_op, schedule): ifm_present = Box([0, 0, 0, 0], [0, 0, 0, 0]) producer_op = sched_op.ifm.connection.producers[0] prev_cmd_gen = generate_high_level_commands_for_sched_op(producer_op, schedule) - ofm_step = op_info.stripe for start_height in range(ofm_start.height, ofm_end.height, ofm_step.height): end_height = min(start_height + ofm_step.height, ofm_end.height) @@ -152,7 +151,6 @@ def generate_high_level_commands_for_sched_op(sched_op, schedule): ofm_box = Box(ofm_box_start.as_list(), ofm_box_end.as_list()) ifm_box = Box([], []) ifm2_box = Box([], []) - # Calculate IFM input box based on the OFM box if ifm: ifm_box, pad_top, pad_bottom = ofm_box.transform_with_strides_and_skirt( @@ -165,8 +163,8 @@ def generate_high_level_commands_for_sched_op(sched_op, schedule): read_offsets[0], read_shapes[0], upscaling, + op.type, ) - # Calculate IFM2 input box based on the OFM box if ifm2: ifm2_box, pad_top, pad_bottom = ofm_box.transform_with_strides_and_skirt( @@ -179,6 +177,7 @@ def generate_high_level_commands_for_sched_op(sched_op, schedule): read_offsets[1], read_shapes[1], upscaling, + op.type, ) ifm_required = ifm_box |