Browse Source

power bug fix

pull/13898/head
lzk 4 years ago
parent
commit
0f900c8d53
3 changed files with 3 additions and 3 deletions
  1. +1
    -1
      mindspore/lite/nnacl/fp16/power_fp16.h
  2. +1
    -1
      mindspore/lite/nnacl/fp32/pack_fp32.c
  3. +1
    -1
      mindspore/lite/nnacl/fp32/power_fp32.h

+ 1
- 1
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);


+ 1
- 1
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;
}


+ 1
- 1
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); }



Loading…
Cancel
Save