diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-11-25 12:43:18 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-11-25 13:43:06 +0000 |
commit | 9785cc7cc3a6f853f9d52228e1a651b41f8f189f (patch) | |
tree | 0bb33a4527b1b3c70290210da6ed0f72667c5248 /ethosu/vela | |
parent | 34d29174c28c724a45602c5ebe8fdab4f86edde7 (diff) | |
download | ethos-u-vela-9785cc7cc3a6f853f9d52228e1a651b41f8f189f.tar.gz |
MLBEDSW-3352 Avoid ifm streaming for some cases
Changed so it is not allowed to do ifm-streaming for
TransposeConv and ResizeBilinear
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I85da279fae6202830c46e4a5500fb1b0dd6ef542
Diffstat (limited to 'ethosu/vela')
-rw-r--r-- | ethosu/vela/scheduler.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index d6e890ab..943c590c 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -557,6 +557,12 @@ class DynamicProgrammingScheduler: return strat_data + def avoid_ifm_streaming(self, ps): + for op in ps.ops: + if op.type in (Op.Conv2DBackpropInputSwitchedBias, Op.ResizeBilinear): + return True + return False + @lru_cache(maxsize=None) def search_output(self, ps): @@ -565,7 +571,7 @@ class DynamicProgrammingScheduler: candidate_list.extend(self.search_weight_streaming_output(ps)) - if self.options.use_ifm_streaming: + if self.options.use_ifm_streaming and not self.avoid_ifm_streaming(ps): candidate_list.extend(self.search_ifm_streaming_output(ps)) best = self.filter_pareto_frontier(candidate_list, remove_equally_good_candidates=True) |