diff options
| author | Tim Hall <tim.hall@arm.com> | 2022-03-16 16:51:16 +0000 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2022-03-31 10:26:56 +0100 |
| commit | 68df8a1f5469daac53b7a418d92204f7026e4228 (patch) | |
| tree | 0db9df020c89e25c20ed4a7a738dedd1b0ceb023 /ethosu/vela/api.py | |
| parent | d85750702229af97c0b0bbda6e397a23254b6144 (diff) | |
| download | ethos-u-vela-68df8a1f5469daac53b7a418d92204f7026e4228.tar.gz | |
vela: Added debug info to external API
- Added optional name attributes to operators and tensors
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I3b5d881a7b1043a6ba4b58fff5d7532b271ba536
Diffstat (limited to 'ethosu/vela/api.py')
| -rw-r--r-- | ethosu/vela/api.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ethosu/vela/api.py b/ethosu/vela/api.py index 3382ea99..399fd46d 100644 --- a/ethosu/vela/api.py +++ b/ethosu/vela/api.py @@ -27,7 +27,7 @@ import numpy API_VERSION_MAJOR = 1 -API_VERSION_MINOR = 2 +API_VERSION_MINOR = 3 API_VERSION = f"{API_VERSION_MAJOR}.{API_VERSION_MINOR}" @@ -253,6 +253,8 @@ class NpuFeatureMap: self.layout: NpuLayout = NpuLayout.NHWC # x/y/c strides used by the NPU when traversing the feature map, if None, vela will use default strides self.strides: Optional[NpuShape3D] = None + # Used for debug + self.name: Optional[str] = None class NpuKernel: @@ -290,6 +292,8 @@ class NpuOperation: def __init__(self, op_type: NpuOperationType): self.op_type = op_type + # Used for debug + self.name: Optional[str] = None class NpuDmaOperation(NpuOperation): |