diff options
author | Ayaan Masood <Ayaan.Masood@arm.com> | 2022-02-22 11:28:55 +0000 |
---|---|---|
committer | Ayaan Masood <Ayaan.Masood@arm.com> | 2022-02-22 11:28:55 +0000 |
commit | b801dda26bbcff8ec4f7967d60f38239fd16912b (patch) | |
tree | bb8aa6bc3141f1dd48ffb22630a1eac1cc209825 /ethosu/vela/scheduler.py | |
parent | 5e0ae5598ab1d7debd603bdd32c7e8f9cad9d581 (diff) | |
download | ethos-u-vela-b801dda26bbcff8ec4f7967d60f38239fd16912b.tar.gz |
MLBEDSW-5880 Fixed Vela verbose weight flag
*Original weights and encoded NPU weight now report correct size instead
of zero when running vela with --verbose-weights flag
(Code to update the aforementioned attributes was missing)
*Removed print references to unencoded NPU weight size
Change-Id: I6d3e41c04cc46d24eeb54cab89818a35e5df27be
Signed-off-by: Ayaan Masood <Ayaan.Masood@arm.com>
Diffstat (limited to 'ethosu/vela/scheduler.py')
-rw-r--r-- | ethosu/vela/scheduler.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index 8f2426c1..6b084593 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -339,7 +339,7 @@ class Scheduler: self.nng = nng self.sg = sg self.arch = arch - self.sched_ops: List(SchedulerOperation) = [] + self.sched_ops: List[SchedulerOperation] = [] self.max_schedule = None self.scheduler_options = options @@ -459,7 +459,6 @@ class Scheduler: def create_initial_schedule(self) -> Schedule: """Creates an initial schedule with no cascading or buffering of any kind""" schedule = Schedule(self.sg, "MAX") - for op in self.sched_ops: cost = op.create_scheduler_info(self.nng, op.ofm.shape) cost.cycles = self.estimate_op_performance(op, cost.block_config, op.ofm.shape.depth) |