From 9033795e399d29871bf95fc7f37959b498c46041 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Thu, 7 May 2020 16:42:35 +0100 Subject: 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 --- ethosu/vela/register_command_stream_generator.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'ethosu/vela') 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 -- cgit v1.2.1