commit 7db78969dc8ead72f3ded81b6d2a6a7ed798ea62
author Louis Verhaard <louis.verhaard@arm.com> Mon May 25 15:05:26 2020 +0200
committer Tim Hall <tim.hall@arm.com> Thu Jun 18 17:53:52 2020 +0100
tree 011bcf579cc8e0f007f9564a98cc5c05df34322b
parent 78792223369fa34dacd0e69e189af035283da2ae

MLBEDSW-2067: added custom exceptions

Added custom exceptions to handle different types of input errors.

Also performed minor formatting changes using flake8/black.

Change-Id: Ie5b05361507d5e569aff045757aec0a4a755ae98
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>

ethosu/vela/shared_buffer_allocation.py

diff --git a/ethosu/vela/shared_buffer_allocation.py b/ethosu/vela/shared_buffer_allocation.py
index 335b863..2bfe594 100644
--- a/ethosu/vela/shared_buffer_allocation.py
+++ b/ethosu/vela/shared_buffer_allocation.py
@@ -22,6 +22,7 @@
 from .architecture_features import Kernel
 from .architecture_features import SharedBufferArea
 from .architecture_features import SHRAMElements
+from .errors import OptionError
 from .operation import NpuBlockType
 
 
@@ -163,8 +164,11 @@
 
     if arch.override_block_config:
         config = alloc.try_block(arch.override_block_config)
-        assert config, "Block config override cannot be used"
-        return [config]
+        raise OptionError(
+            "--force-block-config",
+            str(arch.override_block_config),
+            "This forced block config value cannot be used; it is not compatible",
+        )
 
     # Constrain the search space if the OFM is smaller than the max block size
     # - Add other block search constraints here if required