p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

power bug fix

This commit is contained in:
lzk 2021-03-23 18:14:21 -07:00
parent 51225d58cc
commit 0f900c8d53
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/nnacl/fp16/power_fp16.h
View File

@ -30,7 +30,7 @@ typedef void (*PowerFunFp16)(const float16_t *, const float16_t *, float16_t *,
#ifdef __cplusplus
extern "C" {
#endif
inline bool CheckInteger(float16_t f) { return floorf(f) == f; }
static inline bool CheckInteger(float16_t f) { return floorf(f) == f; }
static inline float16_t StdPowerScalarFp16(float16_t x, const void *exponent) {
return powf(x, *(float16_t *)exponent);

2
mindspore/lite/nnacl/fp32/pack_fp32.c
View File

@ -308,7 +308,7 @@ void PackNHWCToNCHWFp32(const void *src, void *dst, int batches, int plane, int
return;
}
task_end = (task_id + 1) == thread_count ? plane : MSMIN(plane, task_start + offset_hw);
hw8 = task_start + (task_end - task_start) > offset_hw ? offset_hw : 0;
hw8 = task_start + (task_end - task_start) >= offset_hw ? offset_hw : 0;
} else {
hw8 *= C8NUM;
}

2
mindspore/lite/nnacl/fp32/power_fp32.h
View File

@ -30,7 +30,7 @@ typedef float (*PowerScalarFun)(float x, const void *exponent);
#ifdef __cplusplus
extern "C" {
#endif
inline bool CheckInteger(float f) { return floorf(f) == f; }
static inline bool CheckInteger(float f) { return floorf(f) == f; }
static inline float StdPowerScalar(float x, const void *exponent) { return powf(x, *(float *)exponent); }