diff --git a/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/cast_base.c b/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/cast_base.c index d065c0b95ea0431b6c8f00d808add337c573da46..9d3d5c47e331a60cf4dc5209cce3fa13e6a9be0c 100644 --- a/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/cast_base.c +++ b/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/cast_base.c @@ -49,6 +49,12 @@ void Uint8ToFloat32(const uint8_t *input, float *output, int number) { } } +void Int8ToFloat32(const int8_t *input, float *output, int number) { + for (int i = 0; i < number; ++i) { + output[i] = (float)input[i]; + } +} + void Int32ToFloat32(const int32_t *input, float *output, int number); void Int64ToFloat32(const int64_t *input, float *output, int number) { diff --git a/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/cast_base.h b/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/cast_base.h index 5e87bd5f758839c9536c087c24c68caebec3c5a5..21a2f34ebd09d88fb7c8ebb137ca0a3736e1b774 100644 --- a/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/cast_base.h +++ b/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/cast_base.h @@ -27,6 +27,8 @@ void BoolToFloat32(const bool *input, float *output, int number); void Uint8ToFloat32(const uint8_t *input, float *output, int number); +void Int8ToFloat32(const int8_t *input, float *output, int number); + void Int32ToFloat32(const int32_t *input, float *output, int number); void Int64ToFloat32(const int64_t *input, float *output, int number); diff --git a/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/cast_fp32.cc b/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/cast_fp32.cc index 08aec60c23aaf5d017c294ff737828f017df516c..a7ffc534f6c1e3613ff06261fa7d8ef24570e526 100644 --- a/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/cast_fp32.cc +++ b/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/cast_fp32.cc @@ -75,6 +75,10 @@ int CastCPUKernel::CastToFp32(const lite::Tensor *input, lite::Tensor *output, i Uint8ToFloat32(reinterpret_cast(input->data()) + offset, reinterpret_cast(output_data) + offset, data_num); break; + case kNumberTypeInt8: + Int8ToFloat32(reinterpret_cast(input->data()) + offset, + reinterpret_cast(output_data) + offset, data_num); + break; case kNumberTypeInt32: Int32ToFloat32(reinterpret_cast(input->data()) + offset, reinterpret_cast(output_data) + offset, data_num);