1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
/*
* Copyright (c) 2019-2020 Arm Limited. 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.
*/
#ifndef ETHOSU_DRIVER_H
#define ETHOSU_DRIVER_H
/******************************************************************************
* Includes
******************************************************************************/
#include "ethosu_device.h"
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#ifdef __cplusplus
extern "C" {
#endif
/******************************************************************************
* Types
******************************************************************************/
struct ethosu_driver
{
struct ethosu_device dev;
bool abort_inference;
uint64_t fast_memory;
size_t fast_memory_size;
bool status_error;
};
struct ethosu_version_id
{
// Ethos-U id
uint8_t version_status;
uint8_t version_minor;
uint8_t version_major;
uint8_t product_major;
uint8_t arch_patch_rev;
uint8_t arch_minor_rev;
uint8_t arch_major_rev;
// Driver Version
uint8_t driver_patch_rev;
uint8_t driver_minor_rev;
uint8_t driver_major_rev;
};
struct ethosu_version_config
{
uint8_t macs_per_cc;
uint8_t cmd_stream_version;
uint8_t shram_size;
};
struct ethosu_version
{
struct ethosu_version_id id;
struct ethosu_version_config cfg;
};
/******************************************************************************
* Variables
******************************************************************************/
extern struct ethosu_driver ethosu_drv;
/******************************************************************************
* Prototypes
******************************************************************************/
/**
* Initialize the Ethos-U driver.
*/
int ethosu_init_v3(const void *base_address,
const void *fast_memory,
const size_t fast_memory_size,
uint32_t secure_enable,
uint32_t privilege_enable);
#define ethosu_init(base_address) ethosu_init_v3(base_address, NULL, 0, 1, 1)
#define ethosu_init_v2(base_address, fast_memory, fast_memory_size) \
ethosu_init_v3(base_address, fast_memory, fast_memory_size, 1, 1)
/**
* Get Ethos-U version.
*/
int ethosu_get_version(struct ethosu_version *version);
/**
* Invoke Vela command stream.
*/
int ethosu_invoke_v2(const void *custom_data_ptr,
const int custom_data_size,
const uint64_t *base_addr,
const size_t *base_addr_size,
const int num_base_addr);
#define ethosu_invoke(custom_data_ptr, custom_data_size, base_addr, num_base_addr) \
ethosu_invoke_v2(custom_data_ptr, custom_data_size, base_addr, NULL, num_base_addr)
/**
* Abort Ethos-U inference.
*/
void ethosu_abort(void);
/**
* Interrupt handler do be called on IRQ from Ethos-U
*/
void ethosu_irq_handler(void);
#ifdef __cplusplus
}
#endif
#endif // ETHOSU_DRIVER_H
|
