aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTim Hall <tim.hall@arm.com>2020-05-07 16:42:35 +0100
committerTim Hall <tim.hall@arm.com>2020-06-18 17:53:52 +0100
commit9033795e399d29871bf95fc7f37959b498c46041 (patch)
treeb8b2abac1227f7abfb89f9586ce14c0bd59189c5
parent53d4752c8dbd9152238abd854ec23e87e5123881 (diff)
downloadethos-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>
-rw-r--r--ethosu/vela/register_command_stream_generator.py6
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