Browse Source

!29926 [MSLITE][CPU] fuzz problem, cast_fp32.cc bugfix

Merge pull request !29926 from Greatpan/fuzz_problem
feature/build-system-rewrite
i-robot Gitee 4 years ago
parent
commit
87729d9226
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/fp32/cast_fp32.cc

+ 1
- 0
mindspore/lite/src/runtime/kernel/arm/fp32/cast_fp32.cc View File

@@ -106,6 +106,7 @@ int CastCPUKernel::CastToFp16(const lite::Tensor *input, lite::Tensor *output, i
case kNumberTypeInt64:
Int64ToFp16(reinterpret_cast<const int64_t *>(input->data()) + offset,
reinterpret_cast<float16_t *>(output_data) + offset, data_num);
break;
case kNumberTypeInt32:
Int32ToFp16(reinterpret_cast<const int32_t *>(input->data()) + offset,
reinterpret_cast<float16_t *>(output_data) + offset, data_num);


Loading…
Cancel
Save