blob: c3f55798736ddc66fd3b50bf04998c396e65373c [file] [log] [blame]
Kristofer Jonsson116a6352020-08-20 17:25:23 +02001/*
Kristofer Jonssonb42bc0b2023-01-04 17:09:28 +01002 * Copyright 2020-2023 Arm Limited and/or its affiliates
Kristofer Jonsson116a6352020-08-20 17:25:23 +02003 *
4 * This program is free software and is provided to you under the terms of the
5 * GNU General Public License version 2 as published by the Free Software
6 * Foundation, and any use by you of this program is subject to the terms
7 * of such GNU licence.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, you can access it online at
16 * http://www.gnu.org/licenses/gpl-2.0.html.
17 *
18 * SPDX-License-Identifier: GPL-2.0-only
19 */
20
21#ifndef ETHOSU_MAILBOX_H
22#define ETHOSU_MAILBOX_H
23
24/****************************************************************************
25 * Includes
26 ****************************************************************************/
Jonny Svärd7c24c772021-01-14 19:53:17 +010027#include "ethosu_core_interface.h"
Kristofer Jonsson116a6352020-08-20 17:25:23 +020028
29#include <linux/types.h>
30#include <linux/mailbox_client.h>
31#include <linux/workqueue.h>
Davide Grohmann32660f92022-04-27 16:49:07 +020032#include <linux/idr.h>
Kristofer Jonsson116a6352020-08-20 17:25:23 +020033
34/****************************************************************************
35 * Types
36 ****************************************************************************/
37
38struct device;
39struct ethosu_buffer;
40struct ethosu_device;
41struct ethosu_core_msg;
42struct ethosu_core_queue;
43struct resource;
44
45typedef void (*ethosu_mailbox_cb)(void *user_arg);
46
47struct ethosu_mailbox {
48 struct device *dev;
49 struct workqueue_struct *wq;
50 struct work_struct work;
51 struct ethosu_core_queue __iomem *in_queue;
52 struct ethosu_core_queue __iomem *out_queue;
53 struct mbox_client client;
54 struct mbox_chan *rx;
55 struct mbox_chan *tx;
56 ethosu_mailbox_cb callback;
57 void *user_arg;
Davide Grohmann32660f92022-04-27 16:49:07 +020058 struct idr msg_idr;
Kristofer Jonsson442fefb2022-03-17 17:15:52 +010059};
60
61struct ethosu_mailbox_msg {
Davide Grohmann32660f92022-04-27 16:49:07 +020062 int id;
63 void (*fail)(struct ethosu_mailbox_msg *msg);
Kristofer Jonsson116a6352020-08-20 17:25:23 +020064};
65
66/****************************************************************************
67 * Functions
68 ****************************************************************************/
69
70/**
71 * ethosu_mailbox_init() - Initialize mailbox
72 *
73 * Return: 0 on success, else error code.
74 */
75int ethosu_mailbox_init(struct ethosu_mailbox *mbox,
76 struct device *dev,
77 struct resource *in_queue,
78 struct resource *out_queue,
79 ethosu_mailbox_cb callback,
Kristofer Jonssonb42bc0b2023-01-04 17:09:28 +010080 void *user_arg);
Kristofer Jonsson116a6352020-08-20 17:25:23 +020081
82/**
83 * ethosu_mailbox_deinit() - Deinitialize mailbox
84 */
85void ethosu_mailbox_deinit(struct ethosu_mailbox *mbox);
86
87/**
Kristofer Jonsson442fefb2022-03-17 17:15:52 +010088 * ethosu_mailbox_wait_prepare() - Prepare to wait on firmware
89 *
90 * This function must only be called when the firmware is in a
91 * stopped state. It invalidates the firmware queues setting
92 * size, read and write positions to illegal values.
93 */
94void ethosu_mailbox_wait_prepare(struct ethosu_mailbox *mbox);
95
96/**
97 * ethosu_mailbox_wait_firmware() - Waiting for firmware to initialize
98 * message queues
99 *
100 * Following a call to ethosu_mailbox_wait_prepare() this function waits for
101 * the firmware to boot up and initialize the firmware queues.
102 *
103 * Return: 0 on success, else error code.
104 */
105int ethosu_mailbox_wait_firmware(struct ethosu_mailbox *mbox);
106
107/**
Kristofer Jonsson116a6352020-08-20 17:25:23 +0200108 * ethosu_mailbox_read() - Read message from mailbox
109 *
110 * Return: 0 message read, else error code.
111 */
112int ethosu_mailbox_read(struct ethosu_mailbox *mbox,
113 struct ethosu_core_msg *header,
114 void *data,
115 size_t length);
116
117/**
Davide Grohmann32660f92022-04-27 16:49:07 +0200118 * ethosu_mailbox_register() - Register the ethosu_mailbox_msg in ethosu_mailbox
Kristofer Jonsson442fefb2022-03-17 17:15:52 +0100119 *
120 * Return: 0 on success, else error code.
121 */
Davide Grohmann32660f92022-04-27 16:49:07 +0200122int ethosu_mailbox_register(struct ethosu_mailbox *mbox,
123 struct ethosu_mailbox_msg *msg);
124
125/**
126 * ethosu_mailbox_free_id() - Free the id of the ethosu_mailbox_msg
127 */
128void ethosu_mailbox_deregister(struct ethosu_mailbox *mbox,
129 struct ethosu_mailbox_msg *msg);
130
131/**
132 * ethosu_mailbox_find() - Find mailbox message
133 *
134 * Return: a valid pointer on success, otherwise an error ptr.
135 */
136struct ethosu_mailbox_msg *ethosu_mailbox_find(struct ethosu_mailbox *mbox,
137 int msg_id);
Kristofer Jonsson442fefb2022-03-17 17:15:52 +0100138
139/**
140 * ethosu_mailbox_fail() - Fail mailbox messages
141 *
142 * Call fail() callback on all messages in pending list.
143 */
144void ethosu_mailbox_fail(struct ethosu_mailbox *mbox);
145
146/**
Jonny Svärd7c24c772021-01-14 19:53:17 +0100147 * ethosu_mailbox_reset() - Reset to end of queue
148 */
149void ethosu_mailbox_reset(struct ethosu_mailbox *mbox);
150
151/**
Kristofer Jonsson116a6352020-08-20 17:25:23 +0200152 * ethosu_mailbox_ping() - Send ping message
153 *
154 * Return: 0 on success, else error code.
155 */
156int ethosu_mailbox_ping(struct ethosu_mailbox *mbox);
157
158/**
Jonny Svärd7c24c772021-01-14 19:53:17 +0100159 * ethosu_mailbox_pong() - Send pong response
160 *
161 * Return: 0 on success, else error code.
162 */
163int ethosu_mailbox_pong(struct ethosu_mailbox *mbox);
164
165/**
166 * ethosu_mailbox_version_response - Send version request
167 *
168 * Return: 0 on succes, else error code
169 */
170int ethosu_mailbox_version_request(struct ethosu_mailbox *mbox);
171
172/**
Davide Grohmann35ce6c82021-06-01 15:03:51 +0200173 * ethosu_mailbox_capabilities_request() - Send capabilities request
174 *
175 * Return: 0 on success, else error code.
176 */
177int ethosu_mailbox_capabilities_request(struct ethosu_mailbox *mbox,
Davide Grohmann32660f92022-04-27 16:49:07 +0200178 struct ethosu_mailbox_msg *msg);
Davide Grohmann35ce6c82021-06-01 15:03:51 +0200179
180/**
Kristofer Jonsson116a6352020-08-20 17:25:23 +0200181 * ethosu_mailbox_inference() - Send inference
182 *
183 * Return: 0 on success, else error code.
184 */
185int ethosu_mailbox_inference(struct ethosu_mailbox *mbox,
Davide Grohmann32660f92022-04-27 16:49:07 +0200186 struct ethosu_mailbox_msg *msg,
Kristofer Jonssonb74492c2020-09-10 13:26:01 +0200187 uint32_t ifm_count,
188 struct ethosu_buffer **ifm,
189 uint32_t ofm_count,
190 struct ethosu_buffer **ofm,
Per Åstrandf7e407a2020-10-23 21:25:05 +0200191 struct ethosu_buffer *network,
Kristofer Jonsson35de9e62022-03-08 13:25:45 +0100192 uint32_t network_index,
Per Åstrandf7e407a2020-10-23 21:25:05 +0200193 uint8_t *pmu_event_config,
194 uint8_t pmu_event_config_count,
195 uint8_t pmu_cycle_counter_enable);
Kristofer Jonsson116a6352020-08-20 17:25:23 +0200196
Kristofer Jonsson3c6a2602022-03-10 11:17:29 +0100197/**
198 * ethosu_mailbox_network_info_request() - Send network info request
199 *
200 * Return: 0 on success, else error code.
201 */
202int ethosu_mailbox_network_info_request(struct ethosu_mailbox *mbox,
Davide Grohmann32660f92022-04-27 16:49:07 +0200203 struct ethosu_mailbox_msg *msg,
Kristofer Jonsson3c6a2602022-03-10 11:17:29 +0100204 struct ethosu_buffer *network,
205 uint32_t network_index);
206
Davide Grohmann7e8f5082022-03-23 12:48:45 +0100207/**
208 * ethosu_mailbox_cancel_inference() - Send inference cancellation
209 *
210 * Return: 0 on success, else error code.
211 */
212int ethosu_mailbox_cancel_inference(struct ethosu_mailbox *mbox,
Davide Grohmann32660f92022-04-27 16:49:07 +0200213 struct ethosu_mailbox_msg *msg,
214 int inference_handle);
Davide Grohmann7e8f5082022-03-23 12:48:45 +0100215
Kristofer Jonsson116a6352020-08-20 17:25:23 +0200216#endif /* ETHOSU_MAILBOX_H */