diff options
author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-08-28 13:25:14 +0200 |
---|---|---|
committer | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-09-03 10:50:10 +0200 |
commit | 0628a8c0136eebf3af8db7fd40b7aed94ff5d670 (patch) | |
tree | 65e7a49264f2df1f969122d8d7fb812773c0d4db /ethosu/vela/errors.py | |
parent | d2e3355813a33ccefaf112750f86c4f04d6ea12c (diff) | |
download | ethos-u-vela-0628a8c0136eebf3af8db7fd40b7aed94ff5d670.tar.gz |
MLBEDSW-2567: CLI option to specify allocation alignment
Added the CLI option. Only applies to CPU tensors. Added an
AllocationError which is raised when Allocation fails.
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I89164dea3ac7b7add7bc40aec2ce8fe50600105d
Diffstat (limited to 'ethosu/vela/errors.py')
-rw-r--r-- | ethosu/vela/errors.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/ethosu/vela/errors.py b/ethosu/vela/errors.py index 2c93fbc6..1a30d546 100644 --- a/ethosu/vela/errors.py +++ b/ethosu/vela/errors.py @@ -52,6 +52,13 @@ class OptionError(VelaError): self.data = "Incorrect argument to CLI option: {} {}: {}".format(option, option_value, msg) +class AllocationError(VelaError): + """Raised when allocation fails""" + + def __init__(self, msg): + self.data = msg + + def OperatorError(op, msg): """Called when parsing an operator results in errors""" |