MLBEDSW-2337: Intermediate feature maps in fast storage
Attempts to use fast storage for feature maps used in between
cascaded passes.
This is only relevant for system configurations where feature maps
are by default not placed in SRAM, but there is SRAM for fast storage.
Change-Id: I207b7cf32cfcb5bea3e6b93c2da1161c4af5221d
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
diff --git a/ethosu/vela/test/test_compiler_driver.py b/ethosu/vela/test/test_compiler_driver.py
new file mode 100644
index 0000000..56a90c4
--- /dev/null
+++ b/ethosu/vela/test/test_compiler_driver.py
@@ -0,0 +1,44 @@
+# Copyright (C) 2020 Arm Limited or its affiliates. All rights reserved.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the License); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an AS IS BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# Description:
+# Unit tests for compiler driver
+from ethosu.vela.compiler_driver import next_sram_factor
+
+
+def test_next_sram_factor():
+ lower = 0.7
+ assert (1.0, False) == next_sram_factor([])
+ assert (None, False) == next_sram_factor([True])
+ assert (lower, True) == next_sram_factor([False])
+ assert ((1 + lower) / 2, True) == next_sram_factor([False, True])
+ assert (lower / 2, True) == next_sram_factor([False, False])
+ # Tests next_sram_factor for a range of simulated allocator efficiencies
+ for i in range(20):
+ allocator_factor = i / 20.0 # The simulated allocator efficiency
+ alloc_results = []
+ bisected_factor = 0 # The end result of the bisect search
+ while True:
+ factor, dry_test = next_sram_factor(alloc_results)
+ if factor is None:
+ break
+ alloc_result = factor < allocator_factor
+ if alloc_result and not dry_test:
+ bisected_factor = factor
+ alloc_results.append(alloc_result)
+ assert len(alloc_results) < 100
+ assert bisected_factor <= allocator_factor
+ assert abs(bisected_factor - allocator_factor) < 0.02