diff options
| author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-01 12:39:55 +0100 |
|---|---|---|
| committer | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-21 11:09:39 +0100 |
| commit | 845e23200d471e44f274940846e400d170b5ff37 (patch) | |
| tree | 28a01492bf11f0ff69309ead9bd8a1bad9e14cbb /ethosu/vela/register_command_stream_util.py | |
| parent | d2b5510697e7789f5a416f9d80d3cb640eecc092 (diff) | |
| download | ethos-u-vela-845e23200d471e44f274940846e400d170b5ff37.tar.gz | |
MLBEDSW-3367 Add mypy to pre-commit
Add mypy to pre-commit and clean up all reported errors.
Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: If7dc869f5fecdb0e2db40f14e7d9db21aa33df71
Diffstat (limited to 'ethosu/vela/register_command_stream_util.py')
| -rw-r--r-- | ethosu/vela/register_command_stream_util.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ethosu/vela/register_command_stream_util.py b/ethosu/vela/register_command_stream_util.py index 3751d88e..83126ead 100644 --- a/ethosu/vela/register_command_stream_util.py +++ b/ethosu/vela/register_command_stream_util.py @@ -163,7 +163,7 @@ def get_h_ranges( return [get_address_range(fm, strides, y, x0, c0, y, x1, c1) for y in range(y0, y1 + 1)] -def get_address_ranges_for_area(fm: NpuFeatureMap, start: PointXYZ, end: PointXYZ) -> List[NpuAddressRange]: +def get_address_ranges_for_area(fm: NpuFeatureMap, start: PointXYZ, end: PointXYZ) -> List[Optional[NpuAddressRange]]: """ Returns a list of adddress ranges that covers the area start - end (inclusive). Divides the area in horizontal "stripes" of height 1, and returns the address ranges for these "stripes". @@ -183,7 +183,7 @@ def get_address_ranges_for_area(fm: NpuFeatureMap, start: PointXYZ, end: PointXY h, w, c = fm.shape y0, x0, c0 = start.y, start.x, start.z y1, x1, c1 = min(end.y, h - 1), min(end.x, w - 1), min(end.z, c - 1) - ranges = [] + ranges: List[Optional[NpuAddressRange]] = [] if x0 < width_0 and y0 < height_0: # Horizontal ranges for tile 0 ranges.extend(get_h_ranges(fm, strides, y0, x0, c0, min(y1, height_0 - 1), min(x1, width_0 - 1), c1)) @@ -373,7 +373,7 @@ def intersects( else: # The OFM produces a part of the IFM (e.g. a stripe), or the IFM consumes part of the OFM. # In this case, address comparison between the two areas is needed - ifm_ranges = get_address_ranges_for_area(ifm, ifm_start_coord, ifm_end_coord) + ifm_ranges: List[Optional[NpuAddressRange]] = get_address_ranges_for_area(ifm, ifm_start_coord, ifm_end_coord) prev_ofm_ranges = get_address_ranges_for_area(prev_ofm, ofm_start_coord, ofm_end_coord) res = range_lists_overlap(ifm_ranges, prev_ofm_ranges) return res |