apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 00:57:33 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Bug Fix] add __aarch64__ to limit int64 in armv8 (#794)

Browse Source
This commit is contained in:
DefTruth
2022-12-05 18:19:11 +08:00
committed by GitHub
parent cdae9f0482
commit ffea55e038
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/backends/lite/lite_backend.cc
View File

@@ -222,9 +222,13 @@ bool LiteBackend::Infer(std::vector<FDTensor>& inputs,
reinterpret_cast<const uint8_t*>(const_cast<void*>(
inputs[i].CpuData())));
} else if (inputs[i].dtype == FDDataType::INT64) {
#ifdef __aarch64__
tensor->CopyFromCpu<int64_t, paddle::lite_api::TargetType::kARM>(
reinterpret_cast<const int64_t*>(const_cast<void*>(
inputs[i].CpuData())));
#else
FDASSERT(false, "FDDataType::INT64 is not support for Arm v7 now!");
#endif
} else {
FDASSERT(false, "Unexpected data type of %d.", inputs[i].dtype);
}