commit bdb1d6e0fce5e52979f3a5742aaddd3a68b9a0f2
author Jonny Svärd <jonny.svaerd@arm.com> Tue Jan 04 12:38:29 2022 +0100
committer Jonny Svärd <jonny.svaerd@arm.com> Tue Jan 04 15:25:55 2022 +0100
tree 37f80160819a79e8c6b56eb59fc7bbe19ab78ce7
parent 3bbbed68f85c75806cd08b6453cdb678e47ceb2a

Set product in driver action config struct

Signed-off-by: Jonny Svärd <jonny.svaerd@arm.com>
Change-Id: Ib398024c2f41beb4f93f7976c678a9fd54af94a5

ethosu/vela/driver_actions.py

diff --git a/ethosu/vela/driver_actions.py b/ethosu/vela/driver_actions.py
index 317b2a6..90af02c 100644
--- a/ethosu/vela/driver_actions.py
+++ b/ethosu/vela/driver_actions.py
@@ -75,6 +75,10 @@
     log2_macs_cc = int(np.log2(macs_cc) + 0.5)
     shram_size = arch.ncores * int(arch.shram_size_bytes / 1024)
     n = config_r()
+    if arch.is_ethos_u65_system:
+        n.set_product(1)
+    else:
+        n.set_product(0)  # U55
     n.set_shram_size(shram_size)
     n.set_cmd_stream_version(0)  # may be incremented in the future
     n.set_macs_per_cc(log2_macs_cc)