aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/test/extapi/test_extapi_encode_weights.py
diff options
context:
space:
mode:
authorPatrik Gustavsson <patrik.gustavsson@arm.com>2020-11-18 17:05:50 +0100
committerpatrik.gustavsson <patrik.gustavsson@arm.com>2020-11-19 12:12:07 +0000
commitc8a22f198c4fb23735962fee00c062b0325ca6a6 (patch)
treee9a99cb102d43471a6454267170aa7e5ba3b541f /ethosu/vela/test/extapi/test_extapi_encode_weights.py
parentbdf09f9eca75fab14298a94a5402efc15f698e68 (diff)
downloadethos-u-vela-c8a22f198c4fb23735962fee00c062b0325ca6a6.tar.gz
MLBEDSW-3251 Add version to external API
Added version to the external API -Added CLI-option --api_version -Added API function to get the API version Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com> Change-Id: I0143b50adf884a2b05145912a1c7bef8cecc5f02
Diffstat (limited to 'ethosu/vela/test/extapi/test_extapi_encode_weights.py')
-rw-r--r--ethosu/vela/test/extapi/test_extapi_encode_weights.py6
1 files changed, 0 insertions, 6 deletions
diff --git a/ethosu/vela/test/extapi/test_extapi_encode_weights.py b/ethosu/vela/test/extapi/test_extapi_encode_weights.py
index 356bbc1a..0459cae1 100644
--- a/ethosu/vela/test/extapi/test_extapi_encode_weights.py
+++ b/ethosu/vela/test/extapi/test_extapi_encode_weights.py
@@ -66,9 +66,3 @@ def test_encode_weights(
block_traversal=block_traversal,
)
assert type(encoded_stream) == bytearray
-
-
-if __name__ == "__main__":
- # two test candidates for debugging purposes
- test_encode_weights(Accelerator.Ethos_U55_256, ((3, 3, 25, 16), 8), 1, 1, 8, 0)
- test_encode_weights(Accelerator.Ethos_U55_256, ((16, 16, 16, 16), 8), 1, 1, 8, 0)