[MLBEDSW-2749] removed the decorator for typecheck
*the decorator is causing the verification tests to fail when using TF
2.1, but not with TF 2.2, hence removing it for now.
Change-Id: I07357c0fef383d9a65278fe99ad8e4d3f7dc6d9b
Signed-off-by: Manupa Karunaratne <manupa.karunaratne@arm.com>
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py
index 7e03a94..45427a1 100644
--- a/ethosu/vela/weight_compressor.py
+++ b/ethosu/vela/weight_compressor.py
@@ -23,7 +23,6 @@
from .architecture_features import Accelerator
from .architecture_features import ArchitectureFeatures
from .data_type import DataType
-from .errors import typecheck
from .errors import UnsupportedFeatureError
from .nn_graph import SchedulingStrategy
from .numeric_util import round_up
@@ -45,7 +44,6 @@
)
-@typecheck
def encode_weights(
accelerator: Accelerator,
weights_volume: np.ndarray,
@@ -68,6 +66,15 @@
:return: a bytearray of compressed weights
"""
+ # Check arg types
+ assert isinstance(accelerator, Accelerator)
+ assert isinstance(weights_volume, np.ndarray)
+ assert isinstance(dilation_xy, tuple)
+ assert isinstance(ifm_bitdepth, int)
+ assert isinstance(ofm_block_depth, int)
+ assert isinstance(is_depthwise, bool)
+ assert isinstance(is_partkernel, bool)
+
# Checks for weight layout
assert len(weights_volume.shape) == 4, "weights ndarray should have a shape of 4"
@@ -94,7 +101,6 @@
return encoded_stream
-@typecheck
def encode_bias(bias: np.int64, scale: int, shift: int):
"""
Public facing API to pack bias and scale values as required by the hardware
@@ -103,6 +109,11 @@
:param shift: 6bit shift value
:return: packed 80bit [0(2-bits),shift(6-bits),scale(32-bits),bias(40-bits)]
"""
+ # Check arg types
+ assert isinstance(bias, np.int64)
+ assert isinstance(scale, int)
+ assert isinstance(shift, int)
+
assert -(1 << (40 - 1)) <= bias < (1 << (40 - 1)) # signed 40-bit range
assert 0 <= scale < (1 << 32) # unsigned 32-bit range
assert 0 <= shift < (1 << 6) # unsigned 6-bit range