diff --git a/fastdeploy/backends/lite/lite_backend.cc b/fastdeploy/backends/lite/lite_backend.cc index 20afa5e28..c2a783ea4 100755 --- a/fastdeploy/backends/lite/lite_backend.cc +++ b/fastdeploy/backends/lite/lite_backend.cc @@ -249,12 +249,12 @@ bool LiteBackend::Infer(std::vector& inputs, reinterpret_cast(const_cast( inputs[i].CpuData()))); } else if (inputs[i].dtype == FDDataType::INT64) { -#ifdef __aarch64__ +#if (defined(__aarch64__) || defined(__x86_64__) || defined(_M_X64) || defined(_M_ARM64)) tensor->CopyFromCpu( reinterpret_cast(const_cast( inputs[i].CpuData()))); #else - FDASSERT(false, "FDDataType::INT64 is not support for Arm v7 now!"); + FDASSERT(false, "FDDataType::INT64 is not support for x86/armv7 now!"); #endif } else { FDASSERT(false, "Unexpected data type of %d.", inputs[i].dtype);