blob: ff8d1eb024a3afe2f62a13d04f3ae841c4ab7075 [file] [log] [blame]
Kristofer Jonsson49bdee82020-04-06 13:21:21 +02001/*
Kristofer Jonssonb3cde3c2022-01-27 17:30:15 +01002 * Copyright (c) 2019-2022 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 Jonsson3c439172020-08-05 09:38:40 +020019#ifndef ETHOSU_DRIVER_H
20#define ETHOSU_DRIVER_H
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020021
22/******************************************************************************
23 * Includes
24 ******************************************************************************/
25
Jonny Svärd136810f2021-10-13 16:04:26 +020026#include "ethosu_types.h"
Bhavik Pateldae5be02020-06-18 15:25:15 +020027
28#include <stdbool.h>
Kristofer Jonsson2b201c32020-09-02 16:42:43 +020029#include <stddef.h>
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020030#include <stdint.h>
31
32#ifdef __cplusplus
33extern "C" {
34#endif
35
36/******************************************************************************
Jonny Svärda830f172021-06-07 16:57:00 +020037 * Defines
38 ******************************************************************************/
39
40#define ETHOSU_DRIVER_VERSION_MAJOR 0 ///< Driver major version
41#define ETHOSU_DRIVER_VERSION_MINOR 16 ///< Driver minor version
42#define ETHOSU_DRIVER_VERSION_PATCH 0 ///< Driver patch version
43
44/******************************************************************************
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020045 * Types
46 ******************************************************************************/
47
Jonny Svärd136810f2021-10-13 16:04:26 +020048// Forward declare
49struct ethosu_device;
50
Bhavik Pateldae5be02020-06-18 15:25:15 +020051struct ethosu_driver
52{
Jonny Svärd136810f2021-10-13 16:04:26 +020053 struct ethosu_device *dev;
Jonny Svärda830f172021-06-07 16:57:00 +020054 struct ethosu_driver *next;
55 void *semaphore;
Kristofer Jonsson2b201c32020-09-02 16:42:43 +020056 uint64_t fast_memory;
57 size_t fast_memory_size;
Bhavik Patel5f8dad12020-09-30 09:06:52 +020058 bool status_error;
Anton Moberg8d65b6f2020-12-21 09:37:18 +010059 bool dev_power_always_on;
Anton Moberg61da4d32020-12-22 16:00:31 +010060 bool reserved;
Anton Mobergdfed5fd2021-03-11 14:41:11 +010061 volatile bool irq_triggered;
Anton Moberg0a614292021-03-24 14:08:22 +010062 uint8_t clock_request;
63 uint8_t power_request;
Bhavik Pateldae5be02020-06-18 15:25:15 +020064};
65
Jonny Svärda830f172021-06-07 16:57:00 +020066struct ethosu_driver_version
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020067{
Jonny Svärda830f172021-06-07 16:57:00 +020068 uint8_t major;
69 uint8_t minor;
70 uint8_t patch;
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020071};
72
Anton Moberg0a614292021-03-24 14:08:22 +010073enum ethosu_request_clients
74{
75 ETHOSU_PMU_REQUEST = 0,
76 ETHOSU_INFERENCE_REQUEST = 1,
77};
78
Kristofer Jonsson49bdee82020-04-06 13:21:21 +020079/******************************************************************************
Jonny Svärda830f172021-06-07 16:57:00 +020080 * Prototypes (weak functions in driver)
Kristofer Jonsson4dc73dc2020-10-16 12:33:47 +020081 ******************************************************************************/
82
Jonny Svärda830f172021-06-07 16:57:00 +020083/**
84 * Interrupt handler to be called on IRQ from Ethos-U
85 */
86void ethosu_irq_handler(struct ethosu_driver *drv);
87
88/*
89 * Flush/clean the data cache by address and size. Passing NULL as p argument
90 * expects the whole cache to be flushed.
91 */
92
93void ethosu_flush_dcache(uint32_t *p, size_t bytes);
94/*
95 * Invalidate the data cache by address and size. Passing NULL as p argument
96 * expects the whole cache to be invalidated.
97 */
98void ethosu_invalidate_dcache(uint32_t *p, size_t bytes);
99
100/*
101 * Minimal sempahore and mutex implementation for baremetal applications. See
102 * ethosu_driver.c.
103 */
104void *ethosu_mutex_create(void);
105void ethosu_mutex_lock(void *mutex);
106void ethosu_mutex_unlock(void *mutex);
107void *ethosu_semaphore_create(void);
108void ethosu_semaphore_take(void *sem);
109void ethosu_semaphore_give(void *sem);
110
111/*
Kristofer Jonssonb3cde3c2022-01-27 17:30:15 +0100112 * Callbacks for begin/end of inference. user_data pointer is passed to the
113 * ethosu_invoke() call and forwarded to the callback functions.
Jonny Svärda830f172021-06-07 16:57:00 +0200114 */
Kristofer Jonssonb3cde3c2022-01-27 17:30:15 +0100115void ethosu_inference_begin(struct ethosu_driver *drv, void *user_arg);
116void ethosu_inference_end(struct ethosu_driver *drv, void *user_arg);
Kristofer Jonsson4dc73dc2020-10-16 12:33:47 +0200117
118/******************************************************************************
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200119 * Prototypes
120 ******************************************************************************/
121
122/**
123 * Initialize the Ethos-U driver.
124 */
Anton Mobergeffc7aa2021-05-03 09:25:06 +0200125int ethosu_init(struct ethosu_driver *drv,
126 const void *base_address,
127 const void *fast_memory,
128 const size_t fast_memory_size,
129 uint32_t secure_enable,
130 uint32_t privilege_enable);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200131
132/**
Anton Moberg547ca532021-06-14 09:43:53 +0200133 * Deinitialize the Ethos-U driver.
134 */
135void ethosu_deinit(struct ethosu_driver *drv);
136
137/**
Jonny Svärda830f172021-06-07 16:57:00 +0200138 * Get Ethos-U driver version.
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200139 */
Jonny Svärda830f172021-06-07 16:57:00 +0200140void ethosu_get_driver_version(struct ethosu_driver_version *ver);
141
142/**
143 * Get Ethos-U hardware information.
144 */
145void ethosu_get_hw_info(struct ethosu_driver *drv, struct ethosu_hw_info *hw);
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200146
147/**
148 * Invoke Vela command stream.
149 */
Kristofer Jonssonb3cde3c2022-01-27 17:30:15 +0100150int ethosu_invoke_v3(struct ethosu_driver *drv,
151 const void *custom_data_ptr,
152 const int custom_data_size,
153 const uint64_t *base_addr,
154 const size_t *base_addr_size,
155 const int num_base_addr,
156 void *user_arg);
157
158#define ethosu_invoke(drv, custom_data_ptr, custom_data_size, base_addr, base_addr_size, num_base_addr) \
159 ethosu_invoke_v3(drv, custom_data_ptr, custom_data_size, base_addr, base_addr_size, num_base_addr, 0)
Kristofer Jonsson2b201c32020-09-02 16:42:43 +0200160
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200161/**
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100162 * Set Ethos-U power mode.
163 */
Anton Mobergeffc7aa2021-05-03 09:25:06 +0200164void ethosu_set_power_mode(struct ethosu_driver *drv, bool always_on);
Anton Moberg61da4d32020-12-22 16:00:31 +0100165
166/**
Anton Mobergdf386e02021-02-02 11:26:48 +0100167 * Reserves a driver to execute inference with
Anton Moberg61da4d32020-12-22 16:00:31 +0100168 */
169struct ethosu_driver *ethosu_reserve_driver(void);
170
171/**
172 * Change driver status to available
173 */
174void ethosu_release_driver(struct ethosu_driver *drv);
Anton Moberg8d65b6f2020-12-21 09:37:18 +0100175
Anton Mobergdf386e02021-02-02 11:26:48 +0100176/**
Anton Moberg0a614292021-03-24 14:08:22 +0100177 * Set clock and power request bits
178 */
179enum ethosu_error_codes set_clock_and_power_request(struct ethosu_driver *drv,
180 enum ethosu_request_clients client,
181 enum ethosu_clock_q_request clock_request,
182 enum ethosu_power_q_request power_request);
183
Kristofer Jonssonbe0bc462021-06-28 11:44:42 +0000184/**
185 * Static inline for backwards-compatibility
186 */
187static inline int ethosu_invoke_v2(const void *custom_data_ptr,
188 const int custom_data_size,
189 const uint64_t *base_addr,
190 const size_t *base_addr_size,
191 const int num_base_addr)
192{
193 struct ethosu_driver *drv = ethosu_reserve_driver();
Kristofer Jonssonb3cde3c2022-01-27 17:30:15 +0100194 int result = ethosu_invoke_v3(drv, custom_data_ptr, custom_data_size, base_addr, base_addr_size, num_base_addr, 0);
Kristofer Jonssonbe0bc462021-06-28 11:44:42 +0000195 ethosu_release_driver(drv);
196 return result;
197}
198
Kristofer Jonsson49bdee82020-04-06 13:21:21 +0200199#ifdef __cplusplus
200}
201#endif
Kristofer Jonsson3c439172020-08-05 09:38:40 +0200202
203#endif // ETHOSU_DRIVER_H