aboutsummaryrefslogtreecommitdiff
path: root/tests/test_backend_vela_compiler.py
diff options
context:
space:
mode:
authorBenjamin Klimczak <benjamin.klimczak@arm.com>2022-11-28 13:36:03 +0000
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2022-12-06 17:05:37 +0000
commit0241f96fce1bbfa11f21bfbfa161b51f3170a51b (patch)
treef6ceae63670dd58f0b58de0ea55e039eddca3b50 /tests/test_backend_vela_compiler.py
parent6a88ee5315b4ce5b023370c1e55e48bf9f2b6f67 (diff)
downloadmlia-0241f96fce1bbfa11f21bfbfa161b51f3170a51b.tar.gz
MLIA-669 Upgrade dependencies with Vela 3.6
With Vela 3.6 we are able to remove the special treatment of aarch64 in our dependencies, i.e. - upgrade Vela to version 3.6 that resolves a compatibility issue for aarch64 in 3.4 and 3.5. - upgrade to TensorFlow 2.10 which now supports aarch64 (therefore making it obsolete to use 'tensorflow-aarch64'). Change-Id: I86508b667b5ccb55bfd11dcae9defc54e5ef74de
Diffstat (limited to 'tests/test_backend_vela_compiler.py')
-rw-r--r--tests/test_backend_vela_compiler.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_backend_vela_compiler.py b/tests/test_backend_vela_compiler.py
index ff07c74..20121d6 100644
--- a/tests/test_backend_vela_compiler.py
+++ b/tests/test_backend_vela_compiler.py
@@ -27,7 +27,7 @@ def test_default_vela_compiler() -> None:
assert default_compiler.tensor_allocator == TensorAllocator.HillClimb
assert default_compiler.cpu_tensor_alignment == 16
assert default_compiler.optimization_strategy == OptimizationStrategy.Performance
- assert default_compiler.output_dir is None
+ assert default_compiler.output_dir == "output"
assert default_compiler.get_config() == {
"accelerator_config": "ethos-u55-256",
@@ -86,7 +86,7 @@ def test_vela_compiler_with_parameters(test_resources_path: Path) -> None:
tensor_allocator="Greedy",
cpu_tensor_alignment=4,
optimization_strategy="Size",
- output_dir="output",
+ output_dir="custom_output",
)
compiler = VelaCompiler(compiler_options)
@@ -99,7 +99,7 @@ def test_vela_compiler_with_parameters(test_resources_path: Path) -> None:
assert compiler.tensor_allocator == TensorAllocator.Greedy
assert compiler.cpu_tensor_alignment == 4
assert compiler.optimization_strategy == OptimizationStrategy.Size
- assert compiler.output_dir == "output"
+ assert compiler.output_dir == "custom_output"
assert compiler.get_config() == {
"accelerator_config": "ethos-u65-256",