diff options
| author | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2022-03-08 13:25:45 +0100 |
|---|---|---|
| committer | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2022-03-10 15:20:57 +0100 |
| commit | 35de9e63d9c2fe0a557637ac104d7d73382d2d4a (patch) | |
| tree | 41fa348f46f7f76b00625ad3b9768c1ddae5c83b /kernel/ethosu_mailbox.c | |
| parent | 118b05990af26026a1ac2b6d5dfae32ea342a7f4 (diff) | |
| download | ethos-u-linux-driver-stack-35de9e63d9c2fe0a557637ac104d7d73382d2d4a.tar.gz | |
Firmware resident model
Support referencing a network model by index that has been built into
the firmware binary.
Change-Id: Idd5294376ea82503dfeafe1203dcc0694d296dfe
Diffstat (limited to 'kernel/ethosu_mailbox.c')
| -rw-r--r-- | kernel/ethosu_mailbox.c | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/kernel/ethosu_mailbox.c b/kernel/ethosu_mailbox.c index 7f159f3..ef9a07d 100644 --- a/kernel/ethosu_mailbox.c +++ b/kernel/ethosu_mailbox.c @@ -1,5 +1,5 @@ /* - * (C) COPYRIGHT 2020 ARM Limited. All rights reserved. + * Copyright (c) 2020-2022 Arm Limited. * * This program is free software and is provided to you under the terms of the * GNU General Public License version 2 as published by the Free Software @@ -234,6 +234,7 @@ int ethosu_mailbox_inference(struct ethosu_mailbox *mbox, uint32_t ofm_count, struct ethosu_buffer **ofm, struct ethosu_buffer *network, + uint32_t network_index, uint8_t *pmu_event_config, uint8_t pmu_event_config_count, uint8_t pmu_cycle_counter_enable) @@ -262,7 +263,13 @@ int ethosu_mailbox_inference(struct ethosu_mailbox *mbox, for (i = 0; i < ETHOSU_CORE_PMU_MAX; i++) inf.pmu_event_config[i] = pmu_event_config[i]; - ethosu_core_set_size(network, &inf.network); + if (network != NULL) { + inf.network.type = ETHOSU_CORE_NETWORK_BUFFER; + ethosu_core_set_size(network, &inf.network.buffer); + } else { + inf.network.type = ETHOSU_CORE_NETWORK_INDEX; + inf.network.index = network_index; + } return ethosu_queue_write_msg(mbox, ETHOSU_CORE_MSG_INFERENCE_REQ, &inf, sizeof(inf)); |