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
|
/*
* 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_v2(const void *base_address, const void *fast_memory, const size_t fast_memory_size);
#define ethosu_init(base_address) ethosu_init_v2(base_address, NULL, 0)
/**
* 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
|
