blob: 7b2a8ef8ed9b21e9ea640678047514628ced2395 [file] [log] [blame]
Kristofer Jonsson49bdee82020-04-06 13:21:21 +02001/*
Kristofer Jonsson09273d12021-03-15 08:43:08 +01002 * Copyright (c) 2019-2021 Arm Limited. All rights reserved.
Kristofer Jonsson49bdee82020-04-06 13:21:21 +02003 *
4 * SPDX-License-Identifier: Apache-2.0
5 *
6 * Licensed under the Apache License, Version 2.0 (the License); you may
7 * not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
9 *
10 * www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an AS IS BASIS, WITHOUT
14 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
17 */
18
Kristofer Jonsson2b201c32020-09-02 16:42:43 +020019/******************************************************************************
20 * Includes
21 ******************************************************************************/
22
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020023#include "ethosu_driver.h"
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020024#include "ethosu_common.h"
Bhavik Pateldae5be02020-06-18 15:25:15 +020025#include "ethosu_config.h"
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020026#include "ethosu_device.h"
Per Åstrand25d78c02020-04-21 14:19:44 +020027
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020028#include <assert.h>
Per Åstrand25d78c02020-04-21 14:19:44 +020029#include <cmsis_compiler.h>
Per Åstrand14ccfee2020-09-25 10:40:20 +020030#include <inttypes.h>
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020031#include <stdbool.h>
Bhavik Patelbf7ae632020-06-11 21:00:16 +020032#include <stddef.h>
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020033#include <stdio.h>
34#include <stdlib.h>
35
Kristofer Jonsson2b201c32020-09-02 16:42:43 +020036/******************************************************************************
37 * Defines
38 ******************************************************************************/
39
40#define MACS_PER_CYCLE_LOG2_MASK 0x000F
41#define SHRAM_SIZE_MASK 0xFF00
42#define SHRAM_SIZE_RIGHT_SHIFT 8
43#define BYTES_IN_32_BITS 4
44#define CUSTOM_OPTION_LENGTH_32_BIT_WORD 1
45#define DRIVER_ACTION_LENGTH_32_BIT_WORD 1
46#define OPTIMIZER_CONFIG_LENGTH_32_BIT_WORD 2
47#define ETHOSU_FOURCC ('1' << 24 | 'P' << 16 | 'O' << 8 | 'C') // "Custom Operator Payload 1"
48#define APB_START_ADDR_MASK 0x0FFF
49#define APB_NUM_REG_BIT_SHIFT 12
50#define BYTES_1KB 1024
51#define PRODUCT_MAJOR_ETHOSU55 (4)
52#define MASK_16_BYTE_ALIGN (0xF)
53#define FAST_MEMORY_BASE_ADDR_INDEX 2
54
55/******************************************************************************
56 * Types
57 ******************************************************************************/
58
59// Driver actions
60enum DRIVER_ACTION_e
61{
62 RESERVED = 0,
63 OPTIMIZER_CONFIG = 1,
64 COMMAND_STREAM = 2,
65 READ_APB_REG = 3,
66 DUMP_SHRAM = 4,
67 NOP = 5,
68};
69
70// Custom data struct
71struct custom_data_s
72{
73 union
74 {
75 // Driver action data
76 struct
77 {
78 // Driver action command (valid values in DRIVER_ACTION_e)
79 uint8_t driver_action_command;
80
81 // reserved
82 uint8_t reserved;
83
84 // Driver action data
85 union
86 {
87 // DA_CMD_OPT_CFG
88 struct
89 {
90 uint16_t rel_nbr : 4;
91 uint16_t patch_nbr : 4;
92 uint16_t opt_cfg_reserved : 8;
93 };
94
95 // DA_CMD_CMSTRM
96 struct
97 {
98 uint16_t length;
99 };
100
101 // DA_CMD_READAPB
102 struct
103 {
104 uint16_t start_address : 12;
105 uint16_t nbr_reg_minus1 : 4;
106 };
107
108 uint16_t driver_action_data;
109 };
110 };
111
112 uint32_t word;
113 };
114};
115
116// optimizer config struct
117struct opt_cfg_s
118{
119 struct custom_data_s da_data;
120 union
121 {
122 struct
123 {
124 uint32_t macs_per_cc : 4;
125 uint32_t cmd_stream_version : 4;
126 uint32_t shram_size : 8;
127 uint32_t reserved1 : 16;
128 };
129 uint32_t npu_cfg;
130 };
131 union
132 {
133 struct
134 {
135 uint32_t version_status : 4;
136 uint32_t version_minor : 4;
137 uint32_t version_major : 4;
138 uint32_t product_major : 4;
139 uint32_t arch_patch_rev : 4;
140 uint32_t arch_minor_rev : 8;
141 uint32_t arch_major_rev : 4;
142 };
143 uint32_t ethosu_id;
144 };
145};
146
147/******************************************************************************
148 * Functions
149 ******************************************************************************/
150
Kristofer Jonssonef387ea2020-08-25 16:32:21 +0200151struct ethosu_driver ethosu_drv = {
Kristofer Jonssonc6e7a1f2020-11-24 09:20:14 +0100152 .dev = {.base_address = NULL, .proto = 0, .pmccntr = {0}, .pmu_evcntr = {0, 0, 0, 0}, .pmu_evtypr = {0, 0, 0, 0}},
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100153 .abort_inference = false,
154 .status_error = false,
155 .dev_power_always_on = false};
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200156
Anton Moberg61da4d32020-12-22 16:00:31 +0100157// Registered drivers linked list HEAD
158static struct ethosu_driver *registered_drivers = NULL;
159
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100160/*
161 * Following section handles the minimal sempahore and mutex implementation in case of baremetal applications.
162 * Weak symbols will be overwritten by RTOS definitions and implement true thread-safety. (Done in application layer)
163 */
164
165// Baremetal sempahore implementation
166struct ethosu_semaphore_t
167{
168 int count;
169};
170
171// Minimal needed declaration to allow baremetal functionality.
172static void *ethosu_mutex;
173static void *ethosu_semaphore;
174
175void *__attribute__((weak)) ethosu_mutex_create(void) {}
176
177void __attribute__((weak)) ethosu_mutex_lock(void *mutex) {}
178
179void __attribute__((weak)) ethosu_mutex_unlock(void *mutex) {}
180
181// Baremetal implementation of creating a semaphore
182void *__attribute__((weak)) ethosu_semaphore_create(void)
183{
184 struct ethosu_semaphore_t *sem = malloc(sizeof(*sem));
185 sem->count = 1;
186 return sem;
187}
188
189// Baremetal simulation of waiting/sleeping for and then taking a semaphore using intrisics
190void __attribute__((weak)) ethosu_semaphore_take(void *sem)
191{
192 struct ethosu_semaphore_t *s = sem;
193 while (s->count <= 0)
194 {
195 __WFE();
196 }
197 s->count--;
198}
199
200// Baremetal simulation of giving a semaphore and waking up processes using intrinsics
201void __attribute__((weak)) ethosu_semaphore_give(void *sem)
202{
203 struct ethosu_semaphore_t *s = sem;
204 s->count++;
205 __SEV();
206}
207// <--- End of semaphore and mutex implementations
208
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100209static int ethosu_soft_reset_and_restore(struct ethosu_driver *drv);
Anton Moberg61da4d32020-12-22 16:00:31 +0100210
Kristofer Jonsson09273d12021-03-15 08:43:08 +0100211void __attribute__((weak)) ethosu_irq_handler_v2(struct ethosu_driver *drv)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200212{
213 uint8_t irq_raised = 0;
Kristofer Jonsson125429a2020-08-20 16:52:23 +0200214
215 LOG_DEBUG("Interrupt. status=0x%08x, qread=%d\n",
Anton Moberg61da4d32020-12-22 16:00:31 +0100216 ethosu_read_reg(&drv->dev, NPU_REG_STATUS),
217 ethosu_read_reg(&drv->dev, NPU_REG_QREAD));
Kristofer Jonsson125429a2020-08-20 16:52:23 +0200218
219 // Verify that interrupt has been raised
Anton Moberg61da4d32020-12-22 16:00:31 +0100220 (void)ethosu_is_irq_raised(&drv->dev, &irq_raised);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200221 ASSERT(irq_raised == 1);
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100222 drv->irq_triggered = true;
Kristofer Jonsson125429a2020-08-20 16:52:23 +0200223
224 // Clear interrupt
Anton Moberg61da4d32020-12-22 16:00:31 +0100225 (void)ethosu_clear_irq_status(&drv->dev);
Kristofer Jonsson125429a2020-08-20 16:52:23 +0200226
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200227 // Verify that interrupt has been successfully cleared
Anton Moberg61da4d32020-12-22 16:00:31 +0100228 (void)ethosu_is_irq_raised(&drv->dev, &irq_raised);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200229 ASSERT(irq_raised == 0);
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200230
Anton Moberg61da4d32020-12-22 16:00:31 +0100231 if (ethosu_status_has_error(&drv->dev))
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200232 {
Anton Moberg61da4d32020-12-22 16:00:31 +0100233 ethosu_soft_reset_and_restore(drv);
234 drv->status_error = true;
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200235 }
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100236
237 ethosu_semaphore_give(drv->semaphore);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200238}
239
Bhavik Pateldae5be02020-06-18 15:25:15 +0200240static inline void wait_for_irq(struct ethosu_driver *drv)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200241{
242 while (1)
243 {
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100244 if (drv->irq_triggered || drv->abort_inference)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200245 {
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100246 drv->irq_triggered = false;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200247 break;
248 }
249
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100250 ethosu_semaphore_take(drv->semaphore);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200251 }
252}
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200253
Bhavik Pateldae5be02020-06-18 15:25:15 +0200254static int handle_optimizer_config(struct ethosu_driver *drv, struct opt_cfg_s *opt_cfg_p);
255static int handle_command_stream(struct ethosu_driver *drv,
256 const uint8_t *cmd_stream,
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200257 const int cms_length,
258 const uint64_t *base_addr,
Per Åstrand3c8afcc2020-10-20 10:29:59 +0200259 const size_t *base_addr_size,
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200260 const int num_base_addr);
Bhavik Pateldae5be02020-06-18 15:25:15 +0200261static int read_apb_reg(struct ethosu_driver *drv, uint16_t);
262static int dump_shram(struct ethosu_driver *drv);
263static void dump_npu_register(struct ethosu_driver *drv, int npu_reg, int npu_reg_end);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200264static void dump_command_stream(const uint32_t *cmd_stream, const int cms_length, int qread);
Bhavik Pateldae5be02020-06-18 15:25:15 +0200265static void npu_axi_init(struct ethosu_driver *drv);
Anton Mobergdf386e02021-02-02 11:26:48 +0100266static struct ethosu_driver *ethosu_find_and_reserve_driver(void);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200267
Anton Moberg61da4d32020-12-22 16:00:31 +0100268int ethosu_init_v4(struct ethosu_driver *drv,
269 const void *base_address,
Per Åstrande6498f02020-11-09 15:33:12 +0100270 const void *fast_memory,
271 const size_t fast_memory_size,
272 uint32_t secure_enable,
273 uint32_t privilege_enable)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200274{
275 int return_code = 0;
Bhavik Pateldae5be02020-06-18 15:25:15 +0200276
Bhavik Patel033bb1b2020-12-17 15:33:33 +0100277 LOG_INFO("%s. base_address=%p, fast_memory=%p, fast_memory_size=%zu, secure=%" PRIu32 ", privileged=%" PRIu32 "\n",
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200278 __FUNCTION__,
279 base_address,
280 fast_memory,
Per Åstrande6498f02020-11-09 15:33:12 +0100281 fast_memory_size,
282 secure_enable,
283 privilege_enable);
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200284
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100285 if (!ethosu_mutex)
286 {
287 ethosu_mutex = ethosu_mutex_create();
288 }
289
290 if (!ethosu_semaphore)
291 {
292 ethosu_semaphore = ethosu_semaphore_create();
293 }
294
Anton Moberg61da4d32020-12-22 16:00:31 +0100295 ethosu_register_driver(drv);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200296
Anton Moberg61da4d32020-12-22 16:00:31 +0100297 drv->fast_memory = (uint32_t)fast_memory;
298 drv->fast_memory_size = fast_memory_size;
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100299 drv->irq_triggered = false;
300 drv->semaphore = ethosu_semaphore_create();
Anton Moberg61da4d32020-12-22 16:00:31 +0100301
302 if (ETHOSU_SUCCESS != ethosu_dev_init(&drv->dev, base_address, secure_enable, privilege_enable))
Bhavik Pateldae5be02020-06-18 15:25:15 +0200303 {
304 LOG_ERR("Failed in ethosu_dev_init");
305 return -1;
306 }
307
Anton Moberg0a614292021-03-24 14:08:22 +0100308 if (ETHOSU_SUCCESS !=
309 set_clock_and_power_request(drv, ETHOSU_INFERENCE_REQUEST, ETHOSU_CLOCK_Q_DISABLE, ETHOSU_POWER_Q_DISABLE))
Bhavik Patele645fed2020-06-12 14:46:47 +0200310 {
311 LOG_ERR("Failed to disable clock-q & power-q for Ethos-U\n");
312 return -1;
313 }
314
Anton Moberg61da4d32020-12-22 16:00:31 +0100315 if (ETHOSU_SUCCESS != ethosu_soft_reset(&drv->dev))
Per Åstrand849cf692020-11-24 07:39:55 +0100316 {
317 return -1;
318 }
Kristofer Jonssondaa0d202020-05-12 12:23:16 +0200319
Anton Moberg61da4d32020-12-22 16:00:31 +0100320 if (ETHOSU_SUCCESS != ethosu_wait_for_reset(&drv->dev))
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200321 {
322 LOG_ERR("Failed reset of Ethos-U\n");
323 return -1;
324 }
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100325
Anton Moberg61da4d32020-12-22 16:00:31 +0100326 drv->status_error = false;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200327
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200328 return return_code;
329}
330
Anton Moberg61da4d32020-12-22 16:00:31 +0100331int ethosu_get_version_v2(struct ethosu_driver *drv, struct ethosu_version *version)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200332{
333 int return_code = 0;
334
335 if (NULL != version)
336 {
337 struct ethosu_id id;
338 struct ethosu_config cfg;
Anton Moberg61da4d32020-12-22 16:00:31 +0100339 (void)ethosu_get_id(&drv->dev, &id);
340 (void)ethosu_get_config(&drv->dev, &cfg);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200341
342 version->id.version_status = id.version_status;
343 version->id.version_minor = id.version_minor;
344 version->id.version_major = id.version_major;
345 version->id.product_major = id.product_major;
346 version->id.arch_patch_rev = id.arch_patch_rev;
347 version->id.arch_minor_rev = id.arch_minor_rev;
348 version->id.arch_major_rev = id.arch_major_rev;
349 version->id.driver_patch_rev = ETHOSU_DRIVER_VERSION_PATCH;
350 version->id.driver_minor_rev = ETHOSU_DRIVER_VERSION_MINOR;
351 version->id.driver_major_rev = ETHOSU_DRIVER_VERSION_MAJOR;
352 version->cfg.macs_per_cc = cfg.macs_per_cc;
353 version->cfg.cmd_stream_version = cfg.cmd_stream_version;
354 version->cfg.shram_size = cfg.shram_size;
355 }
356 else
357 {
358 return_code = -1;
359 }
360
361 return return_code;
362}
363
Anton Moberg61da4d32020-12-22 16:00:31 +0100364int ethosu_invoke_v3(struct ethosu_driver *drv,
365 const void *custom_data_ptr,
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200366 const int custom_data_size,
367 const uint64_t *base_addr,
368 const size_t *base_addr_size,
369 const int num_base_addr)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200370{
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200371 const struct custom_data_s *data_ptr = custom_data_ptr;
372 const struct custom_data_s *data_end = custom_data_ptr + custom_data_size;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200373 int return_code = 0;
374
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200375 LOG_INFO("%s\n", __FUNCTION__);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200376
377 // First word in custom_data_ptr should contain "Custom Operator Payload 1"
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200378 if (data_ptr->word != ETHOSU_FOURCC)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200379 {
Per Åstrand14ccfee2020-09-25 10:40:20 +0200380 LOG_ERR("Custom Operator Payload: %" PRIu32 " is not correct, expected %x\n", data_ptr->word, ETHOSU_FOURCC);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200381 return -1;
382 }
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200383
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200384 // Custom data length must be a multiple of 32 bits
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200385 if ((custom_data_size % BYTES_IN_32_BITS) != 0)
386 {
387 LOG_ERR("ethosu_invoke ERROR custom_data_size=0x%x not a multiple of 4\n", custom_data_size);
388 return -1;
389 }
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200390
391 ++data_ptr;
392
393 // Adjust base address to fast memory area
Anton Moberg61da4d32020-12-22 16:00:31 +0100394 if (drv->fast_memory != 0 && num_base_addr >= FAST_MEMORY_BASE_ADDR_INDEX)
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200395 {
396 uint64_t *fast_memory = (uint64_t *)&base_addr[FAST_MEMORY_BASE_ADDR_INDEX];
397
Anton Moberg61da4d32020-12-22 16:00:31 +0100398 if (base_addr_size != NULL && base_addr_size[FAST_MEMORY_BASE_ADDR_INDEX] > drv->fast_memory_size)
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200399 {
Kristofer Jonsson4c94b302020-11-06 10:33:21 +0100400 LOG_ERR("Fast memory area too small. fast_memory_size=%u, base_addr_size=%u\n",
Anton Moberg61da4d32020-12-22 16:00:31 +0100401 drv->fast_memory_size,
Kristofer Jonsson4c94b302020-11-06 10:33:21 +0100402 base_addr_size[FAST_MEMORY_BASE_ADDR_INDEX]);
403 return -1;
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200404 }
Kristofer Jonsson4c94b302020-11-06 10:33:21 +0100405
Anton Moberg61da4d32020-12-22 16:00:31 +0100406 *fast_memory = drv->fast_memory;
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200407 }
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200408
Anton Moberg61da4d32020-12-22 16:00:31 +0100409 if (!drv->dev_power_always_on)
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200410 {
Anton Moberg61da4d32020-12-22 16:00:31 +0100411 // Only soft reset if securty state or privilege level needs changing
412 if (drv->dev.proto != ethosu_read_reg(&drv->dev, NPU_REG_PROT))
Per Åstrand849cf692020-11-24 07:39:55 +0100413 {
Anton Moberg61da4d32020-12-22 16:00:31 +0100414 if (ETHOSU_SUCCESS != ethosu_soft_reset(&drv->dev))
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100415 {
416 return -1;
417 }
Per Åstrand849cf692020-11-24 07:39:55 +0100418 }
Anton Moberg61da4d32020-12-22 16:00:31 +0100419
420 drv->status_error = false;
Anton Moberg0a614292021-03-24 14:08:22 +0100421 set_clock_and_power_request(drv, ETHOSU_INFERENCE_REQUEST, ETHOSU_CLOCK_Q_ENABLE, ETHOSU_POWER_Q_DISABLE);
Anton Moberg61da4d32020-12-22 16:00:31 +0100422 ethosu_restore_pmu_config(&drv->dev);
423 npu_axi_init(drv);
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200424 }
Kristofer Jonssonc6e7a1f2020-11-24 09:20:14 +0100425
Anton Moberg61da4d32020-12-22 16:00:31 +0100426 drv->status_error = false;
Kristofer Jonsson125429a2020-08-20 16:52:23 +0200427
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200428 while (data_ptr < data_end)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200429 {
Bhavik Patele645fed2020-06-12 14:46:47 +0200430 int ret = 0;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200431 switch (data_ptr->driver_action_command)
432 {
433 case OPTIMIZER_CONFIG:
434 LOG_INFO("ethosu_invoke OPTIMIZER_CONFIG\n");
435 struct opt_cfg_s *opt_cfg_p = (struct opt_cfg_s *)data_ptr;
436
Anton Moberg61da4d32020-12-22 16:00:31 +0100437 ret = handle_optimizer_config(drv, opt_cfg_p);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200438 data_ptr += DRIVER_ACTION_LENGTH_32_BIT_WORD + OPTIMIZER_CONFIG_LENGTH_32_BIT_WORD;
439 break;
440 case COMMAND_STREAM:
441 LOG_INFO("ethosu_invoke COMMAND_STREAM\n");
442 void *command_stream = (uint8_t *)(data_ptr) + sizeof(struct custom_data_s);
443 int cms_length = (data_ptr->reserved << 16) | data_ptr->length;
444
Anton Moberg61da4d32020-12-22 16:00:31 +0100445 drv->abort_inference = false;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200446 // It is safe to clear this flag without atomic, because npu is not running.
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100447 drv->irq_triggered = false;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200448
Anton Moberg61da4d32020-12-22 16:00:31 +0100449 ret = handle_command_stream(drv, command_stream, cms_length, base_addr, base_addr_size, num_base_addr);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200450
Anton Moberg61da4d32020-12-22 16:00:31 +0100451 if (return_code == -1 && drv->abort_inference)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200452 {
453 uint32_t qread = 0;
Anton Moberg61da4d32020-12-22 16:00:31 +0100454 ethosu_get_qread(&drv->dev, &qread);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200455 LOG_ERR("NPU timeout\n");
456 dump_command_stream(command_stream, cms_length, qread);
Anton Moberg61da4d32020-12-22 16:00:31 +0100457 dump_npu_register(drv, 0x200, 0x2BF);
458 dump_npu_register(drv, 0x800, 0xB3F);
459 dump_shram(drv);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200460 }
461
462 data_ptr += DRIVER_ACTION_LENGTH_32_BIT_WORD + cms_length;
463 break;
464 case READ_APB_REG:
465 LOG_INFO("ethosu_invoke READ_APB_REG\n");
Anton Moberg61da4d32020-12-22 16:00:31 +0100466 ret = read_apb_reg(drv, data_ptr->driver_action_data);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200467 data_ptr += DRIVER_ACTION_LENGTH_32_BIT_WORD;
468 break;
469 case DUMP_SHRAM:
470 LOG_INFO("ethosu_invoke DUMP_SHRAM\n");
Anton Moberg61da4d32020-12-22 16:00:31 +0100471 ret = dump_shram(drv);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200472 data_ptr += DRIVER_ACTION_LENGTH_32_BIT_WORD;
473 break;
474 case NOP:
475 LOG_INFO("ethosu_invoke NOP\n");
476 data_ptr += DRIVER_ACTION_LENGTH_32_BIT_WORD;
477 break;
478 default:
479 LOG_ERR("ethosu_invoke UNSUPPORTED driver_action_command %d \n", data_ptr->driver_action_command);
Bhavik Patele645fed2020-06-12 14:46:47 +0200480 ret = -1;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200481 break;
482 }
Bhavik Patele645fed2020-06-12 14:46:47 +0200483 if (ret != 0)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200484 {
Bhavik Patele645fed2020-06-12 14:46:47 +0200485 return_code = -1;
486 break;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200487 }
488 }
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200489
Anton Moberg61da4d32020-12-22 16:00:31 +0100490 if (!drv->status_error && !drv->dev_power_always_on)
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200491 {
Anton Moberg61da4d32020-12-22 16:00:31 +0100492 ethosu_save_pmu_counters(&drv->dev);
Anton Moberg0a614292021-03-24 14:08:22 +0100493 set_clock_and_power_request(drv, ETHOSU_INFERENCE_REQUEST, ETHOSU_CLOCK_Q_ENABLE, ETHOSU_POWER_Q_ENABLE);
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200494 }
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200495
Bhavik Patele645fed2020-06-12 14:46:47 +0200496 return return_code;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200497}
498
Anton Moberg61da4d32020-12-22 16:00:31 +0100499void ethosu_abort_v2(struct ethosu_driver *drv)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200500{
Anton Moberg61da4d32020-12-22 16:00:31 +0100501 drv->abort_inference = true;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200502}
503
Anton Moberg61da4d32020-12-22 16:00:31 +0100504void ethosu_set_power_mode_v2(struct ethosu_driver *drv, bool always_on)
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100505{
Anton Moberg61da4d32020-12-22 16:00:31 +0100506 drv->dev_power_always_on = always_on;
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100507
508 if (always_on)
509 {
Anton Moberg61da4d32020-12-22 16:00:31 +0100510 npu_axi_init(drv);
511 }
512}
513
514int ethosu_register_driver(struct ethosu_driver *drv)
515{
516 // Safeguard check for if driver is already registered
517 struct ethosu_driver *cur = registered_drivers;
518 while (cur != NULL)
519 {
520 if (cur == drv)
521 {
522 LOG_ERR("%s: NPU driver at address %p is already registered.\n", __FUNCTION__, drv);
523 return -1;
524 }
525 cur = cur->next;
526 }
527
528 drv->next = registered_drivers;
529 // Designate new registered driver HEAD
530 registered_drivers = drv;
531
532 LOG_INFO("%s: New NPU driver at address %p is registered.\n", __FUNCTION__, drv);
Anton Moberg61da4d32020-12-22 16:00:31 +0100533 return 0;
534}
535
536int ethosu_deregister_driver(struct ethosu_driver *drv)
537{
538 struct ethosu_driver *cur = registered_drivers;
539 struct ethosu_driver **prev = &registered_drivers;
540
541 while (cur != NULL)
542 {
543 if (cur == drv)
544 {
545 *prev = cur->next;
546 LOG_INFO("%s: NPU driver at address %p is deregistered.\n", __FUNCTION__, drv);
547 return 0;
548 }
549
550 prev = &cur->next;
551 cur = cur->next;
552 }
553
554 LOG_ERR("%s: NPU driver at address %p does not match a registered driver and therefore may not be deregistered.\n",
555 __FUNCTION__,
556 drv);
Anton Mobergdf386e02021-02-02 11:26:48 +0100557
Anton Moberg61da4d32020-12-22 16:00:31 +0100558 return -1;
559}
560
561struct ethosu_driver *ethosu_reserve_driver(void)
562{
Anton Mobergdf386e02021-02-02 11:26:48 +0100563 struct ethosu_driver *drv = NULL;
564
565 do
566 {
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100567 ethosu_mutex_lock(ethosu_mutex);
Anton Mobergdf386e02021-02-02 11:26:48 +0100568 drv = ethosu_find_and_reserve_driver();
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100569 ethosu_mutex_unlock(ethosu_mutex);
Anton Mobergdf386e02021-02-02 11:26:48 +0100570
571 if (drv != NULL)
572 {
573 break;
574 }
575
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100576 LOG_INFO("%s - Waiting for driver \n", __FUNCTION__);
577 ethosu_semaphore_take(ethosu_semaphore);
Anton Mobergdf386e02021-02-02 11:26:48 +0100578
579 } while (1);
580
581 return drv;
582}
583
584static struct ethosu_driver *ethosu_find_and_reserve_driver(void)
585{
Anton Moberg61da4d32020-12-22 16:00:31 +0100586 struct ethosu_driver *drv = registered_drivers;
587
588 while (drv != NULL)
589 {
590 if (!drv->reserved)
591 {
592 drv->reserved = true;
593 LOG_INFO("%s - Driver %p reserved.\n", __FUNCTION__, drv);
594 return drv;
595 }
596 drv = drv->next;
597 }
598
599 LOG_INFO("%s: No available drivers.\n", __FUNCTION__, drv);
600
601 return NULL;
602}
603
604void ethosu_release_driver(struct ethosu_driver *drv)
605{
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100606 ethosu_mutex_lock(ethosu_mutex);
Anton Moberg61da4d32020-12-22 16:00:31 +0100607 if (drv != NULL && drv->reserved)
608 {
609 drv->reserved = false;
610 LOG_INFO("%s - Driver %p released\n", __FUNCTION__, drv);
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100611 ethosu_semaphore_give(ethosu_semaphore);
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100612 }
Anton Mobergdfed5fd2021-03-11 14:41:11 +0100613 ethosu_mutex_unlock(ethosu_mutex);
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100614}
615
616static int ethosu_soft_reset_and_restore(struct ethosu_driver *drv)
617{
618
619 if (ETHOSU_SUCCESS != ethosu_soft_reset(&drv->dev))
620 {
621 return -1;
622 }
623
Anton Moberg0a614292021-03-24 14:08:22 +0100624 set_clock_and_power_request(drv, ETHOSU_INFERENCE_REQUEST, ETHOSU_CLOCK_Q_ENABLE, ETHOSU_POWER_Q_DISABLE);
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100625
626 npu_axi_init(drv);
627 ethosu_restore_pmu_config(&drv->dev);
628
629 return 0;
630}
631
Anton Moberg0a614292021-03-24 14:08:22 +0100632enum ethosu_error_codes set_clock_and_power_request(struct ethosu_driver *drv,
633 enum ethosu_request_clients client,
634 enum ethosu_clock_q_request clock_request,
635 enum ethosu_power_q_request power_request)
636{
637 // Set clock request bit for client
638 if (clock_request == ETHOSU_CLOCK_Q_DISABLE)
639 {
640 drv->clock_request |= (1 << client);
641 }
642 else
643 {
644 drv->clock_request &= ~(1 << client);
645 }
646 // Get current clock request (ENABLE if both PMU and INFERENCE asks for clock request, else DISABLE)
647 clock_request = drv->clock_request == 0 ? ETHOSU_CLOCK_Q_ENABLE : ETHOSU_CLOCK_Q_DISABLE;
648
649 // Set power request bit for client
650 if (power_request == ETHOSU_CLOCK_Q_DISABLE)
651 {
652 drv->power_request |= (1 << client);
653 }
654 else
655 {
656 drv->power_request &= ~(1 << client);
657 }
658 // Get current power request (ENABLE if both PMU and INFERENCE asks for power request, else DISABLE)
659 power_request = drv->power_request == 0 ? ETHOSU_POWER_Q_ENABLE : ETHOSU_POWER_Q_DISABLE;
660 // Set clock and power
661 enum ethosu_error_codes ret = ethosu_set_clock_and_power(&drv->dev, clock_request, power_request);
662
663 return ret;
664}
665
Bhavik Pateldae5be02020-06-18 15:25:15 +0200666static int handle_optimizer_config(struct ethosu_driver *drv, struct opt_cfg_s *opt_cfg_p)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200667{
668 struct ethosu_config cfg;
669 struct ethosu_id id;
670 int return_code = 0;
671
672 LOG_INFO("handle_optimizer_config:\n");
673 LOG_INFO("Optimizer release nbr: %d patch: %d\n", opt_cfg_p->da_data.rel_nbr, opt_cfg_p->da_data.patch_nbr);
674 LOG_INFO("Optimizer config cmd_stream_version: %d macs_per_cc: %d shram_size: %d\n",
675 opt_cfg_p->cmd_stream_version,
676 opt_cfg_p->macs_per_cc,
677 opt_cfg_p->shram_size);
678 LOG_INFO("Optimizer config Ethos-U version: %d.%d.%d\n",
679 opt_cfg_p->arch_major_rev,
680 opt_cfg_p->arch_minor_rev,
681 opt_cfg_p->arch_patch_rev);
682
Bhavik Pateldae5be02020-06-18 15:25:15 +0200683 (void)ethosu_get_config(&drv->dev, &cfg);
684 (void)ethosu_get_id(&drv->dev, &id);
Per Åstrand14ccfee2020-09-25 10:40:20 +0200685 LOG_INFO("Ethos-U config cmd_stream_version: %" PRIu32 " macs_per_cc: %" PRIu32 " shram_size: %" PRIu32 "\n",
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200686 cfg.cmd_stream_version,
687 cfg.macs_per_cc,
688 cfg.shram_size);
Per Åstrand14ccfee2020-09-25 10:40:20 +0200689 LOG_INFO("Ethos-U version: %" PRIu32 ".%" PRIu32 ".%" PRIu32 "\n",
690 id.arch_major_rev,
691 id.arch_minor_rev,
692 id.arch_patch_rev);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200693
694 if ((cfg.macs_per_cc != opt_cfg_p->macs_per_cc) || (cfg.shram_size != opt_cfg_p->shram_size) ||
695 (cfg.cmd_stream_version != opt_cfg_p->cmd_stream_version))
696 {
697 if (cfg.macs_per_cc != opt_cfg_p->macs_per_cc)
698 {
Per Åstrand14ccfee2020-09-25 10:40:20 +0200699 LOG_ERR("NPU config mismatch: npu.macs_per_cc=%" PRIu32 " optimizer.macs_per_cc=%d\n",
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200700 cfg.macs_per_cc,
701 opt_cfg_p->macs_per_cc);
702 }
703 if (cfg.shram_size != opt_cfg_p->shram_size)
704 {
Per Åstrand14ccfee2020-09-25 10:40:20 +0200705 LOG_ERR("NPU config mismatch: npu.shram_size=%" PRIu32 " optimizer.shram_size=%d\n",
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200706 cfg.shram_size,
707 opt_cfg_p->shram_size);
708 }
709 if (cfg.cmd_stream_version != opt_cfg_p->cmd_stream_version)
710 {
Per Åstrand14ccfee2020-09-25 10:40:20 +0200711 LOG_ERR("NPU config mismatch: npu.cmd_stream_version=%" PRIu32 " optimizer.cmd_stream_version=%d\n",
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200712 cfg.cmd_stream_version,
713 opt_cfg_p->cmd_stream_version);
714 }
715 return_code = -1;
716 }
717
Douglas Troha91e0be52021-01-18 13:57:38 +0100718 if ((id.arch_major_rev != opt_cfg_p->arch_major_rev) || (id.arch_minor_rev < opt_cfg_p->arch_minor_rev))
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200719 {
Per Åstrand14ccfee2020-09-25 10:40:20 +0200720 LOG_ERR("NPU arch mismatch: npu.arch=%" PRIu32 ".%" PRIu32 ".%" PRIu32 " optimizer.arch=%d.%d.%d\n",
Bhavik Patel790ef362020-06-03 10:05:28 +0200721 id.arch_major_rev,
722 id.arch_minor_rev,
723 id.arch_patch_rev,
724 opt_cfg_p->arch_major_rev,
725 opt_cfg_p->arch_minor_rev,
726 opt_cfg_p->arch_patch_rev);
727 return_code = -1;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200728 }
729
730#if !defined(LOG_ENABLED)
731 UNUSED(opt_cfg_p);
732#endif
733 return return_code;
734}
735
Bhavik Pateldae5be02020-06-18 15:25:15 +0200736static void npu_axi_init(struct ethosu_driver *drv)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200737{
Bhavik Pateldae5be02020-06-18 15:25:15 +0200738 ethosu_set_qconfig(&drv->dev, NPU_QCONFIG);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200739
Bhavik Pateldae5be02020-06-18 15:25:15 +0200740 ethosu_set_regioncfg(&drv->dev, 0, NPU_REGIONCFG_0);
741 ethosu_set_regioncfg(&drv->dev, 1, NPU_REGIONCFG_1);
742 ethosu_set_regioncfg(&drv->dev, 2, NPU_REGIONCFG_2);
743 ethosu_set_regioncfg(&drv->dev, 3, NPU_REGIONCFG_3);
744 ethosu_set_regioncfg(&drv->dev, 4, NPU_REGIONCFG_4);
745 ethosu_set_regioncfg(&drv->dev, 5, NPU_REGIONCFG_5);
746 ethosu_set_regioncfg(&drv->dev, 6, NPU_REGIONCFG_6);
747 ethosu_set_regioncfg(&drv->dev, 7, NPU_REGIONCFG_7);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200748
Bhavik Pateldae5be02020-06-18 15:25:15 +0200749 (void)ethosu_set_axi_limit0(&drv->dev,
750 AXI_LIMIT0_MAX_BEATS_BYTES,
Bhavik Patel790ef362020-06-03 10:05:28 +0200751 AXI_LIMIT0_MEM_TYPE,
752 AXI_LIMIT0_MAX_OUTSTANDING_READS,
753 AXI_LIMIT0_MAX_OUTSTANDING_WRITES);
Bhavik Pateldae5be02020-06-18 15:25:15 +0200754 (void)ethosu_set_axi_limit1(&drv->dev,
755 AXI_LIMIT1_MAX_BEATS_BYTES,
Bhavik Patel790ef362020-06-03 10:05:28 +0200756 AXI_LIMIT1_MEM_TYPE,
757 AXI_LIMIT1_MAX_OUTSTANDING_READS,
758 AXI_LIMIT1_MAX_OUTSTANDING_WRITES);
Bhavik Pateldae5be02020-06-18 15:25:15 +0200759 (void)ethosu_set_axi_limit2(&drv->dev,
760 AXI_LIMIT2_MAX_BEATS_BYTES,
Bhavik Patel790ef362020-06-03 10:05:28 +0200761 AXI_LIMIT2_MEM_TYPE,
762 AXI_LIMIT2_MAX_OUTSTANDING_READS,
763 AXI_LIMIT2_MAX_OUTSTANDING_WRITES);
Bhavik Pateldae5be02020-06-18 15:25:15 +0200764 (void)ethosu_set_axi_limit3(&drv->dev,
765 AXI_LIMIT3_MAX_BEATS_BYTES,
Bhavik Patel790ef362020-06-03 10:05:28 +0200766 AXI_LIMIT3_MEM_TYPE,
767 AXI_LIMIT3_MAX_OUTSTANDING_READS,
768 AXI_LIMIT3_MAX_OUTSTANDING_WRITES);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200769}
770
Per Åstrand3c8afcc2020-10-20 10:29:59 +0200771/* Default implementation to flush the data cache. Override if available on the targeted device.
772 * Passing NULL as p argument expects the whole cache to be flushed.
773 */
774void __attribute__((weak)) ethosu_flush_dcache(uint32_t *p, size_t bytes)
775{
776 (void)p;
777 (void)bytes;
778}
779
780/* Default implementation to invalidate the data cache. Override if available on the targeted device.
781 * Passing NULL as p argument expects the whole cache to be flushed.
782 */
783void __attribute__((weak)) ethosu_invalidate_dcache(uint32_t *p, size_t bytes)
784{
785 (void)p;
786 (void)bytes;
787}
788
Bhavik Pateldae5be02020-06-18 15:25:15 +0200789static int handle_command_stream(struct ethosu_driver *drv,
790 const uint8_t *cmd_stream,
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200791 const int cms_length,
792 const uint64_t *base_addr,
Per Åstrand3c8afcc2020-10-20 10:29:59 +0200793 const size_t *base_addr_size,
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200794 const int num_base_addr)
795{
Kristofer Jonssonc6e7a1f2020-11-24 09:20:14 +0100796 uint32_t qread = 0;
797 uint32_t cms_bytes = cms_length * BYTES_IN_32_BITS;
798 ptrdiff_t cmd_stream_ptr = (ptrdiff_t)cmd_stream;
799
Kristofer Jonsson125429a2020-08-20 16:52:23 +0200800 LOG_INFO("handle_command_stream: cmd_stream=%p, cms_length %d\n", cmd_stream, cms_length);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200801
Bhavik Patelbf7ae632020-06-11 21:00:16 +0200802 if (0 != ((ptrdiff_t)cmd_stream & MASK_16_BYTE_ALIGN))
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200803 {
Bhavik Patelbf7ae632020-06-11 21:00:16 +0200804 LOG_ERR("Error: Command stream addr %p not aligned to 16 bytes\n", cmd_stream);
805 return -1;
806 }
807
808 bool base_addr_invalid = false;
809 for (int i = 0; i < num_base_addr; i++)
810 {
811 if (0 != (base_addr[i] & MASK_16_BYTE_ALIGN))
812 {
Per Åstrand14ccfee2020-09-25 10:40:20 +0200813 LOG_ERR("Error: Base addr %d: 0x%llx not aligned to 16 bytes\n", i, base_addr[i]);
Bhavik Patelbf7ae632020-06-11 21:00:16 +0200814 base_addr_invalid = true;
815 }
816 }
Kristofer Jonssonc6e7a1f2020-11-24 09:20:14 +0100817
Bhavik Patelbf7ae632020-06-11 21:00:16 +0200818 if (base_addr_invalid)
819 {
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200820 return -1;
821 }
Kristofer Jonssonc6e7a1f2020-11-24 09:20:14 +0100822
Per Åstrand3c8afcc2020-10-20 10:29:59 +0200823 /* Flush the cache if available on our CPU.
824 * The upcasting to uin32_t* is ok since the pointer never is dereferenced.
825 * The base_addr_size is null if invoking from prior to invoke_V2, in that case
826 * the whole cache is being flushed.
827 */
828
829 if (base_addr_size != NULL)
830 {
Kristofer Jonssonc6e7a1f2020-11-24 09:20:14 +0100831 ethosu_flush_dcache((uint32_t *)cmd_stream_ptr, cms_bytes);
Per Åstrand3c8afcc2020-10-20 10:29:59 +0200832 for (int i = 0; i < num_base_addr; i++)
833 {
Bhavik Patel033bb1b2020-12-17 15:33:33 +0100834 ethosu_flush_dcache((uint32_t *)(uintptr_t)base_addr[i], base_addr_size[i]);
Per Åstrand3c8afcc2020-10-20 10:29:59 +0200835 }
836 }
837 else
838 {
839 ethosu_flush_dcache(NULL, 0);
840 }
841
Bhavik Pateldae5be02020-06-18 15:25:15 +0200842 if (ETHOSU_SUCCESS != ethosu_run_command_stream(&drv->dev, cmd_stream, cms_bytes, base_addr, num_base_addr))
Bhavik Patel790ef362020-06-03 10:05:28 +0200843 {
844 return -1;
845 }
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200846
Bhavik Pateldae5be02020-06-18 15:25:15 +0200847 wait_for_irq(drv);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200848
Bhavik Patel5f8dad12020-09-30 09:06:52 +0200849 if (drv->status_error)
850 {
851 return -1;
852 }
853
Per Åstrand3c8afcc2020-10-20 10:29:59 +0200854 if (base_addr_size != NULL)
855 {
856 for (int i = 0; i < num_base_addr; i++)
857 {
Bhavik Patel033bb1b2020-12-17 15:33:33 +0100858 ethosu_invalidate_dcache((uint32_t *)(uintptr_t)base_addr[i], base_addr_size[i]);
Per Åstrand3c8afcc2020-10-20 10:29:59 +0200859 }
860 }
861 else
862 {
863 ethosu_invalidate_dcache(NULL, 0);
864 }
865
Bhavik Pateldae5be02020-06-18 15:25:15 +0200866 (void)ethosu_get_qread(&drv->dev, &qread);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200867 if (qread != cms_bytes)
868 {
Kristofer Jonsson4dc73dc2020-10-16 12:33:47 +0200869 LOG_WARN(
Per Åstrand14ccfee2020-09-25 10:40:20 +0200870 "Failure: IRQ received but qread (%" PRIu32 ") not at end of stream (%" PRIu32 ").\n", qread, cms_bytes);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200871 return -1;
872 }
873
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200874 return 0;
875}
876
Bhavik Pateldae5be02020-06-18 15:25:15 +0200877static int read_apb_reg(struct ethosu_driver *drv, uint16_t da_data)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200878{
879 uint32_t *reg_p;
880 uint32_t start_address = (uint32_t)(da_data & APB_START_ADDR_MASK);
881 uint16_t num_reg = (da_data >> APB_NUM_REG_BIT_SHIFT) + 1;
882
883 reg_p = (uint32_t *)malloc(num_reg * sizeof(uint32_t));
884 if (reg_p == NULL)
885 {
886 LOG_INFO("read_apb_reg, Error! memory not allocated.");
887 return -1;
888 }
889
Bhavik Pateldae5be02020-06-18 15:25:15 +0200890 if (ETHOSU_SUCCESS == ethosu_read_apb_reg(&drv->dev, start_address, num_reg, reg_p))
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200891 {
892 for (int i = 0; i < num_reg; i++)
893 {
Per Åstrand14ccfee2020-09-25 10:40:20 +0200894 LOG_INFO(
895 "NPU_REG ADDR 0x%04" PRIu32 " = 0x%08" PRIu32 "\n", (start_address + (i * BYTES_IN_32_BITS)), reg_p[i]);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200896 }
897 }
898 else
899 {
900 free(reg_p);
901 return -1;
902 }
903
904 free(reg_p);
905 return 0;
906}
907
Bhavik Pateldae5be02020-06-18 15:25:15 +0200908static int dump_shram(struct ethosu_driver *drv)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200909{
910 struct ethosu_config cfg;
911 uint32_t *shram_p;
Bhavik Pateldae5be02020-06-18 15:25:15 +0200912 (void)ethosu_get_config(&drv->dev, &cfg);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200913
Per Åstrand14ccfee2020-09-25 10:40:20 +0200914 LOG_INFO("dump_shram size = %" PRIu32 " KB\n", cfg.shram_size);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200915
916 shram_p = (uint32_t *)malloc(BYTES_1KB);
917 if (shram_p == NULL)
918 {
919 LOG_ERR("read_shram, Error! memory not allocated.");
920 return -1;
921 }
922
923 for (uint32_t i = 0; i < cfg.shram_size; i++)
924 {
Bhavik Pateldae5be02020-06-18 15:25:15 +0200925 ethosu_get_shram_data(&drv->dev, i, (uint32_t *)shram_p);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200926 // Output 1KB of SHRAM
Per Åstrand14ccfee2020-09-25 10:40:20 +0200927 LOG_INFO("***SHRAM SECTION %" PRIu32 "***\n", i);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200928 for (int j = 0; j < (BYTES_1KB / BYTES_IN_32_BITS); j++)
929 {
Per Åstrand14ccfee2020-09-25 10:40:20 +0200930 LOG_INFO("[0x%04" PRIx32 "] %" PRIx32 "\n", (i * 1024 + j * 4), shram_p[j]);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200931 }
932 }
933 free(shram_p);
934
935 return 0;
936}
937
938typedef struct
939{
940 int number;
941 const char *name;
942} name_lookup_t;
943
944static const name_lookup_t npu_reg_name_tbl[] = {
945 {0x200, "KERNEL_X"},
946 {0x204, "KERNEL_Y"},
947 {0x208, "KERNEL_W_M1"},
948 {0x20C, "KERNEL_H_M1"},
949 {0x210, "OFM_CBLK_WIDTH_M1"},
950 {0x214, "OFM_CBLK_HEIGHT_M1"},
951 {0x218, "OFM_CBLK_DEPTH_M1"},
952 {0x21c, "IFM_CBLK_DEPTH_M1"},
953 {0x220, "OFM_X"},
954 {0x224, "OFM_Y"},
955 {0x228, "OFM_Z"},
956 {0x22C, "IFM_Z"},
957 {0x230, "PAD_TOP"},
958 {0x234, "PAD_LEFT"},
959 {0x238, "IFM_CBLK_WIDTH"},
960 {0x23C, "IFM_CBLK_HEIGHT"},
961 {0x240, "DMA_IFM_SRC"},
962 {0x244, "DMA_IFM_SRC_HI"},
963 {0x248, "DMA_IFM_DST"},
964 {0x24c, "DMA_OFM_SRC"},
965 {0x250, "DMA_OFM_DST"},
966 {0x254, "DMA_OFM_DST_HI"},
967 {0x258, "DMA_WEIGHT_SRC"},
968 {0x25c, "DMA_WEIGHT_SRC_HI"},
969 {0x260, "DMA_CMD_SRC"},
970 {0x264, "DMA_CMD_SRC_HI"},
971 {0x268, "DMA_CMD_SIZE"},
972 {0x26c, "DMA_M2M_SRC"},
973 {0x270, "DMA_M2M_SRC_HI"},
974 {0x274, "DMA_M2M_DST"},
975 {0x278, "DMA_M2M_DST_HI"},
976 {0x27c, "CURRENT_QREAD"},
977 {0x280, "DMA_SCALE_SRC"},
978 {0x284, "DMA_SCALE_SRC_HI"},
979 {0x2BC, "CURRENT_CMD"},
980 {0x800, "IFM_PAD_TOP"},
981 {0x804, "IFM_PAD_LEFT"},
982 {0x808, "IFM_PAD_RIGHT"},
983 {0x80C, "IFM_PAD_BOTTOM"},
984 {0x810, "IFM_DEPTH_M1"},
985 {0x814, "IFM_PRECISION"},
986 {0x81C, "IFM_UPSCALE"},
987 {0x824, "IFM_ZERO_POINT"},
988 {0x828, "IFM_WIDTH0_M1"},
989 {0x82C, "IFM_HEIGHT0_M1"},
990 {0x830, "IFM_HEIGHT1_M1"},
991 {0x834, "IFM_IB_END"},
992 {0x83C, "IFM_REGION"},
993 {0x844, "OFM_WIDTH_M1"},
994 {0x848, "OFM_HEIGHT_M1"},
995 {0x84C, "OFM_DEPTH_M1"},
996 {0x850, "OFM_PRECISION"},
997 {0x854, "OFM_BLK_WIDTH_M1"},
998 {0x858, "OFM_BLK_HEIGHT_M1"},
999 {0x85C, "OFM_BLK_DEPTH_M1"},
1000 {0x860, "OFM_ZERO_POINT"},
1001 {0x868, "OFM_WIDTH0_M1"},
1002 {0x86C, "OFM_HEIGHT0_M1"},
1003 {0x870, "OFM_HEIGHT1_M1"},
1004 {0x87C, "OFM_REGION"},
1005 {0x880, "KERNEL_WIDTH_M1"},
1006 {0x884, "KERNEL_HEIGHT_M1"},
1007 {0x888, "KERNEL_STRIDE"},
1008 {0x88C, "PARALLEL_MODE"},
1009 {0x890, "ACC_FORMAT"},
1010 {0x894, "ACTIVATION"},
1011 {0x898, "ACTIVATION_MIN"},
1012 {0x89C, "ACTIVATION_MAX"},
1013 {0x8A0, "WEIGHT_REGION"},
1014 {0x8A4, "SCALE_REGION"},
1015 {0x8B4, "AB_START"},
1016 {0x8BC, "BLOCKDEP"},
1017 {0x8C0, "DMA0_SRC_REGION"},
1018 {0x8C4, "DMA0_DST_REGION"},
1019 {0x8C8, "DMA0_SIZE0"},
1020 {0x8CC, "DMA0_SIZE1"},
1021 {0x900, "IFM2_BROADCAST"},
1022 {0x904, "IFM2_SCALAR"},
1023 {0x924, "IFM2_ZERO_POINT"},
1024 {0x928, "IFM2_WIDTH0_M1"},
1025 {0x92C, "IFM2_HEIGHT0_M1"},
1026 {0x930, "IFM2_HEIGHT1_M1"},
1027 {0x934, "IFM2_IB_START"},
1028 {0x93C, "IFM2_REGION"},
1029 {0xA00, "IFM_BASE0"},
1030 {0xA04, "IFM_BASE0_HI"},
1031 {0xA08, "IFM_BASE1"},
1032 {0xA0C, "IFM_BASE1_HI"},
1033 {0xA10, "IFM_BASE2"},
1034 {0xA14, "IFM_BASE2_HI"},
1035 {0xA18, "IFM_BASE3"},
1036 {0xA1C, "IFM_BASE3_HI"},
1037 {0xA20, "IFM_STRIDE_X"},
1038 {0xA24, "IFM_STRIDE_X_HI"},
1039 {0xA28, "IFM_STRIDE_Y"},
1040 {0xA2C, "IFM_STRIDE_Y_HI"},
1041 {0xA30, "IFM_STRIDE_C"},
1042 {0xA34, "IFM_STRIDE_C_HI"},
1043 {0xA40, "OFM_BASE0"},
1044 {0xA44, "OFM_BASE0_HI"},
1045 {0xA48, "OFM_BASE1"},
1046 {0xA4C, "OFM_BASE1_HI"},
1047 {0xA50, "OFM_BASE2"},
1048 {0xA54, "OFM_BASE2_HI"},
1049 {0xA58, "OFM_BASE3"},
1050 {0xA5C, "OFM_BASE3_HI"},
1051 {0xA60, "OFM_STRIDE_X"},
1052 {0xA64, "OFM_STRIDE_X_HI"},
1053 {0xA68, "OFM_STRIDE_Y"},
1054 {0xA6C, "OFM_STRIDE_Y_HI"},
1055 {0xA70, "OFM_STRIDE_C"},
1056 {0xA74, "OFM_STRIDE_C_HI"},
1057 {0xA80, "WEIGHT_BASE"},
1058 {0xA84, "WEIGHT_BASE_HI"},
1059 {0xA88, "WEIGHT_LENGTH"},
1060 {0xA8C, "WEIGHT_LENGTH_HI"},
1061 {0xA90, "SCALE_BASE"},
1062 {0xA94, "SCALE_BASE_HI"},
1063 {0xA98, "SCALE_LENGTH"},
1064 {0xAA0, "OFM_SCALE"},
1065 {0xAA4, "OFM_SCALE_SHIFT"},
1066 {0xAA8, "OPA_SCALE "},
1067 {0xAB0, "OPB_SCALE"},
1068 {0xAC0, "DMA0_SRC"},
1069 {0xAC4, "DMA0_SRC_HI"},
1070 {0xAC8, "DMA0_DST"},
1071 {0xACC, "DMA0_DST_HI"},
1072 {0xAD0, "DMA0_LEN"},
1073 {0xAD4, "DMA0_LEN_HI"},
1074 {0xAD8, "DMA0_SKIP0"},
1075 {0xADC, "DMA0_SKIP0_HI"},
1076 {0xAE0, "DMA0_SKIP1"},
1077 {0xAE4, "DMA0_SKIP1_HI"},
1078 {0xB00, "IFM2_BASE0"},
1079 {0xB04, "IFM2_BASE0_HI"},
1080 {0xB08, "IFM2_BASE1"},
1081 {0xB0C, "IFM2_BASE1_HI"},
1082 {0xB10, "IFM2_BASE2"},
1083 {0xB14, "IFM2_BASE2_HI"},
1084 {0xB18, "IFM2_BASE3"},
1085 {0xB1C, "IFM2_BASE3_HI"},
1086 {0xB20, "IFM2_STRIDE_X"},
1087 {0xB24, "IFM2_STRIDE_X_HI"},
1088 {0xB28, "IFM2_STRIDE_Y"},
1089 {0xB2C, "IFM2_STRIDE_Y_HI"},
1090 {0xB30, "IFM2_STRIDE_C"},
1091 {0xB34, "IFM2_STRIDE_C_HI"},
1092 {0xB40, "WEIGHT1_BASE"},
1093 {0xB44, "WEIGHT1_BASE_HI"},
1094 {0xB48, "WEIGHT1_LENGTH"},
1095 {0xB4C, "WEIGHT1_LENGTH_HI"},
1096 {0xB50, "SCALE1_BASE"},
1097 {0xB54, "SCALE1_BASE_HI"},
1098 {0xB58, "SCALE1_LENGTH"},
1099};
1100
1101static const char *lookup_name(const name_lookup_t *lookup_table, int lookup_table_count, int find)
1102{
1103 int n;
1104 for (n = 0; n < lookup_table_count; n++)
1105 {
1106 if (lookup_table[n].number == find)
1107 {
1108 return lookup_table[n].name;
1109 }
1110 }
1111 // Not found
1112 return 0;
1113}
1114
Bhavik Pateldae5be02020-06-18 15:25:15 +02001115static void dump_npu_register(struct ethosu_driver *drv, int npu_reg, int npu_reg_end)
Kristofer Jonsson49bdee82020-04-06 13:21:21 +02001116{
1117 unsigned int reg_val;
1118 const char *reg_name;
1119 int npu_reg_name_tbl_count = sizeof(npu_reg_name_tbl) / sizeof(npu_reg_name_tbl[0]);
1120
1121 LOG_INFO("dump_register %X - %X\n", npu_reg, npu_reg_end);
1122 for (; npu_reg <= npu_reg_end; npu_reg += sizeof(int))
1123 {
Bhavik Pateldae5be02020-06-18 15:25:15 +02001124 reg_val = ethosu_read_reg(&drv->dev, npu_reg);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +02001125 reg_name = lookup_name(npu_reg_name_tbl, npu_reg_name_tbl_count, npu_reg);
1126 LOG_INFO("[0x%.4X] 0x%.8X\t%s\n", npu_reg, reg_val, (reg_name) ? reg_name : "");
1127 }
1128}
1129
1130static const name_lookup_t cmd0_name_tbl[] = {
1131 {0x000, "NPU_OP_STOP"},
1132 {0x001, "NPU_OP_IRQ"},
1133 {0x002, "NPU_OP_CONV"},
1134 {0x003, "NPU_OP_DEPTHWISE"},
1135 {0x004, "NPU_OP_VECTOR_PROD"},
1136 {0x005, "NPU_OP_POOL"},
1137 {0x006, "NPU_OP_ELEMENTWISE"},
1138 {0x010, "NPU_OP_DMA_START"},
1139 {0x011, "NPU_OP_DMA_WAIT"},
1140 {0x012, "NPU_OP_KERNEL_WAIT"},
1141 {0x100, "NPU_SET_IFM_PAD_TOP"},
1142 {0x101, "NPU_SET_IFM_PAD_LEFT"},
1143 {0x102, "NPU_SET_IFM_PAD_RIGHT"},
1144 {0x103, "NPU_SET_IFM_PAD_BOTTOM"},
1145 {0x104, "NPU_SET_IFM_DEPTH_M1"},
1146 {0x105, "NPU_SET_IFM_PRECISION"},
1147 {0x107, "NPU_SET_IFM_UPSCALE"},
1148 {0x109, "NPU_SET_IFM_ZERO_POINT"},
1149 {0x10A, "NPU_SET_IFM_WIDTH0_M1"},
1150 {0x10B, "NPU_SET_IFM_HEIGHT0_M1"},
1151 {0x10C, "NPU_SET_IFM_HEIGHT1_M1"},
1152 {0x10D, "NPU_SET_IFM_IB_END"},
1153 {0x10F, "NPU_SET_IFM_REGION"},
1154 {0x110, "NPU_SET_OFM_BATCH_SIZE_M1"},
1155 {0x111, "NPU_SET_OFM_WIDTH_M1"},
1156 {0x112, "NPU_SET_OFM_HEIGHT_M1"},
1157 {0x113, "NPU_SET_OFM_DEPTH_M1"},
1158 {0x114, "NPU_SET_OFM_PRECISION"},
1159 {0x115, "NPU_SET_OFM_BLK_WIDTH_M1"},
1160 {0x116, "NPU_SET_OFM_BLK_HEIGHT_M1"},
1161 {0x117, "NPU_SET_OFM_BLK_DEPTH_M1"},
1162 {0x118, "NPU_SET_OFM_ZERO_POINT"},
1163 {0x11A, "NPU_SET_OFM_WIDTH0_M1"},
1164 {0x11B, "NPU_SET_OFM_HEIGHT0_M1"},
1165 {0x11C, "NPU_SET_OFM_HEIGHT1_M1"},
1166 {0x11F, "NPU_SET_OFM_REGION"},
1167 {0x120, "NPU_SET_KERNEL_WIDTH_M1"},
1168 {0x121, "NPU_SET_KERNEL_HEIGHT_M1"},
1169 {0x122, "NPU_SET_KERNEL_STRIDE"},
1170 {0x124, "NPU_SET_ACC_FORMAT"},
1171 {0x125, "NPU_SET_ACTIVATION"},
1172 {0x126, "NPU_SET_ACTIVATION_MIN"},
1173 {0x127, "NPU_SET_ACTIVATION_MAX"},
1174 {0x128, "NPU_SET_WEIGHT_REGION"},
1175 {0x129, "NPU_SET_SCALE_REGION"},
1176 {0x12D, "NPU_SET_AB_START"},
1177 {0x12F, "NPU_SET_BLOCKDEP"},
1178 {0x130, "NPU_SET_DMA0_SRC_REGION"},
1179 {0x131, "NPU_SET_DMA0_DST_REGION"},
1180 {0x180, "NPU_SET_IFM2_BROADCAST"},
1181 {0x181, "NPU_SET_IFM2_SCALAR"},
1182 {0x185, "NPU_SET_IFM2_PRECISION"},
1183 {0x189, "NPU_SET_IFM2_ZERO_POINT"},
1184 {0x18A, "NPU_SET_IFM2_WIDTH0_M1"},
1185 {0x18B, "NPU_SET_IFM2_HEIGHT0_M1"},
1186 {0x18C, "NPU_SET_IFM2_HEIGHT1_M1"},
1187 {0x18D, "NPU_SET_IFM2_IB_START"},
1188 {0x18F, "NPU_SET_IFM2_REGION"},
1189};
1190
1191static const name_lookup_t cmd1_name_tbl[] = {
1192 {0x000, "NPU_SET_IFM_BASE0"}, {0x001, "NPU_SET_IFM_BASE1"}, {0x002, "NPU_SET_IFM_BASE2"},
1193 {0x003, "NPU_SET_IFM_BASE3"}, {0x004, "NPU_SET_IFM_STRIDE_X"}, {0x005, "NPU_SET_IFM_STRIDE_Y"},
1194 {0x006, "NPU_SET_IFM_STRIDE_C"}, {0x007, "NPU_SET_IFM_STRIDE_N"}, {0x010, "NPU_SET_OFM_BASE0"},
1195 {0x011, "NPU_SET_OFM_BASE1"}, {0x012, "NPU_SET_OFM_BASE2"}, {0x013, "NPU_SET_OFM_BASE3"},
1196 {0x014, "NPU_SET_OFM_STRIDE_X"}, {0x015, "NPU_SET_OFM_STRIDE_Y"}, {0x016, "NPU_SET_OFM_STRIDE_C"},
1197 {0x017, "NPU_SET_OFM_STRIDE_N"}, {0x020, "NPU_SET_WEIGHT_BASE"}, {0x021, "NPU_SET_WEIGHT_LENGTH"},
1198 {0x022, "NPU_SET_SCALE_BASE"}, {0x023, "NPU_SET_SCALE_LENGTH"}, {0x024, "NPU_SET_OFM_SCALE"},
1199 {0x025, "NPU_SET_OPA_SCALE"}, {0x026, "NPU_SET_OPB_SCALE"}, {0x030, "NPU_SET_DMA0_SRC"},
1200 {0x031, "NPU_SET_DMA0_DST"}, {0x032, "NPU_SET_DMA0_LEN"}, {0x080, "NPU_SET_IFM2_BASE0"},
1201 {0x081, "NPU_SET_IFM2_BASE1"}, {0x082, "NPU_SET_IFM2_BASE2"}, {0x083, "NPU_SET_IFM2_BASE3"},
1202 {0x084, "NPU_SET_IFM2_STRIDE_X"}, {0x085, "NPU_SET_IFM2_STRIDE_Y"}, {0x086, "NPU_SET_IFM2_STRIDE_C"},
1203};
1204
1205static void dump_command_stream(const uint32_t *cmd_stream, const int cms_length, int qread)
1206{
1207 int n;
1208 int offset;
1209 uint32_t cmd_val;
1210 const uint8_t *cmd_ptr;
1211 const char *cmd_name;
1212 int cmd0_name_tbl_count = sizeof(cmd0_name_tbl) / sizeof(cmd0_name_tbl[0]);
1213 int cmd1_name_tbl_count = sizeof(cmd1_name_tbl) / sizeof(cmd1_name_tbl[0]);
1214
1215 LOG_INFO("dump_command_stream cmd_stream = 0x%8p cms_length = %d\n", cmd_stream, cms_length);
1216 for (n = 0; n < cms_length; n++)
1217 {
1218 // Offset
1219 offset = n * sizeof(int);
1220 LOG_INFO("[%.4d] ", offset);
1221 // Command
1222 cmd_ptr = (const uint8_t *)&cmd_stream[n];
1223 LOG_INFO("0x%.2X 0x%.2X 0x%.2X 0x%.2X ", cmd_ptr[0], cmd_ptr[1], cmd_ptr[2], cmd_ptr[3]);
1224 // Command name and payload
1225 if (cmd_stream[n] & 0x4000)
1226 {
1227 cmd_name = lookup_name(cmd1_name_tbl, cmd1_name_tbl_count, cmd_stream[n] & 0x3FF);
1228 n++;
1229 cmd_val = cmd_stream[n];
1230 cmd_ptr = (const uint8_t *)&cmd_stream[n];
1231 LOG_INFO("0x%.2X 0x%.2X 0x%.2X 0x%.2X ", cmd_ptr[0], cmd_ptr[1], cmd_ptr[2], cmd_ptr[3]);
1232 }
1233 else
1234 {
1235 cmd_val = cmd_stream[n] >> 16;
1236 cmd_name = lookup_name(cmd0_name_tbl, cmd0_name_tbl_count, cmd_stream[n] & 0x3FF);
1237 }
1238 if (cmd_name)
1239 {
Per Åstrand14ccfee2020-09-25 10:40:20 +02001240 LOG_INFO("\t%s 0x%.8" PRIX32, cmd_name, cmd_val);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +02001241 }
1242 if (offset == qread)
1243 {
1244 LOG_INFO(" <<== QREAD\n");
1245 }
1246 else
1247 {
1248 LOG_INFO("\n");
1249 }
1250 }
1251}