-
Notifications
You must be signed in to change notification settings - Fork 16
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
shuzhuo
committed
Sep 20, 2024
1 parent
51ff559
commit d6b63d5
Showing
65 changed files
with
843 additions
and
441 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
128 changes: 0 additions & 128 deletions
128
NMSIS/NN/Benchmark/ConvolutionFunctions/test_riscv_convolve_fast_s16.c
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
183 changes: 183 additions & 0 deletions
183
NMSIS/NN/Benchmark/LSTMFunctions/test_riscv_lstm_unidirectional_s16.c
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,183 @@ | ||
| /* | ||
| * SPDX-FileCopyrightText: Copyright 2022-2023 Arm Limited and/or its affiliates <[email protected]> | ||
| * | ||
| * 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. | ||
| */ | ||
|
|
||
| #include <stdbool.h> | ||
| #include <stdlib.h> | ||
|
|
||
| #include <riscv_nnfunctions.h> | ||
|
|
||
|
|
||
| #include "TestData/lstm_1_s16/test_data.h" | ||
| #include "Utils/validate.h" | ||
| #include "nmsis_bench.h" | ||
|
|
||
| BENCH_DECLARE_VAR(); | ||
|
|
||
| // Update the buffer size if adding a unit test with larger buffer. | ||
| #define LARGEST_BUFFER_SIZE lstm_1_s16_hidden_size *lstm_1_s16_batch_size *lstm_1_s16_time_steps | ||
|
|
||
| static int16_t buffer1[LARGEST_BUFFER_SIZE]; | ||
| static int16_t buffer2[LARGEST_BUFFER_SIZE]; | ||
| static int16_t buffer3[LARGEST_BUFFER_SIZE]; | ||
|
|
||
| void lstm_1_riscv_lstm_unidirectional_s16(void) | ||
| { | ||
| int16_t output[lstm_1_s16_batch_size * lstm_1_s16_time_steps * lstm_1_s16_hidden_size] = {0}; | ||
| const riscv_nmsis_nn_status expected = RISCV_NMSIS_NN_SUCCESS; | ||
| const int16_t *output_ref = &lstm_1_s16_output[0]; | ||
| const int32_t output_ref_size = lstm_1_s16_batch_size * lstm_1_s16_time_steps * lstm_1_s16_hidden_size; | ||
|
|
||
| int64_t input_data_kernel_sum[lstm_1_s16_hidden_size]; | ||
| int64_t forget_data_kernel_sum[lstm_1_s16_hidden_size]; | ||
| int64_t cell_data_kernel_sum[lstm_1_s16_hidden_size]; | ||
| int64_t output_data_kernel_sum[lstm_1_s16_hidden_size]; | ||
|
|
||
| int64_t input_hidden_kernel_sum[lstm_1_s16_hidden_size]; | ||
| int64_t forget_hidden_kernel_sum[lstm_1_s16_hidden_size]; | ||
| int64_t cell_hidden_kernel_sum[lstm_1_s16_hidden_size]; | ||
| int64_t output_hidden_kernel_sum[lstm_1_s16_hidden_size]; | ||
|
|
||
| riscv_vector_sum_s8_s64(&input_data_kernel_sum[0], | ||
| lstm_1_s16_input_size, | ||
| lstm_1_s16_hidden_size, | ||
| &lstm_1_s16_input_gate_input_weights[0], | ||
| lstm_1_s16_input_zero_point, | ||
| &lstm_1_s16_input_gate_bias[0]); | ||
| riscv_vector_sum_s8_s64(&forget_data_kernel_sum[0], | ||
| lstm_1_s16_input_size, | ||
| lstm_1_s16_hidden_size, | ||
| &lstm_1_s16_forget_gate_input_weights[0], | ||
| lstm_1_s16_input_zero_point, | ||
| &lstm_1_s16_forget_gate_bias[0]); | ||
| riscv_vector_sum_s8_s64(&cell_data_kernel_sum[0], | ||
| lstm_1_s16_input_size, | ||
| lstm_1_s16_hidden_size, | ||
| &lstm_1_s16_cell_gate_input_weights[0], | ||
| lstm_1_s16_input_zero_point, | ||
| &lstm_1_s16_cell_gate_bias[0]); | ||
| riscv_vector_sum_s8_s64(&output_data_kernel_sum[0], | ||
| lstm_1_s16_input_size, | ||
| lstm_1_s16_hidden_size, | ||
| &lstm_1_s16_output_gate_input_weights[0], | ||
| lstm_1_s16_input_zero_point, | ||
| &lstm_1_s16_output_gate_bias[0]); | ||
|
|
||
| riscv_vector_sum_s8_s64(&input_hidden_kernel_sum[0], | ||
| lstm_1_s16_hidden_size, | ||
| lstm_1_s16_hidden_size, | ||
| &lstm_1_s16_input_gate_hidden_weights[0], | ||
| -lstm_1_s16_output_zero_point, | ||
| NULL); | ||
| riscv_vector_sum_s8_s64(&forget_hidden_kernel_sum[0], | ||
| lstm_1_s16_hidden_size, | ||
| lstm_1_s16_hidden_size, | ||
| &lstm_1_s16_forget_gate_hidden_weights[0], | ||
| -lstm_1_s16_output_zero_point, | ||
| NULL); | ||
| riscv_vector_sum_s8_s64(&cell_hidden_kernel_sum[0], | ||
| lstm_1_s16_hidden_size, | ||
| lstm_1_s16_hidden_size, | ||
| &lstm_1_s16_cell_gate_hidden_weights[0], | ||
| -lstm_1_s16_output_zero_point, | ||
| NULL); | ||
| riscv_vector_sum_s8_s64(&output_hidden_kernel_sum[0], | ||
| lstm_1_s16_hidden_size, | ||
| lstm_1_s16_hidden_size, | ||
| &lstm_1_s16_output_gate_hidden_weights[0], | ||
| -lstm_1_s16_output_zero_point, | ||
| NULL); | ||
|
|
||
| // INPUT GATE | ||
| const nmsis_nn_lstm_gate gate_input = {lstm_1_s16_input_gate_input_multiplier, | ||
| lstm_1_s16_input_gate_input_shift, | ||
| &lstm_1_s16_input_gate_input_weights[0], | ||
| &input_data_kernel_sum[0], | ||
| lstm_1_s16_input_gate_hidden_multiplier, | ||
| lstm_1_s16_input_gate_hidden_shift, | ||
| &lstm_1_s16_input_gate_hidden_weights[0], | ||
| &input_hidden_kernel_sum[0], | ||
| &lstm_1_s16_input_gate_bias[0], | ||
| RISCV_SIGMOID}; | ||
|
|
||
| // FORGET GATE | ||
| const nmsis_nn_lstm_gate gate_forget = {lstm_1_s16_forget_gate_input_multiplier, | ||
| lstm_1_s16_forget_gate_input_shift, | ||
| &lstm_1_s16_forget_gate_input_weights[0], | ||
| &forget_data_kernel_sum[0], | ||
| lstm_1_s16_forget_gate_hidden_multiplier, | ||
| lstm_1_s16_forget_gate_hidden_shift, | ||
| &lstm_1_s16_forget_gate_hidden_weights[0], | ||
| &forget_hidden_kernel_sum[0], | ||
| &lstm_1_s16_forget_gate_bias[0], | ||
| RISCV_SIGMOID}; | ||
|
|
||
| // CELL GATE | ||
| const nmsis_nn_lstm_gate gate_cell = {lstm_1_s16_cell_gate_input_multiplier, | ||
| lstm_1_s16_cell_gate_input_shift, | ||
| &lstm_1_s16_cell_gate_input_weights[0], | ||
| &cell_data_kernel_sum[0], | ||
| lstm_1_s16_cell_gate_hidden_multiplier, | ||
| lstm_1_s16_cell_gate_hidden_shift, | ||
| &lstm_1_s16_cell_gate_hidden_weights[0], | ||
| &cell_hidden_kernel_sum[0], | ||
| &lstm_1_s16_cell_gate_bias[0], | ||
| RISCV_TANH}; | ||
|
|
||
| // OUTPUT GATE | ||
| const nmsis_nn_lstm_gate gate_output = {lstm_1_s16_output_gate_input_multiplier, | ||
| lstm_1_s16_output_gate_input_shift, | ||
| &lstm_1_s16_output_gate_input_weights[0], | ||
| &output_data_kernel_sum[0], | ||
| lstm_1_s16_output_gate_hidden_multiplier, | ||
| lstm_1_s16_output_gate_hidden_shift, | ||
| &lstm_1_s16_output_gate_hidden_weights[0], | ||
| &output_hidden_kernel_sum[0], | ||
| &lstm_1_s16_output_gate_bias[0], | ||
| RISCV_SIGMOID}; | ||
|
|
||
| // LSTM DATA | ||
| const nmsis_nn_lstm_params params = {lstm_1_s16_time_major, | ||
| lstm_1_s16_batch_size, | ||
| lstm_1_s16_time_steps, | ||
| lstm_1_s16_input_size, | ||
| lstm_1_s16_hidden_size, | ||
| lstm_1_s16_input_zero_point, | ||
| lstm_1_s16_forget_to_cell_multiplier, | ||
| lstm_1_s16_forget_to_cell_shift, | ||
| lstm_1_s16_input_to_cell_multiplier, | ||
| lstm_1_s16_input_to_cell_shift, | ||
| lstm_1_s16_cell_clip, | ||
| lstm_1_s16_cell_scale_power, | ||
| lstm_1_s16_output_multiplier, | ||
| lstm_1_s16_output_shift, | ||
| lstm_1_s16_output_zero_point, | ||
| gate_forget, | ||
| gate_input, | ||
| gate_cell, | ||
| gate_output}; | ||
|
|
||
| nmsis_nn_lstm_context buffers; | ||
| buffers.temp1 = buffer1; | ||
| buffers.temp2 = buffer2; | ||
| buffers.cell_state = buffer3; | ||
| BENCH_START(riscv_lstm_unidirectional_s16); | ||
| riscv_nmsis_nn_status result = riscv_lstm_unidirectional_s16(lstm_1_s16_input, output, ¶ms, &buffers); | ||
| BENCH_END(riscv_lstm_unidirectional_s16); | ||
| TEST_ASSERT_EQUAL(expected, result); | ||
| // TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); | ||
| } |
Oops, something went wrong.