diff options
author | Tim Hall <tim.hall@arm.com> | 2020-05-07 16:42:35 +0100 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
commit | 9033795e399d29871bf95fc7f37959b498c46041 (patch) | |
tree | b8b2abac1227f7abfb89f9586ce14c0bd59189c5 /ethosu/vela/register_command_stream_generator.py | |
parent | 53d4752c8dbd9152238abd854ec23e87e5123881 (diff) | |
download | ethos-u-vela-9033795e399d29871bf95fc7f37959b498c46041.tar.gz |
Fix blockdep issue for MLBEDSW-1872 rnnoise
- 5 step rnnoise was failing due to secondary tensors
not being checked for operator dependency. This commit
adds ifm2 comparisons to the dependency check.
Change-Id: I629c8a70997481efb7f596d8b77512d3419eaab4
Signed-off-by: Tim Hall <tim.hall@arm.com>
Diffstat (limited to 'ethosu/vela/register_command_stream_generator.py')
-rw-r--r-- | ethosu/vela/register_command_stream_generator.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py index 0168c649..e767071d 100644 --- a/ethosu/vela/register_command_stream_generator.py +++ b/ethosu/vela/register_command_stream_generator.py @@ -271,10 +271,10 @@ def has_prev_op_dependency(prev_cmd, cmd): if prev_cmd is None: return False if (prev_cmd.cmdtype == cmd.cmdtype == CommandType.NpuStripe) and (prev_cmd.ps != cmd.ps): - if prev_cmd.ofm_tensor == cmd.ifm_tensor: + if prev_cmd.ofm_tensor.equivalence_id == cmd.ifm_tensor.equivalence_id: return True - else: - return prev_cmd.ofm_tensor.equivalence_id == cmd.ifm_tensor.equivalence_id + elif cmd.ifm2_tensor is not None: + return (prev_cmd.ofm_tensor.equivalence_id == cmd.ifm2_tensor.equivalence_id) return False |