Kristofer Jonsson | 116a635 | 2020-08-20 17:25:23 +0200 | [diff] [blame] | 1 | /* |
| 2 | * (C) COPYRIGHT 2020 ARM Limited. All rights reserved. |
| 3 | * |
| 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 | /**************************************************************************** |
| 22 | * Includes |
| 23 | ****************************************************************************/ |
| 24 | |
| 25 | #include "ethosu_mailbox.h" |
| 26 | |
| 27 | #include "ethosu_buffer.h" |
| 28 | #include "ethosu_core_interface.h" |
| 29 | #include "ethosu_device.h" |
| 30 | |
| 31 | #include <linux/resource.h> |
| 32 | #include <linux/uio.h> |
| 33 | |
| 34 | /**************************************************************************** |
| 35 | * Functions |
| 36 | ****************************************************************************/ |
| 37 | |
| 38 | static void ethosu_core_set_size(struct ethosu_buffer *buf, |
| 39 | struct ethosu_core_buffer *cbuf) |
| 40 | { |
| 41 | cbuf->ptr = (uint32_t)buf->dma_addr + buf->offset; |
| 42 | cbuf->size = (uint32_t)buf->size; |
| 43 | } |
| 44 | |
| 45 | static void ethosu_core_set_capacity(struct ethosu_buffer *buf, |
| 46 | struct ethosu_core_buffer *cbuf) |
| 47 | { |
| 48 | cbuf->ptr = (uint32_t)buf->dma_addr + buf->offset + buf->size; |
| 49 | cbuf->size = (uint32_t)buf->capacity - buf->offset - buf->size; |
| 50 | } |
| 51 | |
| 52 | static size_t ethosu_queue_available(struct ethosu_core_queue *queue) |
| 53 | { |
| 54 | size_t size = queue->header.write - queue->header.read; |
| 55 | |
| 56 | if (queue->header.read > queue->header.write) |
| 57 | size += queue->header.size; |
| 58 | |
| 59 | return size; |
| 60 | } |
| 61 | |
| 62 | static size_t ethosu_queue_capacity(struct ethosu_core_queue *queue) |
| 63 | { |
| 64 | return queue->header.size - ethosu_queue_available(queue); |
| 65 | } |
| 66 | |
| 67 | static int ethosu_queue_write(struct ethosu_mailbox *mbox, |
| 68 | const struct kvec *vec, |
| 69 | size_t length) |
| 70 | { |
| 71 | struct ethosu_core_queue *queue = mbox->in_queue; |
| 72 | uint8_t *dst = &queue->data[0]; |
| 73 | uint32_t wpos = queue->header.write; |
| 74 | size_t total_size; |
| 75 | size_t i; |
| 76 | int ret; |
| 77 | |
| 78 | for (i = 0, total_size = 0; i < length; i++) |
| 79 | total_size += vec[i].iov_len; |
| 80 | |
| 81 | if (total_size > ethosu_queue_capacity(queue)) |
| 82 | return -EINVAL; |
| 83 | |
| 84 | for (i = 0; i < length; i++) { |
| 85 | const uint8_t *src = vec[i].iov_base; |
| 86 | const uint8_t *end = src + vec[i].iov_len; |
| 87 | |
| 88 | while (src < end) { |
| 89 | dst[wpos] = *src++; |
| 90 | wpos = (wpos + 1) % queue->header.size; |
| 91 | } |
| 92 | } |
| 93 | |
| 94 | queue->header.write = wpos; |
| 95 | |
| 96 | ret = mbox_send_message(mbox->tx, queue); |
| 97 | if (ret < 0) |
| 98 | return ret; |
| 99 | |
| 100 | return 0; |
| 101 | } |
| 102 | |
| 103 | static int ethosu_queue_write_msg(struct ethosu_mailbox *mbox, |
| 104 | uint32_t type, |
| 105 | void *data, |
| 106 | size_t length) |
| 107 | { |
| 108 | struct ethosu_core_msg msg = { .type = type, .length = length }; |
| 109 | const struct kvec vec[2] = { |
| 110 | { &msg, sizeof(msg) }, |
| 111 | { data, length } |
| 112 | }; |
| 113 | |
| 114 | return ethosu_queue_write(mbox, vec, 2); |
| 115 | } |
| 116 | |
| 117 | static int ethosu_queue_read(struct ethosu_mailbox *mbox, |
| 118 | void *data, |
| 119 | size_t length) |
| 120 | { |
| 121 | struct ethosu_core_queue *queue = mbox->out_queue; |
| 122 | uint8_t *src = &queue->data[0]; |
| 123 | uint8_t *dst = (uint8_t *)data; |
| 124 | const uint8_t *end = dst + length; |
| 125 | uint32_t rpos = queue->header.read; |
| 126 | |
| 127 | if (length > ethosu_queue_available(queue)) |
| 128 | return -ENOMSG; |
| 129 | |
| 130 | while (dst < end) { |
| 131 | *dst++ = src[rpos]; |
| 132 | rpos = (rpos + 1) % queue->header.size; |
| 133 | } |
| 134 | |
| 135 | queue->header.read = rpos; |
| 136 | |
| 137 | return 0; |
| 138 | } |
| 139 | |
| 140 | int ethosu_mailbox_read(struct ethosu_mailbox *mbox, |
| 141 | struct ethosu_core_msg *header, |
| 142 | void *data, |
| 143 | size_t length) |
| 144 | { |
| 145 | int ret; |
| 146 | |
| 147 | /* Read message header */ |
| 148 | ret = ethosu_queue_read(mbox, header, sizeof(*header)); |
| 149 | if (ret) |
| 150 | return ret; |
| 151 | |
| 152 | dev_info(mbox->dev, "mbox: Read msg header. type=%u, length=%u", |
| 153 | header->type, header->length); |
| 154 | |
| 155 | /* Check that payload is not larger than allocated buffer */ |
| 156 | if (header->length > length) |
| 157 | return -ENOMEM; |
| 158 | |
| 159 | /* Ready payload data */ |
| 160 | ret = ethosu_queue_read(mbox, data, header->length); |
| 161 | if (ret) |
| 162 | return -EBADMSG; |
| 163 | |
| 164 | return 0; |
| 165 | } |
| 166 | |
| 167 | int ethosu_mailbox_ping(struct ethosu_mailbox *mbox) |
| 168 | { |
| 169 | return ethosu_queue_write_msg(mbox, ETHOSU_CORE_MSG_PING, NULL, 0); |
| 170 | } |
| 171 | |
| 172 | int ethosu_mailbox_inference(struct ethosu_mailbox *mbox, |
| 173 | void *user_arg, |
Kristofer Jonsson | b74492c | 2020-09-10 13:26:01 +0200 | [diff] [blame] | 174 | uint32_t ifm_count, |
| 175 | struct ethosu_buffer **ifm, |
| 176 | uint32_t ofm_count, |
| 177 | struct ethosu_buffer **ofm, |
Kristofer Jonsson | 116a635 | 2020-08-20 17:25:23 +0200 | [diff] [blame] | 178 | struct ethosu_buffer *network) |
| 179 | { |
| 180 | struct ethosu_core_inference_req inf; |
Kristofer Jonsson | b74492c | 2020-09-10 13:26:01 +0200 | [diff] [blame] | 181 | uint32_t i; |
Kristofer Jonsson | 116a635 | 2020-08-20 17:25:23 +0200 | [diff] [blame] | 182 | |
| 183 | inf.user_arg = (ptrdiff_t)user_arg; |
Kristofer Jonsson | b74492c | 2020-09-10 13:26:01 +0200 | [diff] [blame] | 184 | inf.ifm_count = ifm_count; |
| 185 | inf.ofm_count = ofm_count; |
| 186 | |
| 187 | for (i = 0; i < ifm_count; i++) |
| 188 | ethosu_core_set_size(ifm[i], &inf.ifm[i]); |
| 189 | |
| 190 | for (i = 0; i < ofm_count; i++) |
| 191 | ethosu_core_set_capacity(ofm[i], &inf.ofm[i]); |
| 192 | |
Kristofer Jonsson | 116a635 | 2020-08-20 17:25:23 +0200 | [diff] [blame] | 193 | ethosu_core_set_size(network, &inf.network); |
| 194 | |
| 195 | return ethosu_queue_write_msg(mbox, ETHOSU_CORE_MSG_INFERENCE_REQ, |
| 196 | &inf, sizeof(inf)); |
| 197 | } |
| 198 | |
| 199 | static void ethosu_mailbox_rx_work(struct work_struct *work) |
| 200 | { |
| 201 | struct ethosu_mailbox *mbox = container_of(work, typeof(*mbox), work); |
| 202 | |
| 203 | mbox->callback(mbox->user_arg); |
| 204 | } |
| 205 | |
| 206 | static void ethosu_mailbox_rx_callback(struct mbox_client *client, |
| 207 | void *message) |
| 208 | { |
| 209 | struct ethosu_mailbox *mbox = |
| 210 | container_of(client, typeof(*mbox), client); |
| 211 | |
| 212 | dev_info(mbox->dev, "mbox: Received message.\n"); |
| 213 | |
| 214 | queue_work(mbox->wq, &mbox->work); |
| 215 | } |
| 216 | |
| 217 | static void ethosu_mailbox_tx_done(struct mbox_client *client, |
| 218 | void *message, |
| 219 | int r) |
| 220 | { |
| 221 | if (r) |
| 222 | dev_warn(client->dev, "mbox: Failed sending message (%d)\n", r); |
| 223 | else |
| 224 | dev_info(client->dev, "mbox: Message sent\n"); |
| 225 | } |
| 226 | |
| 227 | int ethosu_mailbox_init(struct ethosu_mailbox *mbox, |
| 228 | struct device *dev, |
| 229 | struct resource *in_queue, |
| 230 | struct resource *out_queue, |
| 231 | ethosu_mailbox_cb callback, |
| 232 | void *user_arg) |
| 233 | { |
| 234 | int ret; |
| 235 | |
| 236 | mbox->dev = dev; |
| 237 | mbox->callback = callback; |
| 238 | mbox->user_arg = user_arg; |
| 239 | |
| 240 | mbox->client.dev = dev; |
| 241 | mbox->client.rx_callback = ethosu_mailbox_rx_callback; |
| 242 | mbox->client.tx_prepare = NULL; /* preparation of data is handled |
| 243 | * through the |
| 244 | * queue functions */ |
| 245 | mbox->client.tx_done = ethosu_mailbox_tx_done; |
| 246 | mbox->client.tx_block = true; |
| 247 | mbox->client.knows_txdone = false; |
| 248 | mbox->client.tx_tout = 500; |
| 249 | |
| 250 | mbox->in_queue = devm_ioremap_resource(mbox->dev, in_queue); |
| 251 | if (IS_ERR(mbox->in_queue)) |
| 252 | return PTR_ERR(mbox->in_queue); |
| 253 | |
| 254 | mbox->out_queue = devm_ioremap_resource(mbox->dev, out_queue); |
| 255 | if (IS_ERR(mbox->out_queue)) { |
| 256 | ret = PTR_ERR(mbox->out_queue); |
| 257 | goto unmap_in_queue; |
| 258 | } |
| 259 | |
| 260 | mbox->wq = create_singlethread_workqueue("ethosu_workqueue"); |
| 261 | if (!mbox->wq) { |
| 262 | dev_err(mbox->dev, "Failed to create work queue\n"); |
| 263 | ret = -EINVAL; |
| 264 | goto unmap_out_queue; |
| 265 | } |
| 266 | |
| 267 | INIT_WORK(&mbox->work, ethosu_mailbox_rx_work); |
| 268 | |
| 269 | mbox->tx = mbox_request_channel_byname(&mbox->client, "tx"); |
| 270 | if (IS_ERR(mbox->tx)) { |
| 271 | dev_warn(mbox->dev, "mbox: Failed to request tx channel\n"); |
| 272 | ret = PTR_ERR(mbox->tx); |
| 273 | goto workqueue_destroy; |
| 274 | } |
| 275 | |
| 276 | mbox->rx = mbox_request_channel_byname(&mbox->client, "rx"); |
| 277 | if (IS_ERR(mbox->rx)) { |
| 278 | dev_info(dev, "mbox: Using same channel for RX and TX\n"); |
| 279 | mbox->rx = mbox->tx; |
| 280 | } |
| 281 | |
| 282 | return 0; |
| 283 | |
| 284 | workqueue_destroy: |
| 285 | destroy_workqueue(mbox->wq); |
| 286 | |
| 287 | unmap_out_queue: |
| 288 | devm_iounmap(mbox->dev, mbox->out_queue); |
| 289 | |
| 290 | unmap_in_queue: |
| 291 | devm_iounmap(mbox->dev, mbox->in_queue); |
| 292 | |
| 293 | return ret; |
| 294 | } |
| 295 | |
| 296 | void ethosu_mailbox_deinit(struct ethosu_mailbox *mbox) |
| 297 | { |
| 298 | if (mbox->rx != mbox->tx) |
| 299 | mbox_free_channel(mbox->rx); |
| 300 | |
| 301 | mbox_free_channel(mbox->tx); |
| 302 | destroy_workqueue(mbox->wq); |
| 303 | devm_iounmap(mbox->dev, mbox->out_queue); |
| 304 | devm_iounmap(mbox->dev, mbox->in_queue); |
| 305 | } |