aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/scheduler.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/scheduler.py')
-rw-r--r--ethosu/vela/scheduler.py8
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)