diff --git a/onert-micro/onert-micro/include/core/OMKernelData.h b/onert-micro/onert-micro/include/core/OMKernelData.h index 64bf96f0521..c3d7a3fc9f6 100644 --- a/onert-micro/onert-micro/include/core/OMKernelData.h +++ b/onert-micro/onert-micro/include/core/OMKernelData.h @@ -42,6 +42,12 @@ struct ConcatenationParams uint32_t axis; }; +struct CumSumParams +{ + bool exclusive; + bool reverse; +}; + struct SoftmaxParams { float beta; diff --git a/onert-micro/onert-micro/include/pal/common/PALCumSum.h b/onert-micro/onert-micro/include/pal/common/PALCumSum.h new file mode 100644 index 00000000000..ea6595ee7f4 --- /dev/null +++ b/onert-micro/onert-micro/include/pal/common/PALCumSum.h @@ -0,0 +1,99 @@ +/* + * Copyright (c) 2024 Samsung Electronics Co., Ltd. All Rights Reserved + * Copyright 2021 The TensorFlow Authors. All Rights Reserved. + * + * 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 + * + * http://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 ONERT_MICRO_PAL_CUM_SUM_COMMON_H +#define ONERT_MICRO_PAL_CUM_SUM_COMMON_H + +#include "core/OMKernelData.h" +#include "core/OMRuntimeShape.h" + +using namespace onert_micro::core; + +namespace onert_micro::execute::pal +{ + +template +OMStatus CumSum(const CumSumParams ¶ms, const OMRuntimeShape &input_shape, const T *input_data, + int32_t axis, T *output_data) +{ + const int32_t rank = input_shape.dimensionsCount(); + + int32_t inner_size = 1; + int32_t outer_size = 1; + int32_t depth_size = 1; + + for (int32_t i = 0; i < rank; ++i) + { + if (i < axis) + inner_size *= input_shape.dims(i); + + else if (i > axis) + outer_size *= input_shape.dims(i); + + else + depth_size *= input_shape.dims(i); + } + + // clang-format off + + auto get_adj_index_fn = [reverse = params.reverse](int32_t index, int32_t dims) + { + if (!reverse) + return index; + + return (dims - 1) - index; + }; + + // clang-format on + + for (int32_t outer = 0; outer < outer_size; ++outer) + { + int32_t outer_adj = get_adj_index_fn(outer, outer_size); + + for (int32_t inner = 0; inner < inner_size; ++inner) + { + T accumulator = 0; + int32_t inner_adj = get_adj_index_fn(inner, inner_size); + + for (int32_t depth = 0; depth < depth_size; ++depth) + { + int32_t depth_adj = get_adj_index_fn(depth, depth_size); + int32_t index = outer_adj; + + index += inner_adj * depth_size * outer_size; + index += depth_adj * outer_size; + + if (params.exclusive) + { + output_data[index] = accumulator; + accumulator += input_data[index]; + + continue; + } + + accumulator += input_data[index]; + output_data[index] = accumulator; + } + } + } + + return Ok; +} + +} // namespace onert_micro::execute::pal + +#endif // ONERT_MICRO_PAL_CUM_SUM_COMMON_H diff --git a/onert-micro/onert-micro/include/pal/mcu/KernelsToBuild.lst b/onert-micro/onert-micro/include/pal/mcu/KernelsToBuild.lst index 23c59c6a2ff..2e056a78d51 100644 --- a/onert-micro/onert-micro/include/pal/mcu/KernelsToBuild.lst +++ b/onert-micro/onert-micro/include/pal/mcu/KernelsToBuild.lst @@ -5,6 +5,7 @@ REGISTER_KERNEL(AVERAGE_POOL_2D, AveragePool2D) REGISTER_KERNEL(ARG_MAX, ArgMax) REGISTER_KERNEL(ARG_MIN, ArgMin) REGISTER_KERNEL(CONCATENATION, Concatenation) +REGISTER_KERNEL(CUMSUM, CumSum) #/*REGISTER_KERNEL(CUSTOM, BroadcastTo)*/ REGISTER_KERNEL(BATCH_TO_SPACE_ND, BatchToSpaceND) REGISTER_KERNEL(CEIL, Ceil) diff --git a/onert-micro/onert-micro/include/test_models/TestDataBase.h b/onert-micro/onert-micro/include/test_models/TestDataBase.h index 4fc6373938e..d44458510b3 100644 --- a/onert-micro/onert-micro/include/test_models/TestDataBase.h +++ b/onert-micro/onert-micro/include/test_models/TestDataBase.h @@ -18,10 +18,10 @@ #define ONERT_MICRO_TEST_MODELS_TEST_DATA_BASE_H #include +#include +#include -namespace onert_micro -{ -namespace test_model +namespace onert_micro::test_model { template class TestDataBase @@ -32,7 +32,17 @@ template class TestDataBase virtual const unsigned char *get_model_ptr() = 0; virtual const std::vector &get_input_data_by_index(int i) = 0; + virtual const std::vector &get_output_data_by_index(int i) = 0; + + // clang-format off + + virtual const uint8_t *get_input_data_ptr_by_index(int i, int32_t &size) + { + return nullptr; + } + + // clang-format on }; class NegTestDataBase @@ -42,7 +52,6 @@ class NegTestDataBase virtual const unsigned char *get_model_ptr() = 0; }; -} // namespace test_model -} // namespace onert_micro +} // namespace onert_micro::test_model #endif // ONERT_MICRO_TEST_MODELS_TEST_DATA_BASE_H diff --git a/onert-micro/onert-micro/include/test_models/cum_sum/FloatCumSumKernel.h b/onert-micro/onert-micro/include/test_models/cum_sum/FloatCumSumKernel.h new file mode 100644 index 00000000000..086a630d54c --- /dev/null +++ b/onert-micro/onert-micro/include/test_models/cum_sum/FloatCumSumKernel.h @@ -0,0 +1,299 @@ +/* + * Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved + * + * 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 + * + * http://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 ONERT_MICRO_TEST_MODELS_FLOAT_CUMSUM_KERNEL_H +#define ONERT_MICRO_TEST_MODELS_FLOAT_CUMSUM_KERNEL_H + +#include "TestDataCumSumBase.h" + +// clang-format off + +namespace onert_micro::test_model +{ +namespace cumsum_float +{ + +/* + * CumSum Kernel: + * + * Input(1, 1, 2, 4): Float32 + * | + * CumSum(axis=2: Int32) + * | + * Output(1, 1, 2, 4): Float32 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x44, 0x00, 0x00, 0x00, 0x78, 0x01, 0x00, 0x00, 0x94, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x30, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x04, 0x00, 0x06, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x8c, 0xff, 0xff, 0xff, 0x90, 0xff, 0xff, 0xff, + 0x94, 0xff, 0xff, 0xff, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, + 0x18, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, + 0x14, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, 0x60, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00, 0x00, + 0x68, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x6d, 0x61, 0x69, 0x6e, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x0c, 0x00, 0x07, 0x00, 0x08, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66, + 0x10, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x04, 0x00, 0x04, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x74, 0x00, 0x00, 0x00, + 0x40, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0xa4, 0xff, 0xff, 0xff, 0x0c, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6f, 0x66, 0x6d, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0f, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, + 0x10, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x03, 0x00, 0x00, 0x00, 0x69, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, 0x11, 0x00, 0x00, 0x00, + 0x4f, 0x4e, 0x45, 0x2d, 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, 0x69, 0x72, 0x63, 0x6c, + 0x65, 0x00, 0x00, 0x00 +}; + +const std::vector input_data = { 1, 2, 3, 4, 5, 6, 7, 8 }; +const std::vector reference_output_data = { 1, 2, 3, 4, 6, 8, 10, 12 }; + +} // namespace cumsum_float + +// ------------------------------------------------------------------------------------------------ + +namespace cumsum_float_two_inputs +{ + +/* + * CumSum Kernel: + * + * Input(1, 1, 2, 4): Float32 Axis(scalar=3): Int32 + * | | + * ---------CumSum----------- + * | + * Output(1, 1, 2, 4) : Float32 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x30, 0x00, 0x00, 0x00, 0x68, 0x01, 0x00, 0x00, 0x84, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x1c, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x88, 0xff, 0xff, 0xff, 0x8c, 0xff, 0xff, 0xff, 0x90, 0xff, 0xff, 0xff, 0x94, 0xff, 0xff, 0xff, + 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x18, 0x00, 0x14, 0x00, + 0x10, 0x00, 0x0c, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x1c, 0x00, 0x00, 0x00, 0x60, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00, 0x00, 0x6c, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x6d, 0x61, 0x69, 0x6e, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, 0x10, 0x00, 0x0c, 0x00, + 0x07, 0x00, 0x08, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66, 0x10, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x04, 0x00, 0x04, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x74, 0x00, 0x00, 0x00, + 0x40, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0xa4, 0xff, 0xff, 0xff, 0x0c, 0x00, 0x00, 0x00, + 0x03, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6f, 0x66, 0x6d, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0f, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, + 0x10, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x03, 0x00, 0x00, 0x00, 0x69, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, 0x11, 0x00, 0x00, 0x00, + 0x4f, 0x4e, 0x45, 0x2d, 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, 0x69, 0x72, 0x63, 0x6c, + 0x65, 0x00, 0x00, 0x00 +}; + +const std::vector axis_data = { 3 }; +const std::vector input_data = { 1, 2, 3, 4, 5, 6, 7, 8 }; +const std::vector reference_output_data = { 1, 3, 6, 10, 5, 11, 18, 26 }; + +} // namespace cumsum_float_two_inputs + +// ------------------------------------------------------------------------------------------------ + +namespace cumsum_float_reverse +{ + +/* + * CumSum Kernel: + * + * Input(1, 1, 2, 4): Float32 + * | + * CumSum(axis=2: Int32, reverse=True) + * | + * Output(1, 1, 2, 4): Float32 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x48, 0x00, 0x00, 0x00, 0x84, 0x01, 0x00, 0x00, 0xa0, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x34, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x04, 0x00, 0x06, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0xf8, 0xff, 0xff, 0xff, 0xfc, 0xff, 0xff, 0xff, + 0x04, 0x00, 0x04, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x0e, 0x00, 0x18, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0e, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, 0x68, 0x00, 0x00, 0x00, + 0x6c, 0x00, 0x00, 0x00, 0x70, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x6d, 0x61, 0x69, 0x6e, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, + 0x14, 0x00, 0x00, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x07, 0x00, 0x08, 0x00, 0x0e, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x66, 0x14, 0x00, 0x00, 0x00, 0x18, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, + 0x08, 0x00, 0x08, 0x00, 0x00, 0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, + 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x74, 0x00, 0x00, 0x00, 0x40, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0xa4, 0xff, 0xff, 0xff, 0x0c, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6f, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0f, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x10, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, + 0x69, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, 0x11, 0x00, 0x00, 0x00, 0x4f, 0x4e, 0x45, 0x2d, + 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, 0x69, 0x72, 0x63, 0x6c, 0x65, 0x00, 0x00, 0x00 +}; + +const std::vector input_data = { 1, 2, 3, 4, 5, 6, 7, 8 }; +const std::vector reference_output_data = { 6, 8, 10, 12, 5, 6, 7, 8 }; + +} // namespace cumsum_float_reverse + +// ------------------------------------------------------------------------------------------------ + +namespace cumsum_float_exclusive +{ + +/* + * CumSum Kernel: + * + * Input(1, 1, 2, 4): Float32 + * | + * CumSum(axis=2: Int32, exclusive=True) + * | + * Output(1, 1, 2, 4): Float32 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x48, 0x00, 0x00, 0x00, 0x84, 0x01, 0x00, 0x00, 0xa0, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x34, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x04, 0x00, 0x06, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0xf8, 0xff, 0xff, 0xff, 0xfc, 0xff, 0xff, 0xff, + 0x04, 0x00, 0x04, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x0e, 0x00, 0x18, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0e, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, 0x68, 0x00, 0x00, 0x00, + 0x6c, 0x00, 0x00, 0x00, 0x70, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x6d, 0x61, 0x69, 0x6e, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, + 0x16, 0x00, 0x00, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x07, 0x00, 0x08, 0x00, 0x0e, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x66, 0x14, 0x00, 0x00, 0x00, 0x18, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x07, 0x00, 0x06, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, + 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x74, 0x00, 0x00, 0x00, 0x40, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0xa4, 0xff, 0xff, 0xff, 0x0c, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6f, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0f, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x10, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, + 0x69, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, 0x11, 0x00, 0x00, 0x00, 0x4f, 0x4e, 0x45, 0x2d, + 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, 0x69, 0x72, 0x63, 0x6c, 0x65, 0x00, 0x00, 0x00 +}; + +const std::vector input_data = { 1, 2, 3, 4, 5, 6, 7, 8 }; +const std::vector reference_output_data = { 0, 0, 1, 2, 4, 6, 9, 12 }; + +} // namespace cumsum_float_exclusive + +// ------------------------------------------------------------------------------------------------ + +class TestDataFloatCumSum : public TestDataCumSumBase +{ +public: + TestDataFloatCumSum() + { + _input_data = cumsum_float::input_data; + _reference_output_data = cumsum_float::reference_output_data; + _test_kernel_model_circle = cumsum_float::test_kernel_model_circle; + } + + ~TestDataFloatCumSum() override = default; +}; + +// ------------------------------------------------------------------------------------------------ + +class TestDataTwoInputsFloatCumSum : public TestDataCumSumBase +{ +public: + TestDataTwoInputsFloatCumSum() + { + _input_data = cumsum_float_two_inputs::input_data; + _axis_data = cumsum_float_two_inputs::axis_data; + _reference_output_data = cumsum_float_two_inputs::reference_output_data; + _test_kernel_model_circle = cumsum_float_two_inputs::test_kernel_model_circle; + } + + ~TestDataTwoInputsFloatCumSum() override = default; +}; + +// ------------------------------------------------------------------------------------------------ + +class TestDataReverseFloatCumSum : public TestDataCumSumBase +{ +public: + TestDataReverseFloatCumSum() + { + _input_data = cumsum_float_reverse::input_data; + _reference_output_data = cumsum_float_reverse::reference_output_data; + _test_kernel_model_circle = cumsum_float_reverse::test_kernel_model_circle; + } + + ~TestDataReverseFloatCumSum() override = default; +}; + +// ------------------------------------------------------------------------------------------------ + +class TestDataExclusiveFloatCumSum : public TestDataCumSumBase +{ +public: + TestDataExclusiveFloatCumSum() + { + _input_data = cumsum_float_exclusive::input_data; + _reference_output_data = cumsum_float_exclusive::reference_output_data; + _test_kernel_model_circle = cumsum_float_exclusive::test_kernel_model_circle; + } + + ~TestDataExclusiveFloatCumSum() override = default; +}; + +} // namespace onert_micro::test_model + +#endif // ONERT_MICRO_TEST_MODELS_FLOAT_CUMSUM_KERNEL_H diff --git a/onert-micro/onert-micro/include/test_models/cum_sum/IntCumSumKernel.h b/onert-micro/onert-micro/include/test_models/cum_sum/IntCumSumKernel.h new file mode 100644 index 00000000000..34a40ec45ee --- /dev/null +++ b/onert-micro/onert-micro/include/test_models/cum_sum/IntCumSumKernel.h @@ -0,0 +1,161 @@ +/* + * Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved + * + * 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 + * + * http://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 ONERT_MICRO_TEST_MODELS_INT_CUMSUM_KERNEL_H +#define ONERT_MICRO_TEST_MODELS_INT_CUMSUM_KERNEL_H + +#include "TestDataCumSumBase.h" +#include + +// clang-format off + +namespace onert_micro::test_model +{ +namespace cumsum_int32 +{ + +/* + * CumSum Kernel: + * + * Input(1, 1, 2, 4): Int32 + * | + * CumSum(axis=2: Int32) + * | + * Output(1, 1, 2, 4): Int32 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x44, 0x00, 0x00, 0x00, 0x74, 0x01, 0x00, 0x00, 0x90, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x30, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x04, 0x00, 0x06, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x8c, 0xff, 0xff, 0xff, 0x90, 0xff, 0xff, 0xff, + 0x94, 0xff, 0xff, 0xff, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, + 0x18, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, + 0x14, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, 0x60, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00, 0x00, + 0x68, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x6d, 0x61, 0x69, 0x6e, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x0c, 0x00, 0x07, 0x00, 0x08, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66, + 0x10, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x04, 0x00, 0x04, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6c, 0x00, 0x00, 0x00, + 0x38, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0xac, 0xff, 0xff, 0xff, 0x10, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, + 0x6f, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0xdc, 0xff, 0xff, 0xff, 0x10, 0x00, 0x00, 0x00, + 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x10, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x14, 0x00, + 0x10, 0x00, 0x0f, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, + 0x69, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, 0x11, 0x00, 0x00, 0x00, 0x4f, 0x4e, 0x45, 0x2d, + 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, 0x69, 0x72, 0x63, 0x6c, 0x65, 0x00, 0x00, 0x00 +}; + +const std::vector input_data = { 1, 2, 3, 4, 5, 6, 7, 8 }; +const std::vector reference_output_data = { 1, 2, 3, 4, 6, 8, 10, 12 }; + +} // namespace cumsum_int + +// ------------------------------------------------------------------------------------------------ + +namespace cumsum_int64 +{ + +/* + * CumSum Kernel: + * + * Input(1, 1, 2, 4): Int64 + * | + * CumSum(axis=2: Int32) + * | + * Output(1, 1, 2, 4): Int64 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x44, 0x00, 0x00, 0x00, 0x74, 0x01, 0x00, 0x00, 0x90, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x30, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x04, 0x00, 0x06, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x8c, 0xff, 0xff, 0xff, 0x90, 0xff, 0xff, 0xff, + 0x94, 0xff, 0xff, 0xff, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, + 0x18, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, + 0x14, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, 0x60, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00, 0x00, + 0x68, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x6d, 0x61, 0x69, 0x6e, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x0c, 0x00, 0x07, 0x00, 0x08, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66, + 0x10, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x04, 0x00, 0x04, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6c, 0x00, 0x00, 0x00, + 0x38, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0xac, 0xff, 0xff, 0xff, 0x10, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, + 0x6f, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0xdc, 0xff, 0xff, 0xff, 0x10, 0x00, 0x00, 0x00, + 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x10, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x14, 0x00, + 0x10, 0x00, 0x0f, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, + 0x69, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, 0x11, 0x00, 0x00, 0x00, 0x4f, 0x4e, 0x45, 0x2d, + 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, 0x69, 0x72, 0x63, 0x6c, 0x65, 0x00, 0x00, 0x00 +}; + +const std::vector input_data = { 1, 2, 3, 4, 5, 6, 7, 8 }; +const std::vector reference_output_data = { 1, 2, 3, 4, 6, 8, 10, 12 }; + +} // namespace cumsum_int64 + +// ------------------------------------------------------------------------------------------------ + +class TestDataInt32CumSum : public TestDataCumSumBase +{ +public: + TestDataInt32CumSum() + { + _input_data = cumsum_int32::input_data; + _reference_output_data = cumsum_int32::reference_output_data; + _test_kernel_model_circle = cumsum_int32::test_kernel_model_circle; + } + + ~TestDataInt32CumSum() override = default; +}; + +// ------------------------------------------------------------------------------------------------ + +class TestDataInt64CumSum : public TestDataCumSumBase +{ +public: + TestDataInt64CumSum() + { + _input_data = cumsum_int64::input_data; + _reference_output_data = cumsum_int64::reference_output_data; + _test_kernel_model_circle = cumsum_int64::test_kernel_model_circle; + } + + ~TestDataInt64CumSum() override = default; +}; + +} // namespace onert_micro::test_model + +#endif // ONERT_MICRO_TEST_MODELS_INT_CUMSUM_KERNEL_H diff --git a/onert-micro/onert-micro/include/test_models/cum_sum/NegCumSumKernel.h b/onert-micro/onert-micro/include/test_models/cum_sum/NegCumSumKernel.h new file mode 100644 index 00000000000..a9692e27742 --- /dev/null +++ b/onert-micro/onert-micro/include/test_models/cum_sum/NegCumSumKernel.h @@ -0,0 +1,226 @@ +/* + * Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved + * + * 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 + * + * http://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 ONERT_MICRO_TEST_MODELS_NEG_CUMSUM_KERNEL_H +#define ONERT_MICRO_TEST_MODELS_NEG_CUMSUM_KERNEL_H + +#include "TestDataCumSumBase.h" + +// clang-format off + +namespace onert_micro::test_model +{ +namespace neg_cumsum_axis_shape_mismatch +{ + +/* + * CumSum Kernel with axis shape mismatch (should be a scalar): + * + * Input(1, 1, 2, 4): Int32 + * | + * CumSum(axis=[2]) + * | + * Output(1, 1, 2, 4): Int32 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x44, 0x00, 0x00, 0x00, 0x7c, 0x01, 0x00, 0x00, 0x98, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x30, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x04, 0x00, 0x06, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x8c, 0xff, 0xff, 0xff, 0x90, 0xff, 0xff, 0xff, + 0x94, 0xff, 0xff, 0xff, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, + 0x18, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, + 0x14, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, 0x60, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00, 0x00, + 0x68, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x6d, 0x61, 0x69, 0x6e, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x0c, 0x00, 0x07, 0x00, 0x08, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66, + 0x10, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x04, 0x00, 0x04, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x78, 0x00, 0x00, 0x00, + 0x40, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0xa0, 0xff, 0xff, 0xff, 0x0c, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6f, 0x66, 0x6d, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0f, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, + 0x10, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x08, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x69, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x04, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, + 0x11, 0x00, 0x00, 0x00, 0x4f, 0x4e, 0x45, 0x2d, 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, + 0x69, 0x72, 0x63, 0x6c, 0x65, 0x00, 0x00, 0x00 +}; + +} // namespace neg_cumsum_axis_shape_mismatch + +// ------------------------------------------------------------------------------------------------ + +namespace neg_cumsum_axis_type_mismatch +{ + +/* + * CumSum Kernel with axis shape mismatch (should be Int32): + * + * Input(1, 1, 2, 4): Int32 + * | + * CumSum(axis=2: Float32) + * | + * Output(1, 1, 2, 4): Int32 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x44, 0x00, 0x00, 0x00, 0x68, 0x01, 0x00, 0x00, 0x84, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x30, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x04, 0x00, 0x06, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x40, 0x8c, 0xff, 0xff, 0xff, 0x90, 0xff, 0xff, 0xff, + 0x94, 0xff, 0xff, 0xff, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6f, 0x66, 0x6d, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0xe0, 0xff, 0xff, 0xff, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x03, 0x00, 0x00, 0x00, 0x69, 0x66, 0x6d, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, 0x11, 0x00, 0x00, 0x00, + 0x4f, 0x4e, 0x45, 0x2d, 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, 0x69, 0x72, 0x63, 0x6c, + 0x65, 0x00, 0x00, 0x00 +}; + +} // namespace neg_cumsum_axis_type_mismatch + +// ------------------------------------------------------------------------------------------------ + +namespace neg_cumsum_input_shape_mismatch +{ + +/* + * CumSum Kernel with input shape mismatch (can't be a scalar): + * + * Input(): Float32 + * | + * CumSum(axis=2: Int32) + * | + * Output(): Int32 + */ + +const unsigned char test_kernel_model_circle[] = { + 0x18, 0x00, 0x00, 0x00, 0x43, 0x49, 0x52, 0x30, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x0c, 0x00, 0x08, 0x00, 0x10, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, + 0x44, 0x00, 0x00, 0x00, 0x58, 0x01, 0x00, 0x00, 0x74, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x30, 0x00, 0x00, 0x00, 0x28, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x06, 0x00, 0x08, 0x00, 0x04, 0x00, 0x06, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x8c, 0xff, 0xff, 0xff, 0x90, 0xff, 0xff, 0xff, + 0x94, 0xff, 0xff, 0xff, 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, + 0x18, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x08, 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, + 0x14, 0x00, 0x00, 0x00, 0x1c, 0x00, 0x00, 0x00, 0x60, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00, 0x00, + 0x68, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x6d, 0x61, 0x69, 0x6e, 0x00, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x14, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x0c, 0x00, 0x07, 0x00, 0x08, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66, + 0x10, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x04, 0x00, 0x04, 0x00, + 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, + 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00, 0x00, + 0x30, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0xb4, 0xff, 0xff, 0xff, 0x0c, 0x00, 0x00, 0x00, + 0x02, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x6f, 0x66, 0x6d, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x14, 0x00, 0x10, 0x00, 0x0f, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, + 0x10, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x61, 0x78, 0x69, 0x73, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x10, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x08, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x00, + 0x03, 0x00, 0x00, 0x00, 0x69, 0x66, 0x6d, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, + 0x10, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x0c, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, + 0x0c, 0x00, 0x00, 0x00, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7f, 0x11, 0x00, 0x00, 0x00, + 0x4f, 0x4e, 0x45, 0x2d, 0x74, 0x66, 0x6c, 0x69, 0x74, 0x65, 0x32, 0x63, 0x69, 0x72, 0x63, 0x6c, + 0x65, 0x00, 0x00, 0x00 +}; + +} // namespace neg_cumsum_input_shape_mismatch + +// ------------------------------------------------------------------------------------------------ + +class NegTestDataCumSumAxisShapeMismatch : public NegTestDataBase +{ +public: + NegTestDataCumSumAxisShapeMismatch() + { + _test_kernel_model_circle = neg_cumsum_axis_shape_mismatch::test_kernel_model_circle; + } + + ~NegTestDataCumSumAxisShapeMismatch() override = default; + + const unsigned char *get_model_ptr() override final + { + return _test_kernel_model_circle; + } + +protected: + const unsigned char *_test_kernel_model_circle; +}; + +// ------------------------------------------------------------------------------------------------ + +class NegTestDataCumSumAxisTypeMismatch : public NegTestDataBase +{ +public: + NegTestDataCumSumAxisTypeMismatch() + { + _test_kernel_model_circle = neg_cumsum_axis_type_mismatch::test_kernel_model_circle; + } + + ~NegTestDataCumSumAxisTypeMismatch() override = default; + + const unsigned char *get_model_ptr() override final + { + return _test_kernel_model_circle; + } + +protected: + const unsigned char *_test_kernel_model_circle; +}; + +// ------------------------------------------------------------------------------------------------ + +class NegTestDataCumSumInputShapeMismatch : public NegTestDataBase +{ +public: + NegTestDataCumSumInputShapeMismatch() + { + _test_kernel_model_circle = neg_cumsum_input_shape_mismatch::test_kernel_model_circle; + } + + ~NegTestDataCumSumInputShapeMismatch() override = default; + + const unsigned char *get_model_ptr() override final + { + return _test_kernel_model_circle; + } + +protected: + const unsigned char *_test_kernel_model_circle; +}; + +} // namespace onert_micro::test_model + +#endif // ONERT_MICRO_TEST_MODELS_NEG_CUMSUM_KERNEL_H diff --git a/onert-micro/onert-micro/include/test_models/cum_sum/TestDataCumSumBase.h b/onert-micro/onert-micro/include/test_models/cum_sum/TestDataCumSumBase.h new file mode 100644 index 00000000000..f0611bf214b --- /dev/null +++ b/onert-micro/onert-micro/include/test_models/cum_sum/TestDataCumSumBase.h @@ -0,0 +1,93 @@ +/* + * Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved + * + * 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 + * + * http://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 ONERT_MICRO_TEST_MODELS_CUMSUM_KERNEL_BASE_H +#define ONERT_MICRO_TEST_MODELS_CUMSUM_KERNEL_BASE_H + +#include "test_models/TestDataBase.h" + +namespace onert_micro::test_model +{ + +template class TestDataCumSumBase : public TestDataBase +{ +public: + TestDataCumSumBase() = default; + ~TestDataCumSumBase() override = default; + + // clang-format off + + const unsigned char *get_model_ptr() override final + { + return _test_kernel_model_circle; + } + + // clang-format on + + const std::vector &get_input_data_by_index(int i) override final + { + switch (i) + { + case 0: + return _input_data; + default: + assert(false && "Wrong input index"); + } + } + + const uint8_t *get_input_data_ptr_by_index(int i, int32_t &size) override final + { + switch (i) + { + case 0: + return get_vector_data(_input_data, size); + case 1: + return get_vector_data(_axis_data, size); + default: + assert(false && "Wrong input index"); + } + } + + const std::vector &get_output_data_by_index(int i) override final + { + assert(i == 0); + return _reference_output_data; + } + +protected: + template + const uint8_t *get_vector_data(const std::vector &vector, int32_t &size) + { + size = vector.size(); + + if (size == 0) + return nullptr; + + size *= sizeof(DataType); + + return reinterpret_cast(vector.data()); + } + +protected: + std::vector _axis_data = {}; + std::vector _input_data; + std::vector _reference_output_data; + const unsigned char *_test_kernel_model_circle; +}; + +} // namespace onert_micro::test_model + +#endif // ONERT_MICRO_TEST_MODELS_CUMSUM_KERNEL_BASE_H diff --git a/onert-micro/onert-micro/src/execute/kernels/CumSum.cpp b/onert-micro/onert-micro/src/execute/kernels/CumSum.cpp new file mode 100644 index 00000000000..827473982e4 --- /dev/null +++ b/onert-micro/onert-micro/src/execute/kernels/CumSum.cpp @@ -0,0 +1,124 @@ +/* + * Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved + * + * 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 + * + * http://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 "core/OMUtils.h" +#include "core/OMKernelData.h" + +#include "execute/OMUtils.h" + +#include "PALCumSum.h" + +using onert_micro::OMStatus; + +using namespace onert_micro::core; +using namespace onert_micro::execute; + +// ------------------------------------------------------------------------------------------------ + +namespace impl +{ + +constexpr static uint32_t axisTensorIdx = 1; +constexpr static uint32_t inputTensorIdx = 0; +constexpr static uint32_t outputTensorIdx = 0; + +template OMStatus CircleCumSum(OMRuntimeKernel &rt_kernel) +{ + const circle::Tensor *axis = rt_kernel.inputs[axisTensorIdx]; + const circle::Tensor *input = rt_kernel.inputs[inputTensorIdx]; + const circle::Tensor *output = rt_kernel.outputs[outputTensorIdx]; + + assert(axis != nullptr); + assert(input != nullptr); + assert(output != nullptr); + + const int32_t *axis_data = utils::castInputData(rt_kernel.inputs_data[axisTensorIdx]); + const T *input_data = utils::castInputData(rt_kernel.inputs_data[inputTensorIdx]); + T *output_data = utils::castOutputData(rt_kernel.outputs_data[outputTensorIdx]); + + assert(input_data != nullptr); + assert(output_data != nullptr); + + OMRuntimeShape input_shape(input); + + const int32_t rank = input_shape.dimensionsCount(); + const int32_t axis_value = (axis_data != nullptr) ? *axis_data : 0; + + assert(rank >= 1); + assert(axis_value >= 0); + assert(axis_value < rank); + + auto options = rt_kernel.first_operator->builtin_options_as_CumsumOptions(); + + assert(options != nullptr); + + CumSumParams params{}; + params.exclusive = options->exclusive(); + params.reverse = options->reverse(); + + return pal::CumSum(params, input_shape, input_data, axis_value, output_data); +} + +} // namespace impl + +// ------------------------------------------------------------------------------------------------ + +namespace onert_micro::execute +{ + +OMStatus execute_kernel_CircleCumSum(const OMExecuteArgs &execute_args) +{ + core::OMRuntimeContext &rt_context = execute_args.runtime_context; + core::OMRuntimeStorage &rt_storage = execute_args.runtime_storage; + + uint16_t op_index = execute_args.kernel_index; + + OMRuntimeKernel rt_kernel; + OMStatus status = rt_kernel.readKernel(op_index, rt_context); + + if (status != Ok) + return status; + + const circle::Tensor *input = rt_kernel.inputs[impl::inputTensorIdx]; + assert(input != nullptr); + + status = rt_kernel.getDataFromStorage(op_index, rt_storage, rt_context); + + if (status != Ok) + return status; + + switch (input->type()) + { + case circle::TensorType_INT32: + return impl::CircleCumSum(rt_kernel); + + case circle::TensorType_INT64: + return impl::CircleCumSum(rt_kernel); + +#ifndef DIS_FLOAT + + case circle::TensorType_FLOAT32: + return impl::CircleCumSum(rt_kernel); + +#endif // DIS__FLOAT + + default: + assert(false && "Unsupported type."); + return UnsupportedType; + } +} + +} // namespace onert_micro::execute diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/CumSum.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/CumSum.test.cpp new file mode 100644 index 00000000000..15da4ed24a1 --- /dev/null +++ b/onert-micro/onert-micro/src/execute/kernels/tests/CumSum.test.cpp @@ -0,0 +1,149 @@ +/* + * Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved + * + * 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 + * + * http://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 "execute/OMTestUtils.h" + +#include "test_models/cum_sum/FloatCumSumKernel.h" +#include "test_models/cum_sum/IntCumSumKernel.h" +#include "test_models/cum_sum/NegCumSumKernel.h" + +namespace onert_micro::execute::testing +{ + +// ------------------------------------------------------------------------------------------------ + +class CumSumTest : public ::testing::Test +{ +}; + +// ------------------------------------------------------------------------------------------------ + +template +std::vector checkCumSumKernel(test_model::TestDataCumSumBase *test_data_base) +{ + onert_micro::OMInterpreter interpreter; + onert_micro::OMConfig config; + + auto model_ptr = reinterpret_cast(test_data_base->get_model_ptr()); + + interpreter.importModel(model_ptr, config); + interpreter.reset(); + interpreter.allocateInputs(); + + T *input_data = reinterpret_cast(interpreter.getInputDataAt(0)); + const auto &input_data_ref = test_data_base->get_input_data_by_index(0); + + std::copy(input_data_ref.begin(), input_data_ref.end(), input_data); + + auto num_inputs = interpreter.getNumberOfInputs(); + + if (num_inputs >= 2) + { + uint8_t *axis_data = reinterpret_cast(interpreter.getInputDataAt(1)); + + int32_t size; + const uint8_t *axis_data_ref = test_data_base->get_input_data_ptr_by_index(1, size); + + if (size > 0) + { + std::copy(axis_data_ref, axis_data_ref + size, axis_data); + } + } + + interpreter.run(config); + + T *output_data = reinterpret_cast(interpreter.getOutputDataAt(0)); + const size_t num_elements = interpreter.getOutputSizeAt(0); + + std::vector output_data_vector(output_data, output_data + num_elements); + + return output_data_vector; +} + +// ------------------------------------------------------------------------------------------------ + +TEST_F(CumSumTest, Float) +{ + test_model::TestDataFloatCumSum test_data; + auto output_data_vector = checkCumSumKernel(&test_data); + + EXPECT_TRUE(output_data_vector == test_data.get_output_data_by_index(0)); +} + +TEST_F(CumSumTest, Int32) +{ + test_model::TestDataInt32CumSum test_data; + auto output_data_vector = checkCumSumKernel(&test_data); + + EXPECT_TRUE(output_data_vector == test_data.get_output_data_by_index(0)); +} + +TEST_F(CumSumTest, Int64) +{ + test_model::TestDataInt64CumSum test_data; + auto output_data_vector = checkCumSumKernel(&test_data); + + EXPECT_TRUE(output_data_vector == test_data.get_output_data_by_index(0)); +} + +TEST_F(CumSumTest, Float_TwoInputs) +{ + test_model::TestDataTwoInputsFloatCumSum test_data; + auto output_data_vector = checkCumSumKernel(&test_data); + + EXPECT_TRUE(output_data_vector == test_data.get_output_data_by_index(0)); +} + +TEST_F(CumSumTest, Float_Reverse) +{ + test_model::TestDataReverseFloatCumSum test_data; + auto output_data_vector = checkCumSumKernel(&test_data); + + EXPECT_TRUE(output_data_vector == test_data.get_output_data_by_index(0)); +} + +TEST_F(CumSumTest, Float_Exclusive) +{ + test_model::TestDataExclusiveFloatCumSum test_data; + auto output_data_vector = checkCumSumKernel(&test_data); + + EXPECT_TRUE(output_data_vector == test_data.get_output_data_by_index(0)); +} + +TEST_F(CumSumTest, AxisShapeMismatch_NEG) +{ + test_model::NegTestDataCumSumAxisShapeMismatch neg_test_data; + + EXPECT_DEATH(checkNEGSISOKernel(&neg_test_data), ""); +} + +TEST_F(CumSumTest, AxisTypeMismatch_NEG) +{ + test_model::NegTestDataCumSumAxisTypeMismatch neg_test_data; + + EXPECT_DEATH(checkNEGSISOKernel(&neg_test_data), ""); +} + +TEST_F(CumSumTest, InputShapeMismatch_NEG) +{ + test_model::NegTestDataCumSumInputShapeMismatch neg_test_data; + + EXPECT_DEATH(checkNEGSISOKernel(&neg_test_data), ""); +} + +// ------------------------------------------------------------------------------------------------ + +} // namespace onert_micro::execute::testing diff --git a/onert-micro/onert-micro/src/import/kernels/CumSum.cpp b/onert-micro/onert-micro/src/import/kernels/CumSum.cpp new file mode 100644 index 00000000000..e8f8cead0a4 --- /dev/null +++ b/onert-micro/onert-micro/src/import/kernels/CumSum.cpp @@ -0,0 +1,73 @@ +/* + * Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved + * + * 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 + * + * http://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 "core/OMUtils.h" +#include "import/OMUtils.h" + +using namespace onert_micro; +using namespace onert_micro::core::utils; + +namespace onert_micro::import +{ + +OMStatus configure_kernel_CircleCumSum(const OMConfigureArgs &config_args) +{ + constexpr static uint32_t axisTensorIdx = 1; + constexpr static uint32_t inputTensorIdx = 0; + constexpr static uint32_t outputTensorIdx = 0; + + execute::OMRuntimeKernel rt_kernel; + core::OMRuntimeContext &rt_context = config_args.runtime_context; + + OMStatus status = rt_kernel.readKernel(config_args.kernel_index, rt_context); + + if (status != Ok) + return status; + + const circle::Tensor *axis = rt_kernel.inputs[axisTensorIdx]; + const circle::Tensor *input = rt_kernel.inputs[inputTensorIdx]; + const circle::Tensor *output = rt_kernel.outputs[outputTensorIdx]; + + assert(axis != nullptr); + assert(input != nullptr); + assert(output != nullptr); + + core::OMRuntimeShape axis_shape(axis); + core::OMRuntimeShape input_shape(input); + + // clang-format off + + std::array conditions = { + input->type() == output->type(), + input_shape.dimensionsCount() > 1, + axis->type() == circle::TensorType_INT32, + axis_shape.isScalar() + }; + + bool status_check = std::all_of(conditions.cbegin(), conditions.cend(), [](bool condition) + { + return checkCondition(condition) == Ok; + }); + + // clang-format on + + if (!status_check) + return FailedCheckCondition; + + return Ok; +} + +} // namespace onert_micro::import