From 37f774079638911177e70fe7020bd13427c8eec7 Mon Sep 17 00:00:00 2001 From: yiguangzheng Date: Mon, 1 Dec 2025 14:14:37 +0800 Subject: [PATCH] fix unsorted_segment_sum --- .../cpu/{fp16_grad => fp16}/unsorted_segment_sum_fp16.cc | 4 ++-- .../cpu/{fp16_grad => fp16}/unsorted_segment_sum_fp16.h | 6 +++--- .../kernel/cpu/{fp32_grad => fp32}/unsorted_segment_sum.cc | 2 +- .../kernel/cpu/{fp32_grad => fp32}/unsorted_segment_sum.h | 6 +++--- .../cpu/nnacl_c/{fp16_grad => fp16}/unsorted_segment_sum.c | 2 +- .../cpu/nnacl_c/{fp16_grad => fp16}/unsorted_segment_sum.h | 6 +++--- 6 files changed, 13 insertions(+), 13 deletions(-) rename mindspore-lite/src/litert/kernel/cpu/{fp16_grad => fp16}/unsorted_segment_sum_fp16.cc (96%) rename mindspore-lite/src/litert/kernel/cpu/{fp16_grad => fp16}/unsorted_segment_sum_fp16.h (83%) rename mindspore-lite/src/litert/kernel/cpu/{fp32_grad => fp32}/unsorted_segment_sum.cc (98%) rename mindspore-lite/src/litert/kernel/cpu/{fp32_grad => fp32}/unsorted_segment_sum.h (84%) rename mindspore-lite/src/litert/kernel/cpu/nnacl_c/{fp16_grad => fp16}/unsorted_segment_sum.c (95%) rename mindspore-lite/src/litert/kernel/cpu/nnacl_c/{fp16_grad => fp16}/unsorted_segment_sum.h (85%) diff --git a/mindspore-lite/src/litert/kernel/cpu/fp16_grad/unsorted_segment_sum_fp16.cc b/mindspore-lite/src/litert/kernel/cpu/fp16/unsorted_segment_sum_fp16.cc similarity index 96% rename from mindspore-lite/src/litert/kernel/cpu/fp16_grad/unsorted_segment_sum_fp16.cc rename to mindspore-lite/src/litert/kernel/cpu/fp16/unsorted_segment_sum_fp16.cc index f6e1eefc..d4d236de 100644 --- a/mindspore-lite/src/litert/kernel/cpu/fp16_grad/unsorted_segment_sum_fp16.cc +++ b/mindspore-lite/src/litert/kernel/cpu/fp16/unsorted_segment_sum_fp16.cc @@ -14,12 +14,12 @@ * limitations under the License. */ -#include "src/litert/kernel/cpu/fp16_grad/unsorted_segment_sum_fp16.h" +#include "src/litert/kernel/cpu/fp16/unsorted_segment_sum_fp16.h" #include #include #include "schema/model_generated.h" #include "src/litert/kernel_registry.h" -#include "nnacl_c/fp16_grad/unsorted_segment_sum.h" +#include "nnacl_c/fp16/unsorted_segment_sum.h" #include "include/errorcode.h" using mindspore::kernel::KERNEL_ARCH::kCPU; diff --git a/mindspore-lite/src/litert/kernel/cpu/fp16_grad/unsorted_segment_sum_fp16.h b/mindspore-lite/src/litert/kernel/cpu/fp16/unsorted_segment_sum_fp16.h similarity index 83% rename from mindspore-lite/src/litert/kernel/cpu/fp16_grad/unsorted_segment_sum_fp16.h rename to mindspore-lite/src/litert/kernel/cpu/fp16/unsorted_segment_sum_fp16.h index 65e11e0e..bbdf813a 100644 --- a/mindspore-lite/src/litert/kernel/cpu/fp16_grad/unsorted_segment_sum_fp16.h +++ b/mindspore-lite/src/litert/kernel/cpu/fp16/unsorted_segment_sum_fp16.h @@ -14,8 +14,8 @@ * limitations under the License. */ -#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP16_GRAD_UNSORTED_SEGMENT_SUM_FP16_H_ -#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP16_GRAD_UNSORTED_SEGMENT_SUM_FP16_H_ +#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP16_UNSORTED_SEGMENT_SUM_FP16_H_ +#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP16_UNSORTED_SEGMENT_SUM_FP16_H_ #include #include "src/executor/kernel_exec.h" @@ -41,4 +41,4 @@ class UnsortedSegmentSumCPUKernelFp16 : public LiteKernel { }; } // namespace mindspore::kernel -#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP16_GRAD_UNSORTED_SEGMENT_SUM_FP16_H_ +#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP16_UNSORTED_SEGMENT_SUM_FP16_H_ diff --git a/mindspore-lite/src/litert/kernel/cpu/fp32_grad/unsorted_segment_sum.cc b/mindspore-lite/src/litert/kernel/cpu/fp32/unsorted_segment_sum.cc similarity index 98% rename from mindspore-lite/src/litert/kernel/cpu/fp32_grad/unsorted_segment_sum.cc rename to mindspore-lite/src/litert/kernel/cpu/fp32/unsorted_segment_sum.cc index ae714355..8f2edc23 100644 --- a/mindspore-lite/src/litert/kernel/cpu/fp32_grad/unsorted_segment_sum.cc +++ b/mindspore-lite/src/litert/kernel/cpu/fp32/unsorted_segment_sum.cc @@ -14,7 +14,7 @@ * limitations under the License. */ -#include "src/litert/kernel/cpu/fp32_grad/unsorted_segment_sum.h" +#include "src/litert/kernel/cpu/fp32/unsorted_segment_sum.h" #include #include #include "schema/model_generated.h" diff --git a/mindspore-lite/src/litert/kernel/cpu/fp32_grad/unsorted_segment_sum.h b/mindspore-lite/src/litert/kernel/cpu/fp32/unsorted_segment_sum.h similarity index 84% rename from mindspore-lite/src/litert/kernel/cpu/fp32_grad/unsorted_segment_sum.h rename to mindspore-lite/src/litert/kernel/cpu/fp32/unsorted_segment_sum.h index 7bac2af1..dcc8d38d 100644 --- a/mindspore-lite/src/litert/kernel/cpu/fp32_grad/unsorted_segment_sum.h +++ b/mindspore-lite/src/litert/kernel/cpu/fp32/unsorted_segment_sum.h @@ -14,8 +14,8 @@ * limitations under the License. */ -#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP32_GRAD_UNSORTED_SEGMENT_SUM_H_ -#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP32_GRAD_UNSORTED_SEGMENT_SUM_H_ +#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP32_UNSORTED_SEGMENT_SUM_H_ +#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP32_UNSORTED_SEGMENT_SUM_H_ #include #include "src/litert/lite_kernel.h" @@ -41,4 +41,4 @@ class UnsortedSegmentSumCPUKernel : public LiteKernel { }; } // namespace mindspore::kernel -#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP32_GRAD_UNSORTED_SEGMENT_SUM_H_ +#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_CPU_FP32_UNSORTED_SEGMENT_SUM_H_ diff --git a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16_grad/unsorted_segment_sum.c b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16/unsorted_segment_sum.c similarity index 95% rename from mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16_grad/unsorted_segment_sum.c rename to mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16/unsorted_segment_sum.c index 8f794d42..5d370893 100644 --- a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16_grad/unsorted_segment_sum.c +++ b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16/unsorted_segment_sum.c @@ -13,7 +13,7 @@ * See the License for the specific language governing permissions and * limitations under the License. */ -#include "nnacl_c/fp16_grad/unsorted_segment_sum.h" +#include "nnacl_c/fp16/unsorted_segment_sum.h" #include "nnacl_c/errorcode.h" int UnsortedSegmentSumFp16(const float16_t *input, int unit_num, int input_dim1, const int *indices, float16_t *output, diff --git a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16_grad/unsorted_segment_sum.h b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16/unsorted_segment_sum.h similarity index 85% rename from mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16_grad/unsorted_segment_sum.h rename to mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16/unsorted_segment_sum.h index 85a54ab2..4ca39c6b 100644 --- a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16_grad/unsorted_segment_sum.h +++ b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp16/unsorted_segment_sum.h @@ -14,8 +14,8 @@ * limitations under the License. */ -#ifndef NNACL_FP16_GRAD_UNSORTED_SEGMENT_SUM_H_ -#define NNACL_FP16_GRAD_UNSORTED_SEGMENT_SUM_H_ +#ifndef NNACL_FP16_UNSORTED_SEGMENT_SUM_H_ +#define NNACL_FP16_UNSORTED_SEGMENT_SUM_H_ #include "nnacl_c/op_base.h" @@ -28,4 +28,4 @@ int UnsortedSegmentSumFp16(const float16_t *input, int unit_num, int input_dim1, #ifdef __cplusplus } #endif -#endif // NNACL_FP16_GRAD_UNSORTED_SEGMENT_SUM_H_ +#endif // NNACL_FP16_UNSORTED_SEGMENT_SUM_H_ -- Gitee