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

!8287 add supported data types for cast ops 

From: @liuwenhao4
Reviewed-by: @zhanghaibo5,@zhanghaibo5,@HilbertDavid,@zhang_xue_tong
Signed-off-by:
This commit is contained in:
mindspore-ci-bot 2020-11-09 09:05:50 +08:00 committed by Gitee
parent 027916cb5f bcaf43e0fb
commit e74355f3e1
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mindspore/lite/src/runtime/kernel/arm/fp32/cast_fp32.cc
View File

@ -17,6 +17,7 @@
#include <vector>
#include "schema/model_generated.h"
#include "src/kernel_registry.h"
#include "src/tensor.h"
#include "nnacl/fp32/cast.h"
#include "nnacl/op_base.h"
#include "src/runtime/runtime_api.h"
@ -70,6 +71,12 @@ int CastCPUKernel::DoCast(int thread_id) {
MS_ASSERT(output_data != nullptr);
auto input_data_type = input->data_type();
auto output_data_type = output->data_type();
if (input_data_type == output_data_type) {
auto datalen = lite::DataTypeSize(input_data_type);
memcpy(reinterpret_cast<char *>(output_data) + offset * datalen,
reinterpret_cast<char *>(input->data_c()) + offset * datalen, data_num * datalen);
return RET_OK;
}
if (output_data_type != kNumberTypeFloat32) {
if (input_data_type == kNumberTypeFloat32 && output_data_type == kNumberTypeInt64) {
Float32ToInt64(reinterpret_cast<float *>(input->data_c()) + offset,
@ -83,9 +90,6 @@ int CastCPUKernel::DoCast(int thread_id) {
} else if (input_data_type == kNumberTypeInt32 && output_data_type == kNumberTypeInt64) {
Int32ToInt64(reinterpret_cast<int32_t *>(input->data_c()) + offset,
reinterpret_cast<int64_t *>(output_data) + offset, data_num);
} else if (input_data_type == kNumberTypeInt32 && output_data_type == kNumberTypeInt32) {
memcpy(reinterpret_cast<int32_t *>(output_data) + offset, reinterpret_cast<int32_t *>(input->data_c()) + offset,
data_num * sizeof(int32_t));
} else {
MS_LOG(ERROR) << "Unsupported datatype from " << input_data_type << " to " << output_data_type;
return RET_ERROR;
@ -108,10 +112,6 @@ int CastCPUKernel::DoCast(int thread_id) {
Fp16ToFloat32(reinterpret_cast<uint16_t *>(input->MutableData()) + offset,
reinterpret_cast<float *>(output_data) + offset, data_num);
break;
case kNumberTypeFloat32:
memcpy(reinterpret_cast<float *>(output_data) + offset, reinterpret_cast<float *>(input->data_c()) + offset,
data_num * sizeof(float));
break;
default:
MS_LOG(ERROR) << "Unsupported input data type " << input_data_type;
return RET_ERROR;